arXiv · 2403.14623
Simplified Diffusion Schr\"odinger Bridge
Abstract
This paper introduces a novel theoretical simplification of the Diffusion Schr\"odinger Bridge (DSB) that facilitates its unification with Score-based Generative Models (SGMs), addressing the limitations of DSB in complex data generation and enabling faster convergence and enhanced performance. By employing SGMs as an initial solution for DSB, our approach capitalizes on the strengths of both frameworks, ensuring a more efficient training process and improving the performance of SGM. We also propose a reparameterization technique that, despite theoretical approximations, practically improves the network's fitting capabilities. Our extensive experimental evaluations confirm the effectiveness of the simplified DSB, demonstrating its significant improvements. We believe the contributions of this work pave the way for advanced generative modeling.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Zhicong Tang, Tiankai Hang, Shuyang Gu, Dong Chen, Baining Guo. 2024-03-21. Simplified Diffusion Schr\"odinger Bridge. https://arxiv.org/abs/2403.14623
Cite the original work for its findings. Save a collection to share your selection of sources.