通过翻转图像比特位,提升AI生成图像检测的鲁棒性与速度。
RAID: Towards Robust AI-Generated Image Detection with Bit-Reversed Images

- 利用比特平面构造反向图像,捕捉真实与生成图像的本质差异。
- 在40多个基准上超越现有方法,零样本性能优异,速度提升近百倍。
- 适合需要高鲁棒性、快速部署的AI图像真实性验证场景。
生成模型的快速发展使人们难以区分真实图像与AI生成图像。为防范虚假图像滥用带来的风险,AI生成图像检测受到广泛关注。现有方法忽视真实与伪造图像间的内在差异,导致鲁棒性和泛化能力不足。本文创新性地引入比特平面分析,提出比特反向图像(bit-reversed image)概念,并设计包含图像构建、基于梯度的块选择和卷积分类器的简单有效流程。从数学角度提供理论分析支持。同时构建两个具有挑战性的新数据集。大量实验验证了方法在跨生成器、跨数据集及零样本设置下的有效性。无需复杂组件,本方法在超过40个基准上表现更优,且速度接近同类方法的100倍。代码已开源。
原文摘要 · Abstract (English)
The rapid advancement of image generation models has made it increasingly difficult for people to distinguish AI-generated images from real ones. To prevent the potential risks associated with the misuse of fake images, AI-generated image detection has gained significant attention. Existing methods neglect the inherent differences between real and fake images, thus lacking robustness and generalization ability. In this work, we innovatively investigate AI-generated image detection using bit-planes, and introduce the bit-reversed image. We propose a simple yet effective pipeline consisting of construction of bit-reversed images, gradient-based patch selection and a convolutional classifier. Besides, we provide a theoretical analysis from the mathematical perspective to demonstrate the validity of our approach. We also introduce two challenging datasets for AI-generated image detection. Extensive experiments verify the effectiveness of our approach across different settings, including cross-generator generalization, cross-dataset generalization and zero-shot performance. Without bells and whistles, our approach outperforms existing methods on over 40 benchmarks, and is nearly 100 times faster than counterparts. The code is at https://github.com/renxi-seu/RAID.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。