用比特平面分析图像噪声,快速精准识别AI生成图片
LOTA: Bit-Planes Guided AI-Generated Image Detection
- 通过低比特平面提取图像内在噪声模式,替代传统高成本重建误差
- 检测准确率达98.9%,毫秒级速度比现有方法快近百倍
- 对不同生成模型间转换具有强泛化能力,适合实际应用部署
生成式对抗网络(GAN)和扩散模型的快速发展使人工智能生成图像与真实图像的区分愈发困难。现有方法常依赖图像重建误差作为特征,但计算开销大且难以捕捉原始图像中的固有噪声。为此,本文创新性地采用比特平面图像处理技术,因低比特平面恰好反映图像噪声结构。我们提出一种基于比特平面的噪声图像生成方法,并结合缩放、阈值等归一化策略;为增强噪声信号以利于检测,设计最大梯度块选择机制,通过多方向梯度计算噪声得分并选取最高分区域。最后,构建轻量级分类头,探索基于噪声的分类器与噪声引导分类器两种结构。在GenImage基准测试中,本方法平均准确率达到98.9%(较前人提升11.9%),展现出优异的跨生成器泛化能力:从GAN到扩散模型准确率超98.2%,反向超99.2%。同时,误差提取仅需毫秒级别,接近百倍于现有方法的速度优势。代码已开源。
原文摘要 · Abstract (English)
The rapid advancement of GAN and Diffusion models makes it more difficult to distinguish AI-generated images from real ones. Recent studies often use image-based reconstruction errors as an important feature for determining whether an image is AI-generated. However, these approaches typically incur high computational costs and also fail to capture intrinsic noisy features present in the raw images. To solve these problems, we innovatively refine error extraction by using bit-plane-based image processing, as lower bit planes indeed represent noise patterns in images. We introduce an effective bit-planes guided noisy image generation and exploit various image normalization strategies, including scaling and thresholding. Then, to amplify the noise signal for easier AI-generated image detection, we design a maximum gradient patch selection that applies multi-directional gradients to compute the noise score and selects the region with the highest score. Finally, we propose a lightweight and effective classification head and explore two different structures: noise-based classifier and noise-guided classifier. Extensive experiments on the GenImage benchmark demonstrate the outstanding performance of our method, which achieves an average accuracy of \textbf{98.9\%} (\textbf{11.9}\%~$\uparrow$) and shows excellent cross-generator generalization capability. Particularly, our method achieves an accuracy of over 98.2\% from GAN to Diffusion and over 99.2\% from Diffusion to GAN. Moreover, it performs error extraction at the millisecond level, nearly a hundred times faster than existing methods. The code is at https://github.com/hongsong-wang/LOTA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。