OT-ALD: Aligning Latent Distributions with Optimal Transport for Accelerated Image-to-Image Translation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Zhanpeng, Cao, Shuting, Lu, Yuhang, Li, Yuhan, Lei, Na, Luo, Zhongxuan
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914158463680512
author Wang, Zhanpeng
Cao, Shuting
Lu, Yuhang
Li, Yuhan
Lei, Na
Luo, Zhongxuan
author_facet Wang, Zhanpeng
Cao, Shuting
Lu, Yuhang
Li, Yuhan
Lei, Na
Luo, Zhongxuan
contents The Dual Diffusion Implicit Bridge (DDIB) is an emerging image-to-image (I2I) translation method that preserves cycle consistency while achieving strong flexibility. It links two independently trained diffusion models (DMs) in the source and target domains by first adding noise to a source image to obtain a latent code, then denoising it in the target domain to generate the translated image. However, this method faces two key challenges: (1) low translation efficiency, and (2) translation trajectory deviations caused by mismatched latent distributions. To address these issues, we propose a novel I2I translation framework, OT-ALD, grounded in optimal transport (OT) theory, which retains the strengths of DDIB-based approach. Specifically, we compute an OT map from the latent distribution of the source domain to that of the target domain, and use the mapped distribution as the starting point for the reverse diffusion process in the target domain. Our error analysis confirms that OT-ALD eliminates latent distribution mismatches. Moreover, OT-ALD effectively balances faster image translation with improved image quality. Experiments on four translation tasks across three high-resolution datasets show that OT-ALD improves sampling efficiency by 20.29% and reduces the FID score by 2.6 on average compared to the top-performing baseline models.
format Preprint
id arxiv_https___arxiv_org_abs_2511_11162
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle OT-ALD: Aligning Latent Distributions with Optimal Transport for Accelerated Image-to-Image Translation
Wang, Zhanpeng
Cao, Shuting
Lu, Yuhang
Li, Yuhan
Lei, Na
Luo, Zhongxuan
Computer Vision and Pattern Recognition
Artificial Intelligence
The Dual Diffusion Implicit Bridge (DDIB) is an emerging image-to-image (I2I) translation method that preserves cycle consistency while achieving strong flexibility. It links two independently trained diffusion models (DMs) in the source and target domains by first adding noise to a source image to obtain a latent code, then denoising it in the target domain to generate the translated image. However, this method faces two key challenges: (1) low translation efficiency, and (2) translation trajectory deviations caused by mismatched latent distributions. To address these issues, we propose a novel I2I translation framework, OT-ALD, grounded in optimal transport (OT) theory, which retains the strengths of DDIB-based approach. Specifically, we compute an OT map from the latent distribution of the source domain to that of the target domain, and use the mapped distribution as the starting point for the reverse diffusion process in the target domain. Our error analysis confirms that OT-ALD eliminates latent distribution mismatches. Moreover, OT-ALD effectively balances faster image translation with improved image quality. Experiments on four translation tasks across three high-resolution datasets show that OT-ALD improves sampling efficiency by 20.29% and reduces the FID score by 2.6 on average compared to the top-performing baseline models.
title OT-ALD: Aligning Latent Distributions with Optimal Transport for Accelerated Image-to-Image Translation
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2511.11162