Attention-Based Deep Learning Pipeline for AI-Created Image Recognition

Abstract

The advent of generative AI models that can generate some of the most realistic images including DALL·E, Midjourney, and Stable Diffusion has posed a significant threat to digital forensics and media authenticity verifications. Conventional methods of forgery detection, which focus on splicing or copy-move manipulations, are not very efficient in the case of AI-generated images, which do not contain any explicit tampering traces. This paper presents Attention-Based Deep Learning Pipeline of AI-Created Image Recognition, intended to resolve the increasing challenge of separating AI-synthesized and authentic imagery. The proposed architecture incorporates three integrated branches, including low-level statistical feature extraction, high-level semantic representation learning, and attention-based feature refinement mechanism. The features that are extracted are combined and fed through a fully connected classifier to generate final predictions. The Synthbuster dataset that consists of images of several generative models, including DALL·E 2, DALL·E 3, Firefly, Glide, and versions of Stable Diffusion, were experimentally evaluated. The suggested framework score had an average AUC of 97.4%, as well as high accuracy, precision, recall, and F1-score. These findings support the robustness and generalization ability of the proposed model in detecting AI-generated images in a variety of generators and conditions.