中文

基于交叉注意力增强双流网络的AI生成图像检测

计算机视觉与模式识别 2023-11-10 v2 密码学与安全

摘要

随着 AI 生成内容(AIGC)的快速发展,相较于传统的计算机生成图形(CG),该技术生成的伪造图像本质上更具欺骗性且需要更少的人为干预。然而,由于 CG 与 AIGC 之间的差异,传统的 CG 检测方法往往不足以识别 AIGC 生成的图像。为解决此问题,我们的研究聚焦于 AIGC 中的文本到图像生成过程。首先,我们利用 DALLE2 和 DreamStudio 两个不同 AI 系统构建了两个文本到图像数据库。为整体捕捉 AIGC 产生的固有异常,我们开发了一个由残差流和内容流组成的鲁棒双流网络。前者采用空间富模型(SRM)从图像中细致提取各类纹理信息,后者旨在捕捉低频中的额外伪造痕迹,从而提取残差流可能忽略的互补信息。为增强两流之间的信息交换,我们引入了交叉多头注意力机制。我们在两个数据库上进行了大量对比实验,结果表明我们的检测方法在各种图像分辨率下始终优于传统 CG 检测技术。此外,通过一系列鲁棒性测试和跨数据库实验,我们的方法展现出优越性能。当应用于 SPL2018 和 DsTok 等广受认可的传统 CG 基准时,我们的方法在 CG 检测领域显著超越其他现有方法的能力。

关键词

引用

@article{arxiv.2306.07005,
  title  = {AI-Generated Image Detection using a Cross-Attention Enhanced Dual-Stream Network},
  author = {Ziyi Xi and Wenmin Huang and Kangkang Wei and Weiqi Luo and Peijia Zheng},
  journal= {arXiv preprint arXiv:2306.07005},
  year   = {2023}
}

备注

8 pages, 41 figures