HEDGE: Heterogeneous Ensemble for Detection of AI-GEnerated Images in the Wild

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wu, Fei, Lu, Dagong, Yao, Mufeng, Xu, Xinlei, Guo, Fengjun
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918427861450752
author Wu, Fei
Lu, Dagong
Yao, Mufeng
Xu, Xinlei
Guo, Fengjun
author_facet Wu, Fei
Lu, Dagong
Yao, Mufeng
Xu, Xinlei
Guo, Fengjun
contents Robust detection of AI-generated images in the wild remains challenging due to the rapid evolution of generative models and varied real-world distortions. We argue that relying on a single training regime, resolution, or backbone is insufficient to handle all conditions, and that structured heterogeneity across these dimensions is essential for robust detection. To this end, we propose HEDGE, a Heterogeneous Ensemble for Detection of AI-GEnerated images, that introduces complementary detection routes along three axes: diverse training data with strong augmentation, multi-scale feature extraction, and backbone heterogeneity. Specifically, Route~A progressively constructs DINOv3-based detectors through staged data expansion and augmentation escalation, Route~B incorporates a higher-resolution branch for fine-grained forensic cues, and Route~C adds a MetaCLIP2-based branch for backbone diversity. All outputs are fused via logit-space weighted averaging, refined by a lightweight dual-gating mechanism that handles branch-level outliers and majority-dominated fusion errors. HEDGE achieves 4th place in the NTIRE 2026 Robust AI-Generated Image Detection in the Wild Challenge and attains state-of-the-art performance with strong robustness on multiple AIGC image detection benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2604_03555
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle HEDGE: Heterogeneous Ensemble for Detection of AI-GEnerated Images in the Wild
Wu, Fei
Lu, Dagong
Yao, Mufeng
Xu, Xinlei
Guo, Fengjun
Computer Vision and Pattern Recognition
Robust detection of AI-generated images in the wild remains challenging due to the rapid evolution of generative models and varied real-world distortions. We argue that relying on a single training regime, resolution, or backbone is insufficient to handle all conditions, and that structured heterogeneity across these dimensions is essential for robust detection. To this end, we propose HEDGE, a Heterogeneous Ensemble for Detection of AI-GEnerated images, that introduces complementary detection routes along three axes: diverse training data with strong augmentation, multi-scale feature extraction, and backbone heterogeneity. Specifically, Route~A progressively constructs DINOv3-based detectors through staged data expansion and augmentation escalation, Route~B incorporates a higher-resolution branch for fine-grained forensic cues, and Route~C adds a MetaCLIP2-based branch for backbone diversity. All outputs are fused via logit-space weighted averaging, refined by a lightweight dual-gating mechanism that handles branch-level outliers and majority-dominated fusion errors. HEDGE achieves 4th place in the NTIRE 2026 Robust AI-Generated Image Detection in the Wild Challenge and attains state-of-the-art performance with strong robustness on multiple AIGC image detection benchmarks.
title HEDGE: Heterogeneous Ensemble for Detection of AI-GEnerated Images in the Wild
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2604.03555