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

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.

Ruoxin Chen, Jiahui Gao, Kaiqing Lin, Keyue Zhang, Yandan Zhao, Isabel Guan, Taiping Yao, Shouhong Ding• 2025

Related benchmarks

TaskDatasetResultRank
AI-generated image detectionGenImage--
65
AIGI DetectionBFree Online
B.Acc97.5
23
AIGI DetectionDRCT-2M
B.Acc98.1
23
AI-generated image detectionWildRF
CommunityUI Score92.9
12
AI-generated image detectionAIGI-Bench
Detection Rate (Civitai)88.8
12
AI-generated image detectionChameleon
Accuracy (Reddit)92.2
12
AI-generated image detectionCO-SPY-Bench in-the-wild
Balanced Accuracy94.3
11
AI-generated image detectionOverall In-the-wild Aggregate
Average Accuracy91.8
11
AIGI DetectionAIGI-Now Nano Banana Sem
Balanced Accuracy95.6
11
AIGI DetectionAIGI-Now GPT-4o Sem
Balanced Acc95.7
11
Showing 10 of 23 rows

Other info

Follow for update