arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.02816cs.CVcs.AI

InpDiffusion:基于条件扩散模型的图像修复定位

InpDiffusion: Image Inpainting Localization via Conditional Diffusion Models

Beijing University of Posts and Telecommunications(北京邮电大学)

Kai Wang, Shaozhang Niu, Qixian Hao, Jiwei Zhang

AI总结:

针对图像修复定位中的过度自信和细微篡改边界难检测问题,提出条件扩散框架 InpDiffusion,结合语义、边缘监督与 DMFE 多尺度特征提取,显著提升定位性能、泛化性和鲁棒性。

摘要:

随着人工智能的快速发展,尤其是 GAN 和扩散模型的出现,图像修复定位(Image Inpainting Localization, IIL)的准确性面临越来越大的挑战。当前 IIL 方法主要面临两个难题:一是存在过度自信倾向,从而导致错误预测;二是难以检测修复图像中细微的篡改边界。为此,我们提出一种新范式,利用扩散模型将 IIL 视为条件掩码生成任务。我们的方法 InpDiffusion 借助融合图像语义条件而增强的去噪过程,逐步细化预测。在去噪期间,我们采用边缘条件,并引入一种新颖的边缘监督策略,以增强模型对修复对象中边缘细节的感知。通过平衡扩散模型的随机采样与对被篡改图像区域的边缘监督,既缓解了过度自信造成错误预测的风险,也避免了因随机性过强而导致细微边界丢失。此外,我们提出创新的双流多尺度特征提取器(Dual-stream Multi-scale Feature Extractor, DMFE),用于提取多尺度特征,并同时考虑修复图像的语义条件和边缘条件,以增强特征表示。在多个具有挑战性的数据集上进行的大量实验表明,InpDiffusion 在 IIL 任务中显著优于现有最先进方法,同时展现出优异的泛化能力和鲁棒性。

英文摘要:

As artificial intelligence advances rapidly, particularly with the advent of GANs and diffusion models, the accuracy of Image Inpainting Localization (IIL) has become increasingly challenging. Current IIL methods face two main challenges: a tendency towards overconfidence, leading to incorrect predictions; and difficulty in detecting subtle tampering boundaries in inpainted images. In response, we propose a new paradigm that treats IIL as a conditional mask generation task utilizing diffusion models. Our method, InpDiffusion, utilizes the denoising process enhanced by the integration of image semantic conditions to progressively refine predictions. During denoising, we employ edge conditions and introduce a novel edge supervision strategy to enhance the model's perception of edge details in inpainted objects. Balancing the diffusion model's stochastic sampling with edge supervision of tampered image regions mitigates the risk of incorrect predictions from overconfidence and prevents the loss of subtle boundaries that can result from overly stochastic processes. Furthermore, we propose an innovative Dual-stream Multi-scale Feature Extractor (DMFE) for extracting multi-scale features, enhancing feature representation by considering both semantic and edge conditions of the inpainted images. Extensive experiments across challenging datasets demonstrate that the InpDiffusion significantly outperforms existing state-of-the-art methods in IIL tasks, while also showcasing excellent generalization capabilities and robustness.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑