cs.CVMar 29, 2026

Diversity Matters: Dataset Diversification and Dual-Branch Network for Generalized AI-Generated Image Detection

Authors: Nusrat Tasnim, Kutub Uddin, Khalid Malik

Organizations: Korea Aerospace University, 10540, Goyang, Gyeonggi, South Korea · University of Michigan-Flint, 48502, Flint, Michigan, United States

Abstract

The rapid proliferation of AI-generated images, powered by generative adversarial networks (GANs), diffusion models, and other synthesis techniques, has raised serious concerns about misinformation, copyright violations, and digital security. However, detecting such images in a generalized and robust manner remains a major challenge due to the vast diversity of generative models and data distributions. In this work, we present Diversity Matters, a novel framework that emphasizes data diversity and feature domain complementarity for AI-generated image detection. The proposed method introduces a feature-domain similarity filtering mechanism that discards redundant or highly similar samples across both inter-class and intra-class distributions, ensuring a more diverse and representative training set. Furthermore, we propose a dual-branch network that combines CLIP features from the pixel domain and the frequency domain to jointly capture semantic and structural cues, leading to improved generalization against unseen generative models and adversarial conditions. Extensive experiments on benchmark datasets demonstrate that the proposed approach significantly improves cross-model and cross-dataset performance compared to existing methods. Diversity Matters highlights the critical role of data and feature diversity in building reliable and robust detectors against the rapidly evolving landscape of synthetic content.

Figures & tables

Explore similar work

CardsList
  1. Combating Pattern and Content Bias: Adversarial Feature Learning for Generalized AI-Generated Image Detection

    Apr 14, 2026Haifeng Zhang, Qinghui He, Xiuli Bi +3Adversarial DebiasingAI-Generated Image Detection

  2. EIB-Net: Entropy-Guided Information Bottleneck for Generalizable AI-Generated Image Detection

    Sep 24, 2026Zhida Zhang, Xinlei Ma, Jie CaoAI-Generated Image DetectionVariational Information Bottleneck