用CLIP引导的扩散Transformer实现高质量图像翻译
Image-to-Image Translation with Diffusion Transformers and CLIP-Based Image Conditioning
- 用DiT架构结合CLIP图像嵌入做条件控制
- 在人脸转漫画、边缘图转鞋子任务上效果优异
- 无需文本标签,保持语义一致与视觉真实
图像到图像翻译旨在学习源域与目标域之间的映射,支持风格迁移、外观变换和领域自适应等任务。本文探索基于扩散模型的图像翻译框架,采用融合去噪能力与全局建模优势的扩散Transformer(DiT)。通过预训练的CLIP编码器提取图像嵌入作为条件,实现无需文本或类别标签的细粒度、结构一致的翻译。训练中引入CLIP相似性损失以保证语义一致性,以及LPIPS感知损失以提升视觉保真度。在两个基准数据集face2comics(真实人脸转漫画)和edges2shoes(边缘图转鞋类图像)上验证,结果表明,结合CLIP条件与感知损失的DiT可生成高质量、语义忠实的翻译结果,为成对图像翻译提供优于GAN的替代方案。
原文摘要 · Abstract (English)
Image-to-image translation aims to learn a mapping between a source and a target domain, enabling tasks such as style transfer, appearance transformation, and domain adaptation. In this work, we explore a diffusion-based framework for image-to-image translation by adapting Diffusion Transformers (DiT), which combine the denoising capabilities of diffusion models with the global modeling power of transformers. To guide the translation process, we condition the model on image embeddings extracted from a pre-trained CLIP encoder, allowing for fine-grained and structurally consistent translations without relying on text or class labels. We incorporate both a CLIP similarity loss to enforce semantic consistency and an LPIPS perceptual loss to enhance visual fidelity during training. We validate our approach on two benchmark datasets: face2comics, which translates real human faces to comic-style illustrations, and edges2shoes, which translates edge maps to realistic shoe images. Experimental results demonstrate that DiT, combined with CLIP-based conditioning and perceptual similarity objectives, achieves high-quality, semantically faithful translations, offering a promising alternative to GAN-based models for paired image-to-image translation tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。