Towards Generalized Image Manipulation Localization via Score-based Model

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Yunfei, Du, Bo, Yang, Zhe, Liu, Xin, Lin, Zhiyu, Xu, Tianxin, Zhou, Ji-Zhe
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914572141592576
author Wang, Yunfei
Du, Bo
Yang, Zhe
Liu, Xin
Lin, Zhiyu
Xu, Tianxin
Zhou, Ji-Zhe
author_facet Wang, Yunfei
Du, Bo
Yang, Zhe
Liu, Xin
Lin, Zhiyu
Xu, Tianxin
Zhou, Ji-Zhe
contents With the rapid evolution of synthetic media, Image Manipulation Localization (IML) has emerged as a critical component in multimedia forensics for ensuring the integrity of digital content. However, generalization remains a core challenge, as existing discriminative methods typically learn a fixed decision boundary that tends to overfit to specific training artifacts and fails to adapt to unseen manipulation types. To address this, we propose DiffIML, a novel framework that introduces score-based generative modeling to IML. Diverging from the direct estimation of hard boundaries, DiffIML approximates the score function, the gradient of the log-likelihood, to capture the intrinsic geometric topology of mask distributions. This paradigm leverages structural priors to iteratively recover coherent masks from noise, thereby circumventing the brittleness associated with discriminative models. Under this formulation, diffusion models serve as an effective numerical solver for the learned score function.To ensure practicality, we respectively resolve the efficiency and stability bottlenecks of standard diffusion by: (1) utilizing a Lightweight Mask-Specific VAE for fast latent-space process and a decoupled architecture with a lightweight denoising UNet, (2) edge supervision and error prior to mitigate error accumulation during sampling. Extensive experiments of two distinct protocols on eight non-generative and three generative benchmarks demonstrate that DiffIML consistently outperforms state-of-the-art methods, yielding remarkable generalization improvements on diverse unseen datasets. The code will be publicly available.
format Preprint
id arxiv_https___arxiv_org_abs_2605_16879
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Towards Generalized Image Manipulation Localization via Score-based Model
Wang, Yunfei
Du, Bo
Yang, Zhe
Liu, Xin
Lin, Zhiyu
Xu, Tianxin
Zhou, Ji-Zhe
Computer Vision and Pattern Recognition
With the rapid evolution of synthetic media, Image Manipulation Localization (IML) has emerged as a critical component in multimedia forensics for ensuring the integrity of digital content. However, generalization remains a core challenge, as existing discriminative methods typically learn a fixed decision boundary that tends to overfit to specific training artifacts and fails to adapt to unseen manipulation types. To address this, we propose DiffIML, a novel framework that introduces score-based generative modeling to IML. Diverging from the direct estimation of hard boundaries, DiffIML approximates the score function, the gradient of the log-likelihood, to capture the intrinsic geometric topology of mask distributions. This paradigm leverages structural priors to iteratively recover coherent masks from noise, thereby circumventing the brittleness associated with discriminative models. Under this formulation, diffusion models serve as an effective numerical solver for the learned score function.To ensure practicality, we respectively resolve the efficiency and stability bottlenecks of standard diffusion by: (1) utilizing a Lightweight Mask-Specific VAE for fast latent-space process and a decoupled architecture with a lightweight denoising UNet, (2) edge supervision and error prior to mitigate error accumulation during sampling. Extensive experiments of two distinct protocols on eight non-generative and three generative benchmarks demonstrate that DiffIML consistently outperforms state-of-the-art methods, yielding remarkable generalization improvements on diverse unseen datasets. The code will be publicly available.
title Towards Generalized Image Manipulation Localization via Score-based Model
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2605.16879