cs.CVApr 4, 2026

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

Authors: Fei Wu, Dagong Lu, Mufeng Yao, Xinlei Xu, Fengjun Guo

Organizations: Shanghai Jiao Tong University · INTSIG Information

Abstract

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, RouteA progressively constructs DINOv3-based detectors through staged data expansion and augmentation escalation, RouteB 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.

Figures & tables

Explore similar work

CardsList
  1. Where Detectors Fail: Probing Generative Space for Generalizable AI-Generated Image Detection

    May 24, 2026Zijie Cao, Weijie Tu, Yao Xiao +3Synthetic Data AugmentationAI-Generated Image Detection

  2. Boosting Robust AIGI Detection with LoRA-based Pairwise Training

    Apr 14, 2026Ruiyang Xia, Qi Zhang, Yaowen Xu +4AI-Generated Image Detection

  3. FeatDistill: A Feature Distillation Enhanced Multi-Expert Ensemble Framework for Robust AI-generated Image Detection

    Mar 23, 2026Zhilin Tu, Kemou Li, Fengpeng Li +3AI-Generated Image Detection