arXiv:2411.19417cs.CVcs.LG2024-11CVPR被引 68

通过频谱学习实现任意分辨率的AI生成图像检测

Any-Resolution AI-Generated Image Detection by Spectral Learning

论文配图:Any-Resolution AI-Generated Image Detection by Spectral Learning
图 1 · 摘自论文原文
  • 利用频谱重建自监督学习捕捉真实图像的频谱规律
  • 在13种生成模型上提升5.5% AUC,对常见干扰保持鲁棒
  • 适配任意分辨率,可识别细微频谱异常

近期研究发现生成图像中存在频谱伪影,并提出依赖标注数据的方法来学习这些特征。然而,不同生成模型间伪影差异大,导致现有方法难以泛化到未见生成器。本文基于真实图像频谱分布具有不变性且高度可区分的特性,采用掩码频谱学习(自监督)进行建模,以频率重建为预训练任务。由于生成图像属于该模型的分布外样本,我们引入频谱重建相似性来捕捉其偏差。同时提出频谱上下文注意力机制,有效捕获任意分辨率图像中的细微频谱不一致。所提方法SPAI在13种近期生成模型上相较之前最优方法提升5.5% AUC,且对常见在线扰动保持鲁棒。代码已公开于https://mever-team.github.io/spai。

原文摘要 · Abstract (English)

Recent works have established that AI models introduce spectral artifacts into generated images and propose approaches for learning to capture them using labeled data. However, the significant differences in such artifacts among different generative models hinder these approaches from generalizing to generators not seen during training. In this work, we build upon the key idea that the spectral distribution of real images constitutes both an invariant and highly discriminative pattern for AI-generated image detection. To model this under a self-supervised setup, we employ masked spectral learning using the pretext task of frequency reconstruction. Since generated images constitute out-of-distribution samples for this model, we propose spectral reconstruction similarity to capture this divergence. Moreover, we introduce spectral context attention, which enables our approach to efficiently capture subtle spectral inconsistencies in images of any resolution. Our spectral AI-generated image detection approach (SPAI) achieves a 5.5% absolute improvement in AUC over the previous state-of-the-art across 13 recent generative approaches, while exhibiting robustness against common online perturbations. Code is available on https://mever-team.github.io/spai.

图像检测频谱分析自监督学习

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