arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.06635cs.LGstat.ML

Orthogonalized SGD and Nested Architectures for Anytime Neural Networks

University of Chicago(芝加哥大学)

Chengcheng Wan, Henry Hoffmann, Shan Lu, Michael Maire

英文摘要:

We propose a novel variant of SGD customized for training network architectures that support anytime behavior: such networks produce a series of increasingly accurate outputs over time. Efficient architectural designs for these networks focus on re-using internal state; subnetworks must produce representations relevant for both immediate prediction as well as refinement by subsequent network stages. We consider traditional branched networks as well as a new class of recursively nested networks. Our new optimizer, Orthogonalized SGD, dynamically re-balances task-specific gradients when training a multitask network. In the context of anytime architectures, this optimizer projects gradients from later outputs onto a parameter subspace that does not interfere with those from earlier outputs. Experiments demonstrate that training with Orthogonalized SGD significantly improves generalization accuracy of anytime networks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑