Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

AEROBLADE: Training-Free Detection of Latent Diffusion Images Using Autoencoder Reconstruction Error

About

With recent text-to-image models, anyone can generate deceptively realistic images with arbitrary contents, fueling the growing threat of visual disinformation. A key enabler for generating high-resolution images with low computational cost has been the development of latent diffusion models (LDMs). In contrast to conventional diffusion models, LDMs perform the denoising process in the low-dimensional latent space of a pre-trained autoencoder (AE) instead of the high-dimensional image space. Despite their relevance, the forensic analysis of LDMs is still in its infancy. In this work we propose AEROBLADE, a novel detection method which exploits an inherent component of LDMs: the AE used to transform images between image and latent space. We find that generated images can be more accurately reconstructed by the AE than real images, allowing for a simple detection approach based on the reconstruction error. Most importantly, our method is easy to implement and does not require any training, yet nearly matches the performance of detectors that rely on extensive training. We empirically demonstrate that AEROBLADE is effective against state-of-the-art LDMs, including Stable Diffusion and Midjourney. Beyond detection, our approach allows for the qualitative analysis of images, which can be leveraged for identifying inpainted regions. We release our code and data at https://github.com/jonasricker/aeroblade .

Jonas Ricker, Denis Lukovnikov, Asja Fischer• 2024

Related benchmarks

TaskDatasetResultRank
Generated Image DetectionGenImage (test)
Average Accuracy48.8
124
AI-generated image detectionGenImage--
106
AI Image DetectionMidjourney
Accuracy43
51
Generated Image DetectionWukong
Accuracy49
41
AI-generated image detectionCelebA-HQ v2 (test)
ACC99.35
40
AI-generated image detectionGauGAN
mAP71
29
AI-generated image detectionCycleGAN
mAP73
29
AI-generated image detectionBigGAN
mAP61
29
AI-generated image detectionStyleGAN
mAP0.59
29
AI-generated image detectionProGAN
mAP49
29
Showing 10 of 59 rows

Other info

Follow for update