arXiv:2506.00874cs.CV2025-06NeurIPS被引 27

通过对抗训练打破生成模型隐空间偏见,提升AI图像检测泛化能力。

Breaking Latent Prior Bias in Detectors for Generalizable AIGC Image Detection

论文配图:Breaking Latent Prior Bias in Detectors for Generalizable AIGC Image Detection
图 1 · 摘自论文原文
  • 在隐空间生成对抗样本,保持生成器输出流形一致性。
  • 在Flux.1、SD3等新生成器上检测准确率提升超20%。
  • 适合需要跨模型检测的AI内容安全研究者使用。

当前AIGC检测器在训练所用生成器上表现接近完美,但在未见生成器上泛化能力差。我们发现其根源之一是隐空间先验偏见:检测器学习了源自初始噪声向量的捷径特征,而非鲁棒的生成痕迹。为此提出基于流形的对抗训练(OMAT):在固定条件约束下优化扩散模型的初始隐变量,生成位于生成流形上的对抗样本。与像素空间攻击不同,该方法不引入生成器无法复现的离流形扰动,能有效暴露真实判别特征。为测试前沿生成模型,我们构建了仅用于测试的基准数据集GenImage++,包含Flux.1和SD3生成的复杂提示与多样风格图像。将该对抗训练范式应用于ResNet50和CLIP基线,在现有及最新挑战数据集上评估。大量实验表明,经对抗训练的检测器显著提升跨生成器性能,无需网络结构修改。对隐空间先验偏见的发现,为未来数据集构建与检测器评估提供重要启示。

原文摘要 · Abstract (English)

Current AIGC detectors often achieve near-perfect accuracy on images produced by the same generator used for training but struggle to generalize to outputs from unseen generators. We trace this failure in part to latent prior bias: detectors learn shortcuts tied to patterns stemming from the initial noise vector rather than learning robust generative artifacts. To address this, we propose On-Manifold Adversarial Training (OMAT): by optimizing the initial latent noise of diffusion models under fixed conditioning, we generate on-manifold adversarial examples that remain on the generator's output manifold-unlike pixel-space attacks, which introduce off-manifold perturbations that the generator itself cannot reproduce and that can obscure the true discriminative artifacts. To test against state-of-the-art generative models, we introduce GenImage++, a test-only benchmark of outputs from advanced generators (Flux.1, SD3) with extended prompts and diverse styles. We apply our adversarial-training paradigm to ResNet50 and CLIP baselines and evaluate across existing AIGC forensic benchmarks and recent challenge datasets. Extensive experiments show that adversarially trained detectors significantly improve cross-generator performance without any network redesign. Our findings on latent-prior bias offer valuable insights for future dataset construction and detector evaluation, guiding the development of more robust and generalizable AIGC forensic methodologies.

AI检测扩散模型对抗训练泛化能力

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