Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Diffusion Bridge or Flow Matching? A Unifying Framework and Comparative Analysis

About

Diffusion Bridge and Flow Matching have both demonstrated compelling empirical performance in transformation between arbitrary distributions. However, there remains confusion about which approach is generally preferable, and the substantial discrepancies in their modeling assumptions and practical implementations have hindered a unified theoretical account of their relative merits. We have, for the first time, provided a unified theoretical and experimental validation of these two models. We recast their frameworks through the lens of Stochastic Optimal Control and prove that the cost function of the Diffusion Bridge is lower, guiding the system toward more stable and natural trajectories. Simultaneously, from the perspective of Optimal Transport, interpolation coefficients $t$ and $1-t$ of Flow Matching become increasingly ineffective when the training data size is reduced. To corroborate these theoretical claims, we propose a novel, powerful architecture for Diffusion Bridge built on a latent Transformer, and implement a Flow Matching model with the same structure to enable a fair performance comparison in various experiments. Comprehensive experiments are conducted across Image Restoration, Translation, and Style Transfer tasks, systematically varying both the distributional discrepancy (different difficulty) and the training data size. Extensive empirical results align perfectly with our theoretical predictions and allow us to delineate the respective advantages and disadvantages of these two models. Our code is available at https://github.com/zhukaizhen/diffusion_bridge_flow_matching.

Kaizhen Zhu, Mokai Pan, Zhechuan Yu, Jingya Wang, Jingyi Yu, Ye Shi• 2025

Related benchmarks

TaskDatasetResultRank
Image InpaintingFFHQ 256x256
LPIPS0.044
21
Image-to-Image Translationedges2shoes
FID20.51
13
4x super-resolutionCelebA-HQ
PSNR27.47
4
Image DeblurringDeblurring
PSNR24.67
2
Image InpaintingFFHQ 256x256
LPIPS0.059
2
Image InpaintingInpainting
PSNR23.57
2
Image-to-Image TranslationEdges to Handbags
MSE0.102
2
Image-to-Image TranslationNights to Days
MSE0.114
2
Super-ResolutionSuper-Resolution 4x
PSNR27.47
2
Showing 9 of 9 rows

Other info

Follow for update