Oct, 2023

黑铁匠:通过混合单步和多步方法实现视觉 Transformer 的快速对抗训练

TL;DR我们提出了一种名为 Blacksmith 的新训练策略,用于解决深度学习算法中的 Catastrophic Overfitting 问题,尤其适用于 Vision Transformers。通过在神经网络的对抗训练中随机使用 PGD-2 或 Fast Gradient Sign Method (FGSM) 攻击,我们的方法有效地防止 Catastrophic Overfitting,并且在性能上超越了其他现有技术。