arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2605.06357cs.LGcs.AIcs.CV

内存高效的全梯度攻击(MEFA)框架用于对抗性防御评估

Memory Efficient Full-gradient Attacks (MEFA) Framework for Adversarial Defense Evaluations

InnoPeak Technology(InnoPeak技术公司)

Yuan Du, Mitchel Hill, HanQin Cai

AI总结:

本文提出MEFA框架,通过梯度检查点技术实现长净化轨迹的精确端到端梯度计算,提升对抗性防御评估的准确性,揭示了近似梯度评估的局限性。

摘要:

本文研究了在白盒对抗攻击下迭代随机净化防御的鲁棒性评估。我们的关键技术见解是梯度检查点技术通过牺牲额外的重新计算来换取显著降低的内存使用,使通过长净化轨迹进行精确端到端梯度计算成为可能。这使能够对基于扩散和兰格-金斯堡的净化防御实施全梯度自适应攻击,而先前的评估往往由于内存限制而依赖于近似反向传播。这些近似方法可能会削弱攻击信号并高估鲁棒性。同时,迭代净化中的随机性常被低估,尽管不同的净化轨迹可以显著改变报告的鲁棒性指标。基于这一见解,我们引入了一个内存高效的全梯度评估框架用于随机净化防御。该框架结合了检查点反向传播与控制随机变化的评估协议,从而减少内存瓶颈同时保持精确梯度。我们评估了基于扩散的净化和能量基模型(EBMs)的兰格-金斯堡采样,证明全梯度攻击揭示了近似梯度评估所遗漏的漏洞。我们的框架产生了更强的最先进ℓ∞和ℓ2白盒攻击,并进一步支持探测分布外鲁棒性。总体而言,我们的结果表明精确梯度评估对于可靠评估迭代随机防御至关重要。

英文摘要:

This work studies the robust evaluation of iterative stochastic purification defenses under white-box adversarial attacks. Our key technical insight is that gradient checkpointing makes exact end-to-end gradient computation through long purification trajectories practical by trading additional recomputation for substantially lower memory usage. This enables full-gradient adaptive attacks against diffusion- and Langevin-based purification defenses, where prior evaluations often resort to approximate backpropagation due to memory constraints. These approximations can weaken the attack signal and risk overestimating robustness. In parallel, stochasticity in iterative purification is frequently under-controlled, even though different purification trajectories can substantially change reported robustness metrics. Building on this insight, we introduce a memory-efficient full-gradient evaluation framework for stochastic purification defenses. The framework combines checkpointed backpropagation with evaluation protocols that control stochastic variability, thereby reducing memory bottlenecks while preserving exact gradients. We evaluate diffusion-based purification and Langevin sampling with Energy-Based Models (EBMs), demonstrating that full-gradient attacks uncover vulnerabilities missed by approximate-gradient evaluations. Our framework yields stronger state-of-the-art $\ell_{\infty}$ and $\ell_{2}$ white-box attacks and further supports probing out-of-distribution robustness. Overall, our results show that exact-gradient evaluation is essential for reliable benchmarking of iterative stochastic defenses.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑