Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Exploiting the Final Component of Generator Architectures for AI-Generated Image Detection

About

With the rapid proliferation of powerful image generators, accurate detection of AI-generated images has become essential for maintaining a trustworthy online environment. However, existing deepfake detectors often generalize poorly to images produced by unseen generators. Notably, despite being trained under vastly different paradigms, such as diffusion or autoregressive modeling, many modern image generators share common final architectural components that serve as the last stage for converting intermediate representations into images. Motivated by this insight, we propose to "contaminate" real images using the generator's final component and train a detector to distinguish them from the original real images. We further introduce a taxonomy based on generators' final components and categorize 21 widely used generators accordingly, enabling a comprehensive investigation of our method's generalization capability. Using only 100 samples from each of three representative categories, our detector-fine-tuned on the DINOv3 backbone-achieves an average accuracy of 98.83% across 22 testing sets from unseen generators.

Yanzhu Liu, Xiao Liu, Yuexuan Wang, Mondal Soumik• 2026

Related benchmarks

TaskDatasetResultRank
AI-generated image detectionSynthbuster Firefly (test)
Accuracy99.7
12
AI-generated image detectionSynthbuster Midjourney (test)
Accuracy99.7
12
AI-generated image detectionFakeBench (test)
Accuracy98.83
12
AI-generated image detectionWildRF Reddit (test)
Accuracy98.67
12
AI-generated image detectionWildRF (Facebook) (test)
Accuracy98.44
12
AI-generated image detectionWildRF Twitter (test)
Accuracy99.31
12
Fake Image DetectionSynthbuster SD 1.3 (test)
Accuracy99.9
12
Fake Image DetectionSynthbuster SD 1.4 (test)
Accuracy99.85
12
Fake Image DetectionSynthbuster SD XL
Accuracy0.999
12
Fake Image DetectionFlux (dev)
Accuracy99.89
12
Showing 10 of 20 rows

Other info

Follow for update