Oct, 2023

快速传播是更好的:通过抽取子网络加速单步对抗训练

TL;DR在这项工作中,我们提出了一种利用模型的内部构建块来提高效率的方法,通过在训练过程中动态地采样轻量级子网络作为替代模型,加速前向和后向传递以实现高效的对抗训练,并通过理论分析证明了单步对抗训练和采样子网络可以提高模型的鲁棒性。相比之前的方法,我们的方法不仅减少了训练成本,而且实现了更好的模型鲁棒性。