Paper Reading AI Learner

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

2025-09-29 09:45:22
Kaizhen Zhu, Mokai Pan, Zhechuan Yu, Jingya Wang, Jingyi Yu, Ye Shi

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 Inpainting, Super-Resolution, Deblurring, Denoising, 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 this https URL.

Abstract (translated)

扩散桥(Diffusion Bridge)和流匹配(Flow Matching)在任意分布之间的转换中都展示了令人信服的实证性能。然而,关于哪一种方法更优的选择仍然存在疑惑,并且这两种模型之间巨大的建模假设差异及实际实现上的不同也阻碍了对它们相对优势进行统一理论阐述的可能性。我们首次提供了这两种模型统合理论和实验验证的方法。我们将这两个框架重新构想为随机最优控制的视角,并证明扩散桥的成本函数较低,这有助于系统趋向更稳定且自然的轨迹变化。同时,从最优传输的角度来看,在训练数据量减少的情况下,流匹配中的插值系数 $t$ 和 $1-t$ 逐渐变得无效。 为了验证这些理论主张,我们提出了一种新颖而强大的基于潜在Transformer架构的扩散桥模型,并实现了一个结构相同的流匹配模型以在各种实验中进行公平性能比较。我们在图像修复、超分辨率、去模糊化、去噪、转换以及风格迁移等任务上进行了全面实验,系统地改变分布差异(不同难度)和训练数据量。广泛的实证结果与我们的理论预测完全一致,并使我们能够明确指出这两种模型各自的优缺点。 有关该工作的代码可以在提供的链接中获取。

URL

https://arxiv.org/abs/2509.24531

PDF

https://arxiv.org/pdf/2509.24531.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot