arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2106.16004cs.LGcs.AIstat.ML
收藏

神经网络损失景观的线性插值能告诉我们什么?

What can linear interpolation of neural network loss landscapes tell us?

Department of Mathematics, University of Edinburgh(爱丁堡大学数学系);CSAIL, MIT(麻省理工学院计算机科学与人工智能实验室)

Tiffany Vlaar, Jonathan Frankle

摘要:

研究神经网络损失景观提供了深入了解基本优化问题的性质。不幸的是,损失景观是出了名的难以想象的人类理解的方式。解决这个问题的一种常见方法是绘制景观的线性切片,例如从网络的初始状态到优化后的最终状态。在此分析的基础上,先前的工作已经得出了更广泛的结论,优化问题的难度。在本文中,我们对这种推断进行了测试,系统地评估了线性插值和最终性能在更改数据、初始化选择以及其他优化器和架构设计选择时的变化。此外,我们使用线性插值来研究网络的各个层和子结构所扮演的角色。我们发现某些层对初始化的选择更敏感,但线性路径的形状并不表示模型测试精度的变化。我们的研究结果质疑更广泛的直觉,即存在或不存在的障碍时,插值必然涉及到优化的成功。

英文摘要:

Studying neural network loss landscapes provides insights into the nature of the underlying optimization problems. Unfortunately, loss landscapes are notoriously difficult to visualize in a human-comprehensible fashion. One common way to address this problem is to plot linear slices of the landscape, for example from the initial state of the network to the final state after optimization. On the basis of this analysis, prior work has drawn broader conclusions about the difficulty of the optimization problem. In this paper, we put inferences of this kind to the test, systematically evaluating how linear interpolation and final performance vary when altering the data, choice of initialization, and other optimizer and architecture design choices. Further, we use linear interpolation to study the role played by individual layers and substructures of the network. We find that certain layers are more sensitive to the choice of initialization, but that the shape of the linear path is not indicative of the changes in test accuracy of the model. Our results cast doubt on the broader intuition that the presence or absence of barriers when interpolating necessarily relates to the success of optimization.

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

↑