Leveraging Representations from Intermediate Encoder-blocks for Synthetic Image Detection
About
The recently developed and publicly available synthetic image generation methods and services make it possible to create extremely realistic imagery on demand, raising great risks for the integrity and safety of online information. State-of-the-art Synthetic Image Detection (SID) research has led to strong evidence on the advantages of feature extraction from foundation models. However, such extracted features mostly encapsulate high-level visual semantics instead of fine-grained details, which are more important for the SID task. On the contrary, shallow layers encode low-level visual information. In this work, we leverage the image representations extracted by intermediate Transformer blocks of CLIP's image-encoder via a lightweight network that maps them to a learnable forgery-aware vector space capable of generalizing exceptionally well. We also employ a trainable module to incorporate the importance of each Transformer block to the final prediction. Our method is compared against the state-of-the-art by evaluating it on 20 test datasets and exhibits an average +10.6% absolute performance improvement. Notably, the best performing models require just a single epoch for training (~8 minutes). Code available at https://github.com/mever-team/rine.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Generated Image Detection | GenImage (test) | Average Accuracy81.78 | 103 | |
| AI-generated image detection | Chameleon | Accuracy58.54 | 63 | |
| Deepfake Detection | UniversalFakeDetect 1.0 (test) | Accuracy (ProGAN)100 | 42 | |
| Image-level detection | OpenSDID cross-domain SD2.1 | F1 Score87.47 | 14 | |
| Image-level detection | OpenSDID SDXL (cross-domain) | F1 Score73.43 | 14 | |
| Image-level detection | OpenSDID Average (aggregated) | F1 Score75.98 | 14 | |
| Image-level detection | OpenSDID SD3 cross-domain | F1 Score72.05 | 14 | |
| Image-level detection | OpenSDID Flux.1 (cross-domain) | F1 Score55.86 | 14 | |
| Image-level detection | OpenSDID SD1.5 (in-domain) | F1 Score91.08 | 14 | |
| Forgery Detection | UniversalFakeDetect (test) | ProGAN100 | 12 |