Alert button

Unpaired Image-to-Image Translation via Neural Schrödinger Bridge

May 24, 2023
Beomsu Kim, Gihyun Kwon, Kwanyoung Kim, Jong Chul Ye

Figure 1 for Unpaired Image-to-Image Translation via Neural Schrödinger Bridge
Figure 2 for Unpaired Image-to-Image Translation via Neural Schrödinger Bridge
Figure 3 for Unpaired Image-to-Image Translation via Neural Schrödinger Bridge
Figure 4 for Unpaired Image-to-Image Translation via Neural Schrödinger Bridge

Share this with someone who'll enjoy it:

Diffusion models are a powerful class of generative models which simulate stochastic differential equations (SDEs) to generate data from noise. Although diffusion models have achieved remarkable progress in recent years, they have limitations in the unpaired image-to-image translation tasks due to the Gaussian prior assumption. Schr\"odinger Bridge (SB), which learns an SDE to translate between two arbitrary distributions, have risen as an attractive solution to this problem. However, none of SB models so far have been successful at unpaired translation between high-resolution images. In this work, we propose the Unpaired Neural Schr\"odinger Bridge (UNSB), which combines SB with adversarial training and regularization to learn a SB between unpaired data. We demonstrate that UNSB is scalable, and that it successfully solves various unpaired image-to-image translation tasks. Code: \url{https://github.com/cyclomon/UNSB}

View paper onarxiv icon

Share this with someone who'll enjoy it: