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

ASAM:深度神经网络尺度不变学习的自适应锐度感知最小化

ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks

Samsung Research(三星研究院)

Jungmin Kwon, Jeongseop Kim, Hyunseo Park, In Kwon Choi

摘要:

近年来,损失面的尖锐度作为泛化间隙的有效度量,作为学习算法的动机,表现出最先进的性能。然而,在具有固定半径的刚性区域中定义的锐度在对参数重新缩放的敏感性方面具有缺点,这使得损失不受影响,从而导致锐度和泛化间隙之间的连接减弱。在本文中,我们引入了自适应锐度的概念,这是规模不变的,并提出了相应的推广界。我们提出了一种新的学习方法,自适应锐度感知最小化(ASAM),利用建议的泛化界。在不同基准数据集上的实验结果表明,ASAM有助于显着提高模型的泛化性能。

英文摘要:

Recently, learning algorithms motivated from sharpness of loss surface as an effective measure of generalization gap have shown state-of-the-art performances. Nevertheless, sharpness defined in a rigid region with a fixed radius, has a drawback in sensitivity to parameter re-scaling which leaves the loss unaffected, leading to weakening of the connection between sharpness and generalization gap. In this paper, we introduce the concept of adaptive sharpness which is scale-invariant and propose the corresponding generalization bound. We suggest a novel learning method, adaptive sharpness-aware minimization (ASAM), utilizing the proposed generalization bound. Experimental results in various benchmark datasets show that ASAM contributes to significant improvement of model generalization performance.

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

↑