AAAIDec, 2023

CR-SAM:曲率规则的锐度感知最小化

TL;DR通过最小化最坏情况损失,使用单步梯度上升作为近似方法,我们提出了曲率正则化 SAM(CR-SAM),通过引入归一化的 Hessian 迹来准确测量训练和测试集上的损失曲线的曲率。我们的实证评估结果显示,CR-SAM 在各种数据集上持续提高了 ResNet 和 Vision Transformer(ViT)模型的分类性能。