arXiv · 2403.11172
Artifact Feature Purification for Cross-domain Detection of AI-generated Images
Abstract
In the era of AIGC, the fast development of visual content generation technologies, such as diffusion models, bring potential security risks to our society. Existing generated image detection methods suffer from performance drop when faced with out-of-domain generators and image scenes. To relieve this problem, we propose Artifact Purification Network (APN) to facilitate the artifact extraction from generated images through the explicit and implicit purification processes. For the explicit one, a suspicious frequency-band proposal method and a spatial feature decomposition method are proposed to extract artifact-related features. For the implicit one, a training strategy based on mutual information estimation is proposed to further purify the artifact-related features. Experiments show that for cross-generator detection, the average accuracy of APN is 5.6% ~ 16.4% higher than the previous 10 methods on GenImage dataset and 1.7% ~ 50.1% on DiffusionForensics dataset. For cross-scene detection, APN maintains its high performance. Via visualization analysis, we find that the proposed method extracts flexible forgery patterns and condenses the forgery information diluted in irrelevant features. We also find that the artifact features APN focuses on across generators and scenes are global and diverse. The code will be available on GitHub.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Zheling Meng, Bo Peng, Jing Dong, Tieniu Tan. 2024-03-17. Artifact Feature Purification for Cross-domain Detection of AI-generated Images. https://arxiv.org/abs/2403.11172
Cite the original work for its findings. Save a collection to share your selection of sources.