REVIEW 1 cited by
A theoretical and empirical comparison claiming diffusion bridges have lower stochastic-optimal-control cost and greater robustness than flow matching when training data are scarce.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · deepseek-v4-flash
2026-08-04 13:52 UTC pith:J5VZQJ7C
Diffusion Bridge or Flow Matching? A Unifying Framework and Comparative Analysis
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
They claim two theoretical results. First, using stochastic optimal control, they say the bridge's control cost is always no larger than flow matching's cost, because the bridge adds a drift term that pushes the trajectory toward the target. The inequality itself is a simple consequence of their setup, but the paper goes further and says lower control cost means more stable and natural images, which is an interpretation, not a proof. Second, they claim that flow matching's linear interpolation between paired samples, the t and 1-t schedule, becomes invalid when the training data are limited, because the empirical measures are discrete and do not satisfy the smoothness condition of Brenier's optimal transport theorem. That step is the weakest part: flow matching trains on sample pairs, and linear interpolation of pairs remains a well-defined path even for discrete data. The empirical measures being atomic does not by itself predict that flow matching should collapse at small sample sizes, so the theoretical story does not actually explain the experimental pattern.
The experiments themselves show a consistent pattern: the diffusion bridge has better perceptual scores, especially when data are scarce or the transformation is large, while flow matching trains and samples faster. That pattern is useful evidence, but it needs
Core claim
The paper's load-bearing assertion is Theorem 4.2: J(u*DB) <= J(u*FM), interpreted as DB producing 'more stable and natural trajectories,' and Remark 4.3: for finite empirical measures, the interpolation coefficients t and 1-t of FM 'lose their validity' and the path is no longer an optimal transport interpolation. If correct, DB would be the preferable method for small training data and large distributional discrepancy.
Load-bearing premise
The Section 4.2 argument assumes that the validity of FM's linear schedule is governed by Brenier's absolute-continuity condition. The paper identifies the training measures as atomic and concludes that McCann's interpolation is undefined, therefore the t,1-t coefficients become ineffective. This conflates the well-definedness of sample-wise linear interpolation (what FM actually trains on) with existence of a continuous Brenier map; a discrete optimal transport plan between paired samples still defines a valid Wasserstein geodesic. Without this premise, the theoretical prediction of FM collapse at small n has no support. Location: Section 4.2, Remark 4.3 and Appendix A.3.
Editorial analysis
A structured set of objections, weighed in public.
Axiom & Free-Parameter Ledger
free parameters (4)
- lambda^2 (steady variance level) =
30^2/255^2
- terminal margin e^{-theta_bar_T} =
0.005
- cosine schedule parameter s =
0.008
- network and training hyperparameters =
patch=2, hidden=1024, depth=24, heads=16, MLP ratio=4.0, batch=8, steps=600k, LR=1e-4
axioms (6)
- standard math Brenier theorem and McCann interpolation for absolutely continuous measures
- domain assumption Certainty equivalence principle reduces the SDE SOC problem to an ODE SOC problem
- domain assumption The closed-form DB optimal controller from UniDB is correct
- ad hoc to paper Lower control cost J implies more stable and natural trajectories and better perceptual quality
- ad hoc to paper Absence of absolute continuity for empirical measures implies the t,1-t interpolation coefficients lose validity
- domain assumption All finite-pixel images lie in a compact space and a relevant class is absolutely continuous
read the original abstract
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.
Figures
Forward citations
Cited by 1 Pith paper
-
Residual Diffusion Bridge Model for Image Restoration
RDBM reformulates generalized diffusion bridge SDEs to use distribution residuals for adaptive noise modulation, unifying prior bridge models as special cases and achieving SOTA on image restoration tasks.
Reference graph
Works this paper leans on
-
[3]
Both training time and inference time are evaluated on a single NVIDIA H20 GPU
Task Dataset Training Data Size Method Training Time Inference Time Inpainting CelebA-HQ 500 Flow Matching 18 minutes 0.287 s 1000 36 minutes 0.287 s 5000 3 hours 0.287 s 27000 16 hours 0.287 s 500 Diffusion Bridge 4.5 hours 0.668 s 1000 9 hours 0.668 s 5000 12 hours 0.668 s 27000 66 hours 0.668 s Super-Resolution CelebA-HQ 27000 Flow Matching 16 hours 0....
arXiv 2014
-
[4]
Generative modeling with phase stochastic bridges.arXiv preprint arXiv:2310.07805, 2023a
Tianrong Chen, Jiatao Gu, Laurent Dinh, Evangelos A Theodorou, Joshua Susskind, and Shuangfei Zhai. Generative modeling with phase stochastic bridges.arXiv preprint arXiv:2310.07805, 2023a. Tianrong Chen, Jiatao Gu, Laurent Dinh, Evangelos A Theodorou, Joshua Susskind, and Shuangfei Zhai. Generative Modeling with Phase Stochastic Bridges.arXiv preprint ar...
-
[5]
Efficient Image Restoration via Latent Consistency Flow Matching.arXiv preprint arXiv:2502.03500,
Elad Cohen, Idan Achituve, Idit Diamant, Arnon Netzer, and Hai Victor Habi. Efficient Image Restoration via Latent Consistency Flow Matching.arXiv preprint arXiv:2502.03500,
-
[7]
Shutong Ding, Ke Hu, Shan Zhong, Haoyang Luo, Weinan Zhang, Jingya Wang, Jun Wang, and Ye Shi. Genpo: Generative Diffusion Models Meet On-Policy Reinforcement Learning.arXiv preprint arXiv:2505.18763,
-
[8]
Carles Domingo-Enrich, Michal Drozdzal, Brian Karrer, and Ricky TQ Chen. Adjoint matching: Fine-tuning flow and diffusion generative models with memoryless stochastic optimal control. arXiv preprint arXiv:2409.08861,
-
[9]
A Formal Evaluation of PSNR as Quality Measurement Parameter for Image Segmentation Algorithms
Fernando A Fardo, Victor H Conforto, Francisco C de Oliveira, and Paulo S Rodrigues. A Formal Evaluation of PSNR as Quality Measurement Parameter for Image Segmentation Algorithms. arXiv preprint arXiv:1605.07116,
-
[11]
Tero Karras. Progressive Growing of GANs for Improved Quality, Stability, and Variation.arXiv preprint arXiv:1710.10196,
-
[14]
Flow matching for generative modeling.arXiv preprint arXiv:2210.02747, 2022a
Yaron Lipman, Ricky TQ Chen, Heli Ben-Hamu, Maximilian Nickel, and Matt Le. Flow matching for generative modeling.arXiv preprint arXiv:2210.02747, 2022a. Yaron Lipman, Ricky TQ Chen, Heli Ben-Hamu, Maximilian Nickel, and Matthew Le. Flow Matching for Generative Modeling. InThe Eleventh International Conference on Learning Representations, 2022b. Yaron Lip...
Pith/arXiv arXiv 2023
-
[15]
Flow matching guide and code.arXiv preprint arXiv:2412.06264,
Yaron Lipman, Marton Havasi, Peter Holderrieth, Neta Shaul, Matt Le, Brian Karrer, Ricky TQ Chen, David Lopez-Paz, Heli Ben-Hamu, and Itai Gat. Flow matching guide and code.arXiv preprint arXiv:2412.06264,
-
[16]
Rectified flow: A marginal preserving approach to optimal transport.arXiv preprint arXiv:2209.14577,
Qiang Liu. Rectified flow: A marginal preserving approach to optimal transport.arXiv preprint arXiv:2209.14577,
-
[17]
Qihao Liu, Xi Yin, Alan Yuille, Andrew Brown, and Mannat Singh
URLhttps://icml.cc/virtual/2025/40011. Qihao Liu, Xi Yin, Alan Yuille, Andrew Brown, and Mannat Singh. Flowing from Words to Pixels: A Noise-Free Framework for cross-modality evolution. InProceedings of the Computer Vision and Pattern Recognition Conference, pp. 2755–2765,
2025
-
[18]
Xingchao Liu, Chengyue Gong, and Qiang Liu. Flow straight and fast: Learning to generate and transfer data with rectified flow.arXiv preprint arXiv:2209.03003,
-
[19]
Cheng Lu, Yuhao Zhou, Fan Bao, Jianfei Chen, Chongxuan Li, and Jun Zhu. DPM-Solver: A fast ODE solver for diffusion probabilistic model sampling in around 10 steps.Advances in Neural Information Processing Systems, 35:5775–5787, 2022a. Cheng Lu, Yuhao Zhou, Fan Bao, Jianfei Chen, Chongxuan Li, and Jun Zhu. DPM-Solver++: Fast solver for guided sampling of ...
-
[20]
Pnp-flow: Plug-and-play image restoration with flow matching.arXiv preprint arXiv:2410.02423,
13 Preprint S´egol`ene Martin, Anne Gagneux, Paul Hagemann, and Gabriele Steidl. Pnp-flow: Plug-and-play image restoration with flow matching.arXiv preprint arXiv:2410.02423,
-
[23]
Variational control for guidance in diffusion models.arXiv preprint arXiv:2502.03686,
Kushagra Pandey, Farrin Marouf Sofian, Felix Draxler, Theofanis Karaletsos, and Stephan Mandt. Variational control for guidance in diffusion models.arXiv preprint arXiv:2502.03686,
-
[25]
Moritz Reuss, Maximilian Li, Xiaogang Jia, and Rudolf Lioutikov. Goal-conditioned Imitation Learning using Score-based Diffusion Policies.arXiv preprint arXiv:2304.02532,
-
[26]
Litu Rout, Yujia Chen, Nataniel Ruiz, Abhishek Kumar, Constantine Caramanis, Sanjay Shakkottai, and Wen-Sheng Chu. RB-Modulation: Training-Free Personalization of Diffusion Models using Stochastic Optimal Control.arXiv preprint arXiv:2405.17401,
-
[27]
Denoising Diffusion Implicit Models.International Conference on Learning Representations, 2020a
Jiaming Song, Chenlin Meng, and Stefano Ermon. Denoising Diffusion Implicit Models.International Conference on Learning Representations, 2020a. 14 Preprint Yang Song, Jascha Sohl-Dickstein, Diederik P Kingma, Abhishek Kumar, Stefano Ermon, and Ben Poole. Score-based Generative Modeling Through Stochastic Differential Equations.arXiv preprint arXiv:2011.13...
Pith/arXiv arXiv 2011
-
[28]
Alexander Tong, Kilian Fatras, Nikolay Malkin, Guillaume Huguet, Yanlei Zhang, Jarrid Rector- Brooks, Guy Wolf, and Yoshua Bengio. Improving and generalizing flow-based generative models with minibatch optimal transport.arXiv preprint arXiv:2302.00482,
-
[29]
Zhendong Wang, Jonathan J Hunt, and Mingyuan Zhou. Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learning.arXiv preprint arXiv:2208.06193,
-
[31]
Tianyu Xie, Yu Zhu, Longlin Yu, Tong Yang, Ziheng Cheng, Shiyue Zhang, Xiangyu Zhang, and Cheng Zhang. Reflected Flow Matching. InInternational Conference on Machine Learning, pp. 54614–54634. PMLR, 2024a. Tianyu Xie, Yu Zhu, Longlin Yu, Tong Yang, Ziheng Cheng, Shiyue Zhang, Xiangyu Zhang, and Cheng Zhang. Reflected flow matching.arXiv preprint arXiv:240...
-
[32]
Denoising Diffusion Bridge Models
Linqi Zhou, Aaron Lou, Samar Khanna, and Stefano Ermon. Denoising Diffusion Bridge Models. arXiv preprint arXiv:2309.16948,
-
[33]
Kaizhen Zhu, Mokai Pan, Yuexin Ma, Yanwei Fu, Jingyi Yu, Jingya Wang, and Ye Shi. UniDB: A Uni- fied Diffusion Bridge Framework via Stochastic Optimal Control.arXiv preprint arXiv:2502.05749,
-
[34]
Mei et al
for the first time, systematically unifies Flow Matching with SOC, and on this basis proposes a novel, theoretically unbiased reward fine-tuning framework. Mei et al. (2024) proposes a Flow-Matching-based framework for SOC, which reformulates the classical SOC problem into a data-driven optimization task, thereby circumventing the intractability of solvin...
2024
-
[1958]
Unidb++: Fast Sampling of Unified Diffusion Bridge.arXiv preprint arXiv:2505.21528,
Mokai Pan, Kaizhen Zhu, Yuexin Ma, Yanwei Fu, Jingyi Yu, Jingya Wang, and Ye Shi. Unidb++: Fast Sampling of Unified Diffusion Bridge.arXiv preprint arXiv:2505.21528,
-
[1997]
Flow matching for stochastic linear control systems.arXiv preprint arXiv:2412.00617,
Yuhang Mei, Mohammad Al-Jarrah, Amirhossein Taghvaei, and Yongxin Chen. Flow matching for stochastic linear control systems.arXiv preprint arXiv:2412.00617,
-
[2004]
Shijie Wu, Yihang Zhu, Yunao Huang, Kaizhen Zhu, Jiayuan Gu, Jingyi Yu, Ye Shi, and Jingya Wang. AffordDP: Generalizable Diffusion Policy with Transferable Affordance.arXiv preprint arXiv:2412.03142,
-
[2008]
Diffusion models and gaussian flow matching: Two sides of the same coin
Ruiqi Gao, Emiel Hoogeboom, Jonathan Heek, Valentin De Bortoli, Kevin Patrick Murphy, and Tim Salimans. Diffusion models and gaussian flow matching: Two sides of the same coin. InThe Fourth Blogpost Track at ICLR 2025,
2025
-
[2014]
Auto-encoding variational bayes.arXiv preprint arXiv:1312.6114,
Diederik P Kingma and Max Welling. Auto-encoding variational bayes.arXiv preprint arXiv:1312.6114,
-
[2017]
Introduction to Stochastic Differential Equations, 1988a
R Ahmad. Introduction to Stochastic Differential Equations, 1988a. R Ahmad. Introduction to Stochastic Differential Equations, 1988b. Michael S Albergo and Eric Vanden-Eijnden. Building normalizing flows with stochastic interpolants. arXiv preprint arXiv:2209.15571, 2022a. Michael Samuel Albergo and Eric Vanden-Eijnden. Building normalizing flows with sto...
-
[2021]
Stochastic Optimal Control for Diffusion Bridges in Function Spaces.arXiv preprint arXiv:2405.20630,
Byoungwoo Park, Jungwon Choi, Sungbin Lim, and Juho Lee. Stochastic Optimal Control for Diffusion Bridges in Function Spaces.arXiv preprint arXiv:2405.20630,
-
[2022]
ADAM: A method for stochastic optimization.arXiv preprint arXiv:1412.6980,
Diederik P Kingma. ADAM: A method for stochastic optimization.arXiv preprint arXiv:1412.6980,
-
[2023]
Flow matching on general geometries.arXiv preprint arXiv:2302.03660,
Ricky TQ Chen and Yaron Lipman. Flow matching on general geometries.arXiv preprint arXiv:2302.03660,
-
[2024]
NTIRE 2017 Challenge on Single Image Super-Resolution
Eirikur Agustsson and Radu Timofte. NTIRE 2017 Challenge on Single Image Super-Resolution. In Proceedings of the IEEE conference on computer vision and pattern recognition workshops, pp. 126–135,
2017
-
[2025]
Flow matching in latent space.arXiv preprint arXiv:2307.08698,
Quan Dao, Hao Phung, Binh Nguyen, and Anh Tran. Flow matching in latent space.arXiv preprint arXiv:2307.08698,
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.