研究人员在神经网络的快速对抗性训练(FAT)中识别出一种称为灾难性过拟合(CO)的现象。当对多步攻击的鲁棒性迅速下降时,即使网络在单步攻击和干净数据上的性能持续提高,也会发生 CO。这是因为训练损失对这种权重漂移基本不敏感。该研究提出了一种简单的解决方案:使用权重的指数移动平均值(EMA),该平均值保持鲁棒性,并在验证鲁棒性下降时将权重重置为其 EMA。此方法可有效消除 CO,并在 CIFAR-100 等基准数据集上实现最先进的准确度,同时降低了训练成本。 AI
影响 识别出一种关键的训练故障模式,并提供了一种实用的、低成本的解决方案来提高模型鲁棒性。
排序理由 详细介绍机器学习训练中新发现和拟议解决方案的学术论文。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →