Jun, 2024

无偏锐度感知最小化

TL;DR在这项工作中,我们探索了锐度感知极小化(SAM)和模型不可知元学习(MAML)之间的联系,特别是在增强模型泛化方面。我们引入了一种新方法 Agnostic-SAM,它结合了 SAM 和 MAML 的原则。Agnostic-SAM 通过优化模型朝向具有更广泛的局部最小值,使用训练数据来调整 SAM 的核心思想,并同时在验证数据上保持较低的损失值。通过这样做,它寻求不仅对小扰动具有鲁棒性,而且对数据分布转换问题更加稳健的最小值。我们的实验结果表明,Agnostic-SAM 在各种数据集和具有挑战性的条件下(如噪声标签和数据限制)显著改善了泛化能力。