OmniDFA: A Unified Framework for Open Set Synthesis Image Detection and Few-Shot Attribution

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wu, Shiyu, Li, Shuyan, Li, Jing, Liu, Jing, Wang, Yequan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916979232735232
author Wu, Shiyu
Li, Shuyan
Li, Jing
Liu, Jing
Wang, Yequan
author_facet Wu, Shiyu
Li, Shuyan
Li, Jing
Liu, Jing
Wang, Yequan
contents AI-generated image (AIGI) detection and source model attribution remain central challenges in combating deepfake abuses, primarily due to the structural diversity of generative models. Current detection methods are prone to overfitting specific forgery traits, whereas source attribution offers a robust alternative through fine-grained feature discrimination. However, synthetic image attribution remains constrained by the scarcity of large-scale, well-categorized synthetic datasets, limiting its practicality and compatibility with detection systems. In this work, we propose a new paradigm for image attribution called open-set, few-shot source identification. This paradigm is designed to reliably identify unseen generators using only limited samples, making it highly suitable for real-world application. To this end, we introduce OmniDFA (Omni Detector and Few-shot Attributor), a novel framework for AIGI that not only assesses the authenticity of images, but also determines the synthesis origins in a few-shot manner. To facilitate this work, we construct OmniFake, a large class-aware synthetic image dataset that curates $1.17$ M images from $45$ distinct generative models, substantially enriching the foundational resources for research on both AIGI detection and attribution. Experiments demonstrate that OmniDFA exhibits excellent capability in open-set attribution and achieves state-of-the-art generalization performance on AIGI detection. Our dataset and code will be made available.
format Preprint
id arxiv_https___arxiv_org_abs_2509_25682
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle OmniDFA: A Unified Framework for Open Set Synthesis Image Detection and Few-Shot Attribution
Wu, Shiyu
Li, Shuyan
Li, Jing
Liu, Jing
Wang, Yequan
Computer Vision and Pattern Recognition
AI-generated image (AIGI) detection and source model attribution remain central challenges in combating deepfake abuses, primarily due to the structural diversity of generative models. Current detection methods are prone to overfitting specific forgery traits, whereas source attribution offers a robust alternative through fine-grained feature discrimination. However, synthetic image attribution remains constrained by the scarcity of large-scale, well-categorized synthetic datasets, limiting its practicality and compatibility with detection systems. In this work, we propose a new paradigm for image attribution called open-set, few-shot source identification. This paradigm is designed to reliably identify unseen generators using only limited samples, making it highly suitable for real-world application. To this end, we introduce OmniDFA (Omni Detector and Few-shot Attributor), a novel framework for AIGI that not only assesses the authenticity of images, but also determines the synthesis origins in a few-shot manner. To facilitate this work, we construct OmniFake, a large class-aware synthetic image dataset that curates $1.17$ M images from $45$ distinct generative models, substantially enriching the foundational resources for research on both AIGI detection and attribution. Experiments demonstrate that OmniDFA exhibits excellent capability in open-set attribution and achieves state-of-the-art generalization performance on AIGI detection. Our dataset and code will be made available.
title OmniDFA: A Unified Framework for Open Set Synthesis Image Detection and Few-Shot Attribution
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2509.25682