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

DeepInv: 一种用于快速准确扩散逆向的新型自监督学习方法

DeepInv: A Novel Self-supervised Learning Approach for Fast and Accurate Diffusion Inversion

Xiamen University(厦门大学);National University of Defense Technology(国防科技大学)

Ziyue Zhang, Luxi Lin, Xiaolin Hu, Chao Chang, HuaiXi Wang, Yiyi Zhou, Rongrong Ji

AI总结:

DeepInv通过自监督学习和数据增强策略,实现快速准确的扩散逆向,优于现有方法。

摘要:

扩散逆向是恢复扩散模型中图像噪声的任务,对于可控的扩散图像编辑至关重要。目前,扩散逆向仍然是一项具有挑战性的任务,由于缺乏有效的监督信号。因此,大多数现有方法依赖于基于近似的解决方案,但这些方案往往以性能或效率为代价。为了弥补这些不足,本文提出了一种新的自监督扩散逆向方法,称为Deep Inversion(DeepInv)。不同于需要真实噪声注释,我们引入了一个自监督目标以及数据增强策略,从真实图像中生成高质量的伪噪声,而无需人工干预。基于这两种创新设计,DeepInv还配备了迭代和多尺度的训练制度,以训练一个参数化的逆向求解器,从而实现快速且准确的图像到噪声映射。据我们所知,这是首次尝试提出可训练的求解器,逐步预测逆向噪声。广泛的实验表明,我们的DeepInv在性能和推理速度上均优于比较方法,例如在COCO数据集上,比EasyInv高40.435%的SSIM,比ReNoise快9887.5%。此外,我们对可训练求解器的精心设计也可以为社区提供见解。代码和模型参数将在https://github.com/potato-kitty/DeepInv上发布。

英文摘要:

Diffusion inversion is a task of recovering the noise of an image in a diffusion model, which is vital for controllable diffusion image editing. At present, diffusion inversion still remains a challenging task due to the lack of viable supervision signals. Thus, most existing methods resort to approximation-based solutions, which however are often at the cost of performance or efficiency. To remedy these shortcomings, we propose a novel self-supervised diffusion inversion approach in this paper, termed Deep Inversion (DeepInv). Instead of requiring ground-truth noise annotations, we introduce a self-supervised objective as well as a data augmentation strategy to generate high-quality pseudo noises from real images without manual intervention. Based on these two innovative designs, DeepInv is also equipped with an iterative and multi-scale training regime to train a parameterized inversion solver, thereby achieving the fast and accurate image-to-noise mapping. To the best of our knowledge, this is the first attempt of presenting a trainable solver to predict inversion noise step by step. The extensive experiments show that our DeepInv can achieve much better performance and inference speed than the compared methods, e.g., +40.435% SSIM than EasyInv and +9887.5% speed than ReNoise on COCO dataset. Moreover, our careful designs of trainable solvers can also provide insights to the community. Codes and model parameters will be released in https://github.com/potato-kitty/DeepInv.

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

↑