A Bias-Free Training Paradigm for More General AI-generated Image Detection

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Guillaro, Fabrizio, Zingarini, Giada, Usman, Ben, Sud, Avneesh, Cozzolino, Davide, Verdoliva, Luisa
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910902297559040
author Guillaro, Fabrizio
Zingarini, Giada
Usman, Ben
Sud, Avneesh
Cozzolino, Davide
Verdoliva, Luisa
author_facet Guillaro, Fabrizio
Zingarini, Giada
Usman, Ben
Sud, Avneesh
Cozzolino, Davide
Verdoliva, Luisa
contents Successful forensic detectors can produce excellent results in supervised learning benchmarks but struggle to transfer to real-world applications. We believe this limitation is largely due to inadequate training data quality. While most research focuses on developing new algorithms, less attention is given to training data selection, despite evidence that performance can be strongly impacted by spurious correlations such as content, format, or resolution. A well-designed forensic detector should detect generator specific artifacts rather than reflect data biases. To this end, we propose B-Free, a bias-free training paradigm, where fake images are generated from real ones using the conditioning procedure of stable diffusion models. This ensures semantic alignment between real and fake images, allowing any differences to stem solely from the subtle artifacts introduced by AI generation. Through content-based augmentation, we show significant improvements in both generalization and robustness over state-of-the-art detectors and more calibrated results across 27 different generative models, including recent releases, like FLUX and Stable Diffusion 3.5. Our findings emphasize the importance of a careful dataset design, highlighting the need for further research on this topic. Code and data are publicly available at https://grip-unina.github.io/B-Free/.
format Preprint
id arxiv_https___arxiv_org_abs_2412_17671
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Bias-Free Training Paradigm for More General AI-generated Image Detection
Guillaro, Fabrizio
Zingarini, Giada
Usman, Ben
Sud, Avneesh
Cozzolino, Davide
Verdoliva, Luisa
Computer Vision and Pattern Recognition
Successful forensic detectors can produce excellent results in supervised learning benchmarks but struggle to transfer to real-world applications. We believe this limitation is largely due to inadequate training data quality. While most research focuses on developing new algorithms, less attention is given to training data selection, despite evidence that performance can be strongly impacted by spurious correlations such as content, format, or resolution. A well-designed forensic detector should detect generator specific artifacts rather than reflect data biases. To this end, we propose B-Free, a bias-free training paradigm, where fake images are generated from real ones using the conditioning procedure of stable diffusion models. This ensures semantic alignment between real and fake images, allowing any differences to stem solely from the subtle artifacts introduced by AI generation. Through content-based augmentation, we show significant improvements in both generalization and robustness over state-of-the-art detectors and more calibrated results across 27 different generative models, including recent releases, like FLUX and Stable Diffusion 3.5. Our findings emphasize the importance of a careful dataset design, highlighting the need for further research on this topic. Code and data are publicly available at https://grip-unina.github.io/B-Free/.
title A Bias-Free Training Paradigm for More General AI-generated Image Detection
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.17671