arXiv:2508.17174cs.LG2025-08

提出SaGD框架,让模型在对抗攻击下仍能准确区分异常数据和恶意样本。

Sharpness-Aware Geometric Defense for Robust Out-Of-Distribution Detection

  • 通过平滑对抗训练带来的尖锐损失曲面,优化嵌入空间几何结构。
  • 在多种攻击下对六组外部数据集的检测FPR显著降低,AUC提升明显。
  • 适合关注模型安全性与鲁棒性检测的研究者或工程落地应用。

分布外(OOD)检测保障模型安全部署。现有基于几何投影的OOD方法将对抗性内分布(ID)样本误判为OOD,导致错误检测。针对这一问题,本文提出一种鲁棒的OOD检测方法——尖锐感知几何防御(SaGD),可有效区分对抗性ID样本与真实OOD样本。对抗训练产生的尖锐损失曲面阻碍模型收敛,影响用于计算OOD分数的潜在嵌入质量。为此,我们引入SaGD框架,平滑投影后潜在空间中的陡峭损失地形。增强的几何嵌入收敛能力使内分布数据表征更准确,从而提升对抗攻击下的检测性能。采用基于抖动的扰动进行对抗训练,扩展了对未见攻击的防御能力。实验表明,在多种攻击下,该方法在将CIFAR-100与六个其他OOD数据集区分时,显著优于现有先进防御方法,在不同对抗训练强度下,还揭示了尖锐损失曲面与对抗性OOD检测之间的关系。

原文摘要 · Abstract (English)

Out-of-distribution (OOD) detection ensures safe and reliable model deployment. Contemporary OOD algorithms using geometry projection can detect OOD or adversarial samples from clean in-distribution (ID) samples. However, this setting regards adversarial ID samples as OOD, leading to incorrect OOD predictions. Existing efforts on OOD detection with ID and OOD data under attacks are minimal. In this paper, we develop a robust OOD detection method that distinguishes adversarial ID samples from OOD ones. The sharp loss landscape created by adversarial training hinders model convergence, impacting the latent embedding quality for OOD score calculation. Therefore, we introduce a {\bf Sharpness-aware Geometric Defense (SaGD)} framework to smooth out the rugged adversarial loss landscape in the projected latent geometry. Enhanced geometric embedding convergence enables accurate ID data characterization, benefiting OOD detection against adversarial attacks. We use Jitter-based perturbation in adversarial training to extend the defense ability against unseen attacks. Our SaGD framework significantly improves FPR and AUC over the state-of-the-art defense approaches in differentiating CIFAR-100 from six other OOD datasets under various attacks. We further examine the effects of perturbations at various adversarial training levels, revealing the relationship between the sharp loss landscape and adversarial OOD detection.

OOD检测对抗防御几何方法

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