Oct, 2023
黑铁匠:通过混合单步和多步方法实现视觉 Transformer 的快速对抗训练
Blacksmith: Fast Adversarial Training of Vision Transformers via a Mixture of Single-step and Multi-step Methods
Mahdi Salmani, Alireza Dehghanpour Farashah, Mohammad Azizmalayeri, Mahdi Amiri, Navid Eslami...
TL;DR我们提出了一种名为 Blacksmith 的新训练策略,用于解决深度学习算法中的 Catastrophic Overfitting 问题,尤其适用于 Vision Transformers。通过在神经网络的对抗训练中随机使用 PGD-2 或 Fast Gradient Sign Method (FGSM) 攻击,我们的方法有效地防止 Catastrophic Overfitting,并且在性能上超越了其他现有技术。