arXiv · 2509.25682
Few-Shot Synthetic Image Attribution: Identifying Unseen Generators with Limited Samples
Abstract
AI-generated image (AIGI) attribution presents a pressing challenge that goes beyond mere AIGI detection, aiming to identify the source model or technique responsible for a synthetic image. However, most previous source attribution methods operate in a closed-set manner, which necessitates retraining to recognize any novel category, preventing adaptation to the rapid evolution of image generation. In this work, we propose a new paradigm for synthetic image attribution, termed few-shot attribution. This paradigm targets the reliable identification of unseen generators using only limited samples, making it highly suitable for real-world applications. To facilitate this work, we construct OmniFake, a large-scale, well-categorized synthetic image dataset that contains $1.17$ million images from $45$ distinct generators. We further introduce OmniDFA (Omni Detector and Few-shot Attributor), a few-shot attribution baseline that not only assesses the authenticity of images but also determines their synthesis origins. Experiments demonstrate that OmniDFA exhibits excellent capability in few-shot attribution and achieves state-of-the-art generalization performance in AIGI detection. Our dataset and code are available at https://github.com/teheperinko541/OmniDFA.
Explore related subjects
Keep this discovery
Shiyu Wu, Shuyan Li, Jing Li, Jing Liu, Yequan Wang. 2025-09-30. Few-Shot Synthetic Image Attribution: Identifying Unseen Generators with Limited Samples. https://arxiv.org/abs/2509.25682
Cite the original work for its findings. Save a collection to share your selection of sources.