mlsanityData to 14 Oct 2026

PatchHead: Learning Spatial Patch Evidence for Generalizable AI-Generated Image Detection

arXiv 2608.09223 ↗First posted 10 Aug 2026cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

AI-generated image detectors generalize poorly when their training and test images originate from different generators or datasets. Despite the rich spatial representations produced by vision foundation models like DINO, existing detectors typically classify images using only the globally aggregated CLS token. We hypothesize that globally aggregating DINO features into a single CLS token obscures spatially distributed generation traces. To test this hypothesis, we introduce PatchHead, a lightweight spatial aggregation head that preserves the two-dimensional organization of DINO patch tokens and integrates evidence across neighboring regions. During training, we freeze the pretrained DINO backbone and optimize only the inserted LoRA adapters, PatchHead, and auxiliary projection head. Across nine cross-dataset benchmarks spanning manually curated and in-the-wild settings, PatchHead ranks first on seven datasets and second on the remaining two. It improves the strongest prior method from 91.6% to 94.6% in average balanced accuracy (+3.0 points) and raises the worst-case accuracy from 82.4% to 89.4% (+6.9 points), while introducing only 8.6% more trainable parameters and 0.08% additional FLOPs. Further qualitative analysis suggests that PatchHead (i) reduces class-conditional domain discrepancy, and (ii) redirects the representation from content-dominated saliency toward spatially distributed authenticity evidence. Together, these observations provide a representation-level account of why spatial patch aggregation transfers more reliably across generators and datasets than a single CLS-based global representation. Our code and models will be made available upon acceptance.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 14 of its 20 references that are in the map. First posted on arXiv 10 Aug 2026.

Builds on

14 of 20 references.

References · 14

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  2. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  3. 2017MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applicationscited by 3,845
  4. 2025DINOv3cited by 1,414
  5. 2019CNN-generated images are surprisingly easy to spot... for nowcited by 847
  6. 2023Towards Universal Fake Image Detectors that Generalize Across Generative Modelscited by 490
  7. 2023Forgery-aware Adaptive Transformer for Generalizable Synthetic Image Detectioncited by 182
  8. 2023Raising the Bar of AI-generated Image Detection with CLIPcited by 164
  9. 2024C2P-CLIP: Injecting Category Common Prompt in CLIP to Enhance Generalization in Deepfake Detectioncited by 113
  10. 2024Improving Synthetic Image Detection Towards Generalization: An Image Transformation Perspectivecited by 89
  11. 2024Real-Time Deepfake Detection in the Real-Worldcited by 51
  12. 2023Rethinking the Up-Sampling Operations in CNN-based Generative Network for Generalizable Deepfake Detectioncited by 41
2 more
  1. 2024Aligned Datasets Improve Detection of Latent Diffusion-Generated Imagescited by 36
  2. 2026How well are open sourced AI-generated image detection models out-of-the-box: A comprehensive benchmark studycited by 15

Built on by

Not cited yet.