arXiv:2410.06921stat.MLcs.LG2024-10

数据流形内不可分导致对抗脆弱性,二阶优化可显著提升鲁棒性。

Adversarial Vulnerability as a Consequence of On-Manifold Inseparibility

  • 将数据视为低维流形,用高/低方差特征区分流形内外方向。
  • 一阶优化器在流形外方向收敛差,导致对抗样本易生成。
  • 采用二阶优化和长训练可大幅提升模型鲁棒性,批量归一化反而有害。

近期研究已从理论和实证上表明,冗余数据维度是对抗脆弱性的根源。然而实践中,降维技术并未带来预期的鲁棒性提升。本文聚焦分类任务,将数据分布建模为低维流形,以高/低方差特征定义流形内/外方向。我们指出,干净训练在流形外方向因梯度下降等常用一阶优化器的病态条件而收敛不佳,当数据在流形内方向不可分时,此问题会引发对抗脆弱性。本文给出了逻辑回归和两层线性网络的理论分析。进一步建议使用对病态不敏感的二阶优化方法,能有效提升鲁棒性。实验显示,通过长训练与二阶优化可实现显著的鲁棒性提升,验证了该框架。此外,我们发现批量归一化层会抑制此类鲁棒性增益,归因于传统网络与批归一化网络隐式偏置的不同。

原文摘要 · Abstract (English)

Recent works have shown theoretically and empirically that redundant data dimensions are a source of adversarial vulnerability. However, the inverse doesn't seem to hold in practice; employing dimension-reduction techniques doesn't exhibit robustness as expected. In this work, we consider classification tasks and characterize the data distribution as a low-dimensional manifold, with high/low variance features defining the on/off manifold direction. We argue that clean training experiences poor convergence in the off-manifold direction caused by the ill-conditioning in widely used first-order optimizers like gradient descent. The poor convergence then acts as a source of adversarial vulnerability when the dataset is inseparable in the on-manifold direction. We provide theoretical results for logistic regression and a 2-layer linear network on the considered data distribution. Furthermore, we advocate using second-order methods that are immune to ill-conditioning and lead to better robustness. We perform experiments and exhibit tremendous robustness improvements in clean training through long training and the employment of second-order methods, corroborating our framework. Additionally, we find the inclusion of batch-norm layers hinders such robustness gains. We attribute this to differing implicit biases between traditional and batch-normalized neural networks.

对抗鲁棒性二阶优化流形学习批量归一化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。