用扩散模型实现无配对图像翻译,提升生成质量和结构一致性。
CycleDiff: Cycle Diffusion Models for Unpaired Image-to-image Translation
- 将扩散过程与图像翻译对齐,实现端到端联合学习。
- 时间依赖的翻译网络提升复杂映射学习能力。
- 在多模态翻译任务中超越现有方法性能。
我们提出一种基于扩散模型的跨域图像翻译方法,无需成对训练数据。与基于GAN的方法不同,该方法利用扩散模型更全面地建模数据分布,提升跨域翻译效果。然而,扩散过程作用于噪声信号,而翻译过程针对清晰信号,两者难以对齐。现有方法采用分步训练或浅层融合,易陷入局部最优。为此,我们设计一种新的联合学习框架,通过扩散模型提取图像成分以表示干净信号,并在此基础上进行翻译,实现端到端联合优化。同时引入时间依赖的翻译网络,有效学习复杂映射关系。实验在RGB↔RGB及多种跨模态任务(如RGB↔Edge、RGB↔Semantics、RGB↔Depth)上验证,生成效果优于当前最先进方法。
原文摘要 · Abstract (English)
We introduce a diffusion-based cross-domain image translator in the absence of paired training data. Unlike GAN-based methods, our approach integrates diffusion models to learn the image translation process, allowing for more coverable modeling of the data distribution and performance improvement of the cross-domain translation. However, incorporating the translation process within the diffusion process is still challenging since the two processes are not aligned exactly, i.e., the diffusion process is applied to the noisy signal while the translation process is conducted on the clean signal. As a result, recent diffusion-based studies employ separate training or shallow integration to learn the two processes, yet this may cause the local minimal of the translation optimization, constraining the effectiveness of diffusion models. To address the problem, we propose a novel joint learning framework that aligns the diffusion and the translation process, thereby improving the global optimality. Specifically, we propose to extract the image components with diffusion models to represent the clean signal and employ the translation process with the image components, enabling an end-to-end joint learning manner. On the other hand, we introduce a time-dependent translation network to learn the complex translation mapping, resulting in effective translation learning and significant performance improvement. Benefiting from the design of joint learning, our method enables global optimization of both processes, enhancing the optimality and achieving improved fidelity and structural consistency. We have conducted extensive experiments on RGB$\leftrightarrow$RGB and diverse cross-modality translation tasks including RGB$\leftrightarrow$Edge, RGB$\leftrightarrow$Semantics and RGB$\leftrightarrow$Depth, showcasing better generative performances than the state of the arts.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。