通过主动探索生成空间边界,提升图像检测器对未知生成器的泛化能力。
Where Detectors Fail: Probing Generative Space for Generalizable AI-Generated Image Detection

- 用检测器作为评判标准,引导生成器产生难辨样本
- 在多个基准上实现对未见生成器的检测性能提升
- 适合关注AI生成内容安全与鲁棒检测的研究者
AI生成图像(AIGI)检测仍面临挑战,因现有检测器难以泛化至未见过的生成器。尽管训练数据规模大,但当生成设置变化时性能仍下降,表明仅靠数据量不足,训练中对生成变体覆盖有限是关键原因。研究发现,生成模型内部表示的微小改动即可产生多样且有意义的图像差异,而这些常未在标准采样中体现。基于此,我们提出PROBE(通过边界探索提升鲁棒性),框架通过主动探测生成过程中的困难区域来增强检测器泛化能力。不将生成器视为固定数据源,而是以检测器为评判标准,引导生成器进行流形级修改,生成难以分类的真实样本。这些样本揭示了标准采样下罕见的失败案例,并用于优化检测器。多基准实验表明,PROBE显著提升了对未见生成器的检测泛化性能。代码与模型已公开于https://github.com/Amamiya-C/PROBE-AIGI-Detection。
原文摘要 · Abstract (English)
Detecting AI-generated images (AIGI) remains challenging because detectors often fail to generalize to unseen generators. Although existing methods are trained on large datasets, their performance still degrades when generation settings change, indicating that data scale alone is insufficient and that limited coverage of generative variations during training is a key factor. Studies on generative model editing show that small changes in internal representations can produce diverse and meaningful image variations, many of which are not explored under standard sampling. Leveraging this insight, we propose PROBE (Probing Robustness via Boundary Exploration), a framework that improves detector generalization by actively exploring challenging regions of the generative process. Instead of treating the generator as a fixed data source, PROBE uses the detector as a critic to steer the generator through manifold-level modifications, producing realistic samples that are difficult to classify. These samples expose failure cases that are uncommon under standard data sampling strategies and are used to refine the detector. Experimental results across multiple benchmarks indicate that PROBE enhances generalization to unseen generators, resulting in more generalizable AIGI detection performance. Code and models are available at https://github.com/Amamiya-C/PROBE-AIGI-Detection
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。