arXiv:2608.26963cs.LG2026-08

无需输入梯度的对抗训练,用低秩豪斯霍尔德展开快速生成小扰动对抗样本。

Adversarial Training Without Input Gradients via Low-Rank Householder Expansions

论文配图:Adversarial Training Without Input Gradients via Low-Rank Householder Expansions
图 1 · 摘自论文原文
  • 通过低秩豪斯霍尔德展开直接从网络参数计算对抗样本,跳过输入梯度迭代。
  • 每轮训练仅需2.8次PGD等效开销,比40步训练快8.7倍,性能接近3步PGD。
  • 适合追求高效对抗训练且关注小范数扰动的场景,如资源受限部署。

本文研究由深度神经网络固有输入不稳定性引发的小范数对抗样本。此类样本在相对ℓ²范数下极小,位于模型近似线性的邻域内,因此扰动不可察觉。我们首次证明,可通过一种称为低秩豪斯霍尔德展开(LRHE)的线性化方法,直接从训练后的网络参数获取对抗样本,无需输入梯度迭代。该展开描述的是组合仿射映射而非单个层,其方向可直接从前向传播中的激活模式读取。基于此,我们提出一种新对抗训练方案:全程不进行输入微分,仅需额外前向评估,权重更新仍用标准反向传播,且完全消除传统最小最大优化中的内层最大化。该正则化项的开销相当于每轮2.8次PGD步骤,较MNIST上40步对抗训练降低8.7倍,低于3步训练成本。所得模型在相对ℓ²预算ε≤0.02时表现媲美3步PGD训练,在ε≤0.012时匹配40步训练,超出后性能下降,符合展开的局部性特征。

原文摘要 · Abstract (English)

This work concerns adversarial training against the small-norm adversarial examples that arise from the inherent input instability of a trained deep neural network. Examples in this class are small as measured in the relative $\ell^2$-norm, and therefore lie in the neighborhood of the input on which the model acts approximately linearly, the regime in which the perturbation remains imperceptible. We first show that such examples can be computed directly from the trained network parameters, without input gradient iterations, by means of a linearization called the low-rank Householder expansion (LRHE). The expansion describes the composed affine map rather than any individual layer, and the directions it identifies are read from the activation pattern already available in the forward pass. We then propose a simple adversarial training scheme built on this construction. No differentiation with respect to the input is performed at any point: training requires only additional forward evaluations, with weight parameters updated by the standard backward pass, and the inner maximization of the usual min-max formulation is eliminated entirely. That such a regularizer exists is our main finding: the methods that dispense with the inner search all obtain their local geometry by differentiating with respect to the input, and we show this is not necessary. The regularizer costs the equivalent of $2.8$ PGD steps per epoch, an $8.7\times$ reduction relative to 40-step adversarial training on MNIST and below the cost of 3-step training. The resulting models match three-step PGD adversarial training for relative $\ell^2$ budgets $\varepsilon \le 0.02$ and 40-step training for $\varepsilon \le 0.012$, falling away beyond, consistent with the locality of the expansion.

对抗训练梯度替代高效学习线性化

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