用融合特征与轻量微调提升AI图像检测的泛化与鲁棒性
DeeCLIP: A Robust and Generalizable Transformer-Based Framework for Detecting AI-Generated Images
- 融合高低层特征,增强对压缩模糊等退化的抗干扰能力
- 仅用4类ProGAN数据训练,19个测试子集平均准确率达89.00%
- 采用LoRA实现轻量微调,适合快速适配新生成模型
本文提出DeeCLIP,一种基于CLIP-ViT和融合学习的AI生成图像检测框架。针对现有方法在不同生成模型间泛化能力差、对轻微扰动敏感的问题,DeeCLIP引入DeeFuser融合模块,结合高层与低层特征,提升对压缩、模糊等退化处理的鲁棒性。通过三元组损失优化嵌入空间,增强真实与合成内容的区分能力。为实现轻量级适应并保留预训练知识,采用低秩适应(LoRA)进行参数高效微调,支持零样本学习而不牺牲泛化性。模型仅在4类ProGAN数据上训练,却在由GAN与扩散模型组成的19个测试子集上达到平均89.00%的准确率。尽管可训练参数更少,仍优于现有方法,展现出对多种生成模型及现实扭曲的强鲁棒性。代码已开源。
原文摘要 · Abstract (English)
This paper introduces DeeCLIP, a novel framework for detecting AI-generated images using CLIP-ViT and fusion learning. Despite significant advancements in generative models capable of creating highly photorealistic images, existing detection methods often struggle to generalize across different models and are highly sensitive to minor perturbations. To address these challenges, DeeCLIP incorporates DeeFuser, a fusion module that combines high-level and low-level features, improving robustness against degradations such as compression and blurring. Additionally, we apply triplet loss to refine the embedding space, enhancing the model's ability to distinguish between real and synthetic content. To further enable lightweight adaptation while preserving pre-trained knowledge, we adopt parameter-efficient fine-tuning using low-rank adaptation (LoRA) within the CLIP-ViT backbone. This approach supports effective zero-shot learning without sacrificing generalization. Trained exclusively on 4-class ProGAN data, DeeCLIP achieves an average accuracy of 89.00% on 19 test subsets composed of generative adversarial network (GAN) and diffusion models. Despite having fewer trainable parameters, DeeCLIP outperforms existing methods, demonstrating superior robustness against various generative models and real-world distortions. The code is publicly available at https://github.com/Mamadou-Keita/DeeCLIP for research purposes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。