AlignGemini: Generalizable AI-Generated Image Detection Through Task-Model Alignment
About
Vision Language Models (VLMs) are increasingly used for detecting AI-generated images (AIGI). However, converting VLMs into reliable detectors is resource-intensive, and the resulting models often suffer from hallucination and poor generalization. To investigate the root cause, we conduct an empirical analysis and identify two consistent behaviors. First, fine-tuning VLMs with semantic supervision improves semantic discrimination and generalizes well to unseen data. Second, fine-tuning VLMs with pixel-artifact supervision leads to weak generalization. These findings reveal a fundamental task-model misalignment. VLMs are optimized for high-level semantic reasoning and lack inductive bias toward low-level pixel artifacts. In contrast, conventional vision models effectively capture pixel-level artifacts but are less sensitive to semantic inconsistencies. This indicates that different models are naturally suited to different subtasks. Based on this insight, we formulate AIGI detection as two orthogonal subtasks: semantic consistency checking and pixel-artifact detection. Neglecting either subtask leads to systematic detection failures. We further propose the Task-Model Alignment principle and instantiate it in a two-branch detector, AlignGemini. The detector combines a VLM trained with pure semantic supervision and a vision model trained with pure pixel-artifact supervision. By enforcing clear specialization, each branch captures complementary cues. Experiments on in-the-wild benchmarks show that AlignGemini improves average accuracy by 9.5 percent using simplified training data. These results demonstrate that task-model alignment is an effective principle for generalizable AIGI detection.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| AI-generated image detection | GenImage | -- | 65 | |
| AIGI Detection | BFree Online | B.Acc97.5 | 23 | |
| AIGI Detection | DRCT-2M | B.Acc98.1 | 23 | |
| AI-generated image detection | WildRF | CommunityUI Score92.9 | 12 | |
| AI-generated image detection | AIGI-Bench | Detection Rate (Civitai)88.8 | 12 | |
| AI-generated image detection | Chameleon | Accuracy (Reddit)92.2 | 12 | |
| AI-generated image detection | CO-SPY-Bench in-the-wild | Balanced Accuracy94.3 | 11 | |
| AI-generated image detection | Overall In-the-wild Aggregate | Average Accuracy91.8 | 11 | |
| AIGI Detection | AIGI-Now Nano Banana Sem | Balanced Accuracy95.6 | 11 | |
| AIGI Detection | AIGI-Now GPT-4o Sem | Balanced Acc95.7 | 11 |