新方法用二阶信息加速深度学习训练,能避开鞍点
Symmetric Rank-One Quasi-Newton Methods for Deep Learning Using Cubic Regularization
- 采用对称秩一拟牛顿法,允许负曲率方向更新
- 结合立方正则化,每步可闭式求解,提升稳定性
- 适合追求更快收敛的模型优化研究者
随机梯度下降及其变体(如Adam、AdaGrad)因计算高效、内存占用低而广泛用于深度学习,但缺乏曲率信息,易陷入鞍点或次优局部极小。相比之下,拟牛顿方法通过重构海森矩阵利用曲率信息,且计算开销相当。主流的L-BFGS方法保证海森矩阵半正定,适用于线搜索,但在非凸的深度神经网络损失函数中,海森矩阵可能不定。本文提出一种有限记忆对称秩一拟牛顿法,支持不定海森逼近,可利用负曲率方向。同时采用改进的自适应立方正则化策略,生成一系列具有闭式解的立方子问题。我们在自编码器和前馈神经网络上评估该方法,与当前主流的一阶自适应方法及其他拟牛顿方法进行对比。
原文摘要 · Abstract (English)
Stochastic gradient descent and other first-order variants, such as Adam and AdaGrad, are commonly used in the field of deep learning due to their computational efficiency and low-storage memory requirements. However, these methods do not exploit curvature information. Consequently, iterates can converge to saddle points or poor local minima. On the other hand, Quasi-Newton methods compute Hessian approximations which exploit this information with a comparable computational budget. Quasi-Newton methods re-use previously computed iterates and gradients to compute a low-rank structured update. The most widely used quasi-Newton update is the L-BFGS, which guarantees a positive semi-definite Hessian approximation, making it suitable in a line search setting. However, the loss functions in DNNs are non-convex, where the Hessian is potentially non-positive definite. In this paper, we propose using a limited-memory symmetric rank-one quasi-Newton approach which allows for indefinite Hessian approximations, enabling directions of negative curvature to be exploited. Furthermore, we use a modified adaptive regularized cubics approach, which generates a sequence of cubic subproblems that have closed-form solutions with suitable regularization choices. We investigate the performance of our proposed method on autoencoders and feed-forward neural network models and compare our approach to state-of-the-art first-order adaptive stochastic methods as well as other quasi-Newton methods.x
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。