Dissect and Prune: Enhancing Robustness in AI-Generated Image Detection
Dahye Kim, Jaehyun Choi, Hyun Seok Seong, Seongho Kim, Donghun Lee, Sungwon Yi, Jang-Ho Choi
Abstract
While existing AI-generated image detectors report high performance, we identify that this is largely driven by a critical prediction asymmetry : a bias toward the real class that severely limits sensitivity to generated content, especially under standard post-processing operations such as compression and resizing. We hypothesize that this stems from the model's reliance on spurious features, distracting signals that obscure true generative artifacts. To address this, we propose DEAR (Dissect and Prune), which leverages inpainted images to identify and prune these interfering components. Specifically, we find that features strongly aligned to either inpainted or non-inpainted regions are less robust to post-processing. By measuring the alignment between channel activations and inpaint masks, DEAR removes features at both extremes, retaining only those that capture genuine generative artifacts. Experimental results demonstrate that our approach significantly enhances robustness against unseen generators and post-processing, effectively mitigating the prediction asymmetry. Our code is available at https://github.com/dahyedahye/dear.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on40
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- SDXL: Improving Latent Diffusion Models for High-Resolution Image SynthesisDustin Podell, Zion English, Kyle Lacey, Andreas Blattmann et al.ICLR 2024 · 4,569 citations
- PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image SynthesisJunsong Chen, Jincheng Yu, Chongjian Ge, Lewei Yao et al.ICLR 2024 · 831 citations
- DIRE for Diffusion-Generated Image DetectionZhendong Wang, Jianmin Bao, Wengang Zhou, Weilun Wang et al.ICCV 2023 · 479 citations
Related papers
- Stay-Positive: A Case for Ignoring Real Image Features in Fake Image DetectionAnirudh Sundara Rajan, Yong Jae LeeICML 2025
- Beyond Semantic Features: Pixel-level Mapping for Generalized AI-Generated Image DetectionChenming Zhou, Jiaan Wang, Yu Li, Lei Li et al.AAAI 2026 · 1 citation
- Beyond Pixels: Mining Compressed Domain Artifacts for Efficient AI-Generated Video DetectionAnran Zhu, Zhengli Shi, Chende Zheng, Chenhao Lin et al.ICML 2026
- Breaking Semantic Artifacts for Generalized AI-generated Image DetectionChende Zheng, Chenhao Lin, Zhengyu Zhao, Hang Wang et al.NeurIPS 2024 · 57 citations
- Beyond Generation: A Diffusion-based Low-level Feature Extractor for Detecting AI-generated ImagesNan Zhong, Haoyu Chen, Yiran Xu, Zhenxing Qian et al.CVPR 2025
