Pith. sign in

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 →

arxiv 2509.24531 v2 pith:J5VZQJ7C submitted 2025-09-29 cs.CV

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

classification cs.CV
keywords bridgediffusionflowmatchingtheoreticaldataempiricalexperiments
verification ladder T0 review T1 audit T2 compute T3 formal T4 reserved

The pith

A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.

This paper asks a practical question: when you want a model that turns one image distribution into another, for example low-quality to high-quality or a photo to a painting, should you use a diffusion bridge or flow matching? The authors build the same transformer backbone for both methods so the comparison is fair, then test them on inpainting, super-resolution, deblurring, denoising, style transfer, and translation, with different training set sizes.

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.

Desk editor's note, referee report, simulated authors' rebuttal, and a circularity audit.

Axiom & Free-Parameter Ledger

4 free parameters · 6 axioms · 0 invented entities

The central cost inequality depends on the closed-form DB controller imported from UniDB (Zhu et al., 2025, same authors), and the data-size prediction depends on a controversial reading of Brenier's theorem applied to empirical measures. The practical experiments use several chosen hyperparameters (lambda^2, schedule s, terminal margin, network dimensions) that are not tied to the theory.

free parameters (4)
  • lambda^2 (steady variance level) = 30^2/255^2
    Chosen in Appendix D from prior bridge work; scales the noise schedule and is a hyperparameter, though the cost inequality in Theorem 4.2 holds for any positive lambda^2.
  • terminal margin e^{-theta_bar_T} = 0.005
    Chosen in Appendix D to avoid exact terminal hitting; affects the practical DB sampling but not the proof.
  • cosine schedule parameter s = 0.008
    Flipped cosine noise schedule for theta_t; taken from prior bridge papers and adjusted by hand.
  • network and training hyperparameters = patch=2, hidden=1024, depth=24, heads=16, MLP ratio=4.0, batch=8, steps=600k, LR=1e-4
    Selected by hand for the DiT architecture; not tied to the theoretical claims but needed for the experiments.
axioms (6)
  • standard math Brenier theorem and McCann interpolation for absolutely continuous measures
    Invoked in Section 4.2 to characterize the optimal transport map and the dynamic OT solution.
  • domain assumption Certainty equivalence principle reduces the SDE SOC problem to an ODE SOC problem
    Used in Section 4.1 to make the DB and FM SOC formulations comparable; cited from prior work.
  • domain assumption The closed-form DB optimal controller from UniDB is correct
    Equation (17) takes u*DB from Zhu et al. 2025, a paper by the same authors; Theorem 4.2 depends on it.
  • ad hoc to paper Lower control cost J implies more stable and natural trajectories and better perceptual quality
    Stated after Theorem 4.2 and used to connect the cost inequality to image quality; no argument is given.
  • ad hoc to paper Absence of absolute continuity for empirical measures implies the t,1-t interpolation coefficients lose validity
    Remark 4.3 makes this inference, but the paper does not derive it from the failure of McCann interpolation.
  • domain assumption All finite-pixel images lie in a compact space and a relevant class is absolutely continuous
    Assumed in Section 4.2 before invoking Brenier; real images are discrete-valued, so the assumption is inconsistent with the actual data.

pith-pipeline@v1.3.0-alltime-deepseek · 21803 in / 14072 out tokens · 113037 ms · 2026-08-04T13:52:01.826478+00:00 · methodology

0 comments
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

Figures reproduced from arXiv: 2509.24531 by Jingya Wang, Jingyi Yu, Kaizhen Zhu, Mokai Pan, Ye Shi, Zhechuan Yu.

Figure 1
Figure 1. Figure 1: Overview of Diffusion Bridge (DB) versus Flow Matching (FM) on mapping one distri [PITH_FULL_IMAGE:figures/full_fig_p002_1.png] view at source ↗
Figure 2
Figure 2. Figure 2: Qualitative comparison of visual results between Diffusion Bridge and Flow Matching on [PITH_FULL_IMAGE:figures/full_fig_p008_2.png] view at source ↗
Figure 3
Figure 3. Figure 3: The performance (LPIPS and FID) of Flow Matching and Diffusion Bridge under Image [PITH_FULL_IMAGE:figures/full_fig_p008_3.png] view at source ↗
Figure 4
Figure 4. Figure 4: Qualitative comparison of visual results between Diffusion Bridge and Flow Matching [PITH_FULL_IMAGE:figures/full_fig_p009_4.png] view at source ↗
Figure 5
Figure 5. Figure 5: The performance (FID) of Flow Matching and Diffusion Bridge under Image Inpainting [PITH_FULL_IMAGE:figures/full_fig_p009_5.png] view at source ↗
Figure 6
Figure 6. Figure 6: Training (with training data size 27000) and inference time (NFEs = 20) comparison on [PITH_FULL_IMAGE:figures/full_fig_p010_6.png] view at source ↗
Figure 7
Figure 7. Figure 7: Qualitative comparison of visual results between Diffusion Bridge and Flow Matching in [PITH_FULL_IMAGE:figures/full_fig_p023_7.png] view at source ↗
Figure 8
Figure 8. Figure 8: Additional Diffusion Bridge visual results on Image Translation with CelebFaces dataset. [PITH_FULL_IMAGE:figures/full_fig_p023_8.png] view at source ↗
Figure 9
Figure 9. Figure 9: Additional Diffusion Bridge visual results on Style-Transfer with CelebA-HQ dataset. [PITH_FULL_IMAGE:figures/full_fig_p024_9.png] view at source ↗
Figure 10
Figure 10. Figure 10: Additional visual comparison between Diffusion Bridge and Flow Matching on 4 [PITH_FULL_IMAGE:figures/full_fig_p024_10.png] view at source ↗
Figure 11
Figure 11. Figure 11: Additional Diffusion Bridge and Flow Matching visual results on Image Inpainting [PITH_FULL_IMAGE:figures/full_fig_p025_11.png] view at source ↗
Figure 12
Figure 12. Figure 12: Additional Flow Matching visual results on Image Inpainting (different masks and different [PITH_FULL_IMAGE:figures/full_fig_p026_12.png] view at source ↗
Figure 13
Figure 13. Figure 13: Additional Flow Matching visual results on Image Inpainting (different masks and different [PITH_FULL_IMAGE:figures/full_fig_p027_13.png] view at source ↗

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 1 Pith paper

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. Residual Diffusion Bridge Model for Image Restoration

    cs.CV 2025-10 unverdicted novelty 6.0

    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

35 extracted references · 27 linked inside Pith · cited by 1 Pith paper

  1. [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....

  2. [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...

  3. [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,

  4. [7]

    Genpo: Generative Diffusion Models Meet On-Policy Reinforcement Learning.arXiv preprint arXiv:2505.18763,

    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,

  5. [8]

    Adjoint matching: Fine-tuning flow and diffusion generative models with memoryless stochastic optimal control

    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,

  6. [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,

  7. [11]

    Progressive Growing of GANs for Improved Quality, Stability, and Variation.arXiv preprint arXiv:1710.10196,

    Tero Karras. Progressive Growing of GANs for Improved Quality, Stability, and Variation.arXiv preprint arXiv:1710.10196,

  8. [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...

  9. [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,

  10. [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,

  11. [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,

  12. [18]

    Flow straight and fast: Learning to generate and transfer data with rectified flow.arXiv preprint arXiv:2209.03003,

    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,

  13. [19]

    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: 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 ...

  14. [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,

  15. [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,

  16. [25]

    Goal-conditioned Imitation Learning using Score-based Diffusion Policies.arXiv preprint arXiv:2304.02532,

    Moritz Reuss, Maximilian Li, Xiaogang Jia, and Rudolf Lioutikov. Goal-conditioned Imitation Learning using Score-based Diffusion Policies.arXiv preprint arXiv:2304.02532,

  17. [26]

    RB-Modulation: Training-Free Personalization of Diffusion Models using Stochastic Optimal Control.arXiv preprint arXiv:2405.17401,

    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,

  18. [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...

  19. [28]

    Improving and generalizing flow-based generative models with minibatch optimal transport.arXiv preprint arXiv:2302.00482,

    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,

  20. [29]

    Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learning.arXiv preprint arXiv:2208.06193,

    Zhendong Wang, Jonathan J Hunt, and Mingyuan Zhou. Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learning.arXiv preprint arXiv:2208.06193,

  21. [31]

    Reflected Flow Matching

    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...

  22. [32]

    Denoising Diffusion Bridge Models

    Linqi Zhou, Aaron Lou, Samar Khanna, and Stefano Ermon. Denoising Diffusion Bridge Models. arXiv preprint arXiv:2309.16948,

  23. [33]

    UniDB: A Uni- fied Diffusion Bridge Framework via Stochastic Optimal Control.arXiv preprint arXiv:2502.05749,

    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,

  24. [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...

  25. [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,

  26. [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,

  27. [2004]

    AffordDP: Generalizable Diffusion Policy with Transferable Affordance.arXiv preprint arXiv:2412.03142,

    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,

  28. [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,

  29. [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,

  30. [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...

  31. [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,

  32. [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,

  33. [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,

  34. [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,

  35. [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,