ReTD: Reconstruction-Based Traceability Detection for Generated Images

The objective of generated image traceability is to accurately identify and locate the source models. In this paper, we propose ReTD (Reconstruction-Based Traceability Detection), a generalized model for generated image traceability detection. Firstly, we use VAE to reconstruct images which are compared with the original ones to extract numerical distinguishing features. Secondly, we use Vision Transformer to learn the fine-grained distribution features to realize the generated image traceability classification. Finally, we conduct traceability experiments using images generated by ten GAN and Diffusion models. The experimental results demonstrate that ReTD only training a unified classifier improves accuracy by 9.4% compared to the state-of-the-art method. The ReTD-related code is availble at https://github.com/chenweizhuo/ReTD.