Entropy Across the Bridge: Conditional-Marginal Discretization for Flow and Schr"odinger Samplers
Not on the map. Why?
Abstract (arXiv)
For a fixed flow-based generative model under a small inference budget, sample quality can depend strongly on where the sampler spends its few function evaluations. Flow matching and Schr"odinger bridges define probability paths, yet their inference grids are usually heuristic or inherited from one-endpoint diffusion. We derive a conditional-marginal entropy-rate objective for bridge-aware discretization, separating endpoint-conditioned bridge geometry from marginal flow evolution, and use it to build a training-free entropic inference-time scheduler from first principles. For Gaussian Brownian bridges this rate is closed-form and U-shaped, motivating boundary-heavy nonuniform grids. On trained two-dimensional bridge/flow models, the estimated profile recovers the predicted shape and improves 10-step ODE-Heun MMD over linear by 18.1%, with a paired 22.7% SDE-Heun improvement in the same low-NFE sweep. On EDM/CIFAR-10, the entropic time-discretization gives the best tested five-step FID (186.3 ± 4.0 versus 200.5 ± 2.9 for linear and 238.0 ± 5.3 for cosine). On AlphaFlow protein generation, entropic conditional-marginal (cond-marg) scheduling shows advantage in low-NFE regimes on both CAMEO22 and ATLAS benchmarks. These results support entropy-rate scheduling as a practical low-budget allocation signal for high-dimensional bridge and flow samplers.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 42 of its 55 references that are in the map. First posted on arXiv 15 May 2026.
Builds on
42 of 55 references.
References · 42
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,761
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 3,045
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,316
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Stepscited by 1,979
- 2021Variational Diffusion Modelscited by 1,153
- 2022Building Normalizing Flows with Stochastic Interpolantscited by 995
- 2022DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Modelscited by 938
30 more
- 2023Improving and generalizing flow-based generative models with minibatch optimal transportcited by 879
- 2016InfoGAN: Interpretable Representation Learning by Information Maximizing Generative Adversarial Netscited by 860
- 2018MINE: Mutual Information Neural Estimationcited by 683
- 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modelingcited by 602
- 2022Fast Sampling of Diffusion Models with Exponential Integratorcited by 532
- 2023UniPC: A Unified Predictor-Corrector Framework for Fast Sampling of Diffusion Modelscited by 450
- 2023Common Diffusion Noise Schedules and Sample Steps are Flawedcited by 310
- 2013A survey of the Schr"odinger problem and some of its connections with optimal transportcited by 292
- 2023I2SB: Image-to-Image Schr"odinger Bridgecited by 218
- 2023On the Importance of Noise Scheduling for Diffusion Modelscited by 194
- 2024AlphaFold Meets Flow Matching for Generating Protein Ensemblescited by 110
- 2024FlowMM: Generating Materials with Riemannian Flow Matchingcited by 109
- 2024Align Your Steps: Optimizing Sampling Schedules in Diffusion Modelscited by 96
- 2023Simulation-free Schr"odinger bridges via score and flow matchingcited by 83
- 2024Accelerating Diffusion Sampling with Optimized Time Stepscited by 67
- 2025Proteina: Scaling Flow-based Protein Structure Generative Modelscited by 65
- 2024Wasserstein Flow Matching: Generative modeling over families of distributionscited by 28
- 2024Consistency Diffusion Bridge Modelscited by 23
- 2025Adjoint Schr"odinger Bridge Samplercited by 21
- 2025Joint Distillation for Fast Likelihood Evaluation and Sampling in Flow-based Modelscited by 18
- 2025Entropic Time Schedulers for Generative Diffusion Modelscited by 16
- 2025Inverse Bridge Matching Distillationcited by 14
- 2025Optimal Scheduling of Dynamic Transportcited by 13
- 2025The Information Dynamics of Generative Diffusioncited by 13
- 2025B'ezierFlow: Learning B'ezier Stochastic Interpolant Schedulers for Few-Step Generationcited by 7
- 2025Data-to-Energy Stochastic Dynamicscited by 6
- 2024Improving Inverse Folding for Peptide Design with Diversity-regularized Direct Preference Optimizationcited by 5
- 2025Categorical Schr"odinger Bridge Matchingcited by 5
- 2025STORK: Faster Diffusion And Flow Matching Sampling By Resolving Both Stiffness And Structure-Dependencecited by 5
- 2025Branched Schr"odinger Bridge Matchingcited by 5
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Foundations of Schr"odinger Bridges for Generative Modeling
- 2026Bridge Graphical Models: Coupling, Projection, and Current-Preserving Dynamics for Generative Modeling
- 2024Reflected Schr"odinger Bridge for Constrained Generative Modeling
- 2025Entering the Era of Discrete Diffusion Models: A Benchmark for Schr"odinger Bridges and Entropic Optimal Transport
- 2026Flow Sampling: Learning to Sample from Unnormalized Densities via Denoising Conditional Processes
- 2023The probability flow ODE is provably fast
- 2026Efficient Generative Modeling beyond Memoryless Diffusion via Adjoint Schr"odinger Bridge Matching
- 2026A Unified Kullback--Leibler Divergence Analysis of Generative Diffusion Models via Entropy Production Rate