Stochastic interpolants with data-dependent couplings
Not on the map. Why?
Cited by 66 in Diffusion models, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
Generative models inspired by dynamical transport of measure -- such as flows and diffusions -- construct a continuous-time map between two probability densities. Conventionally, one of these is the target density, only accessible through samples, while the other is taken as a simple base density that is data-agnostic. In this work, using the framework of stochastic interpolants, we formalize how to textitcouple the base and the target densities, whereby samples from the base are computed conditionally given samples from the target in a way that is different from (but does preclude) incorporating information about class labels or continuous embeddings. This enables us to construct dynamical transport maps that serve as conditional generative models. We show that these transport maps can be learned by solving a simple square loss regression problem analogous to the standard independent setting. We demonstrate the usefulness of constructing dependent couplings in practice through experiments in super-resolution and in-painting.
Key facts. Cited by 100 papers in the map (data through 14 Oct 2026). Builds on 36 of its 47 references that are in the map. First posted on arXiv 5 Oct 2023.
Builds on
36 of 47 references.
References · 36
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Classifier-Free Diffusion Guidancecited by 5,847
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,761
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,316
- 2022Video Diffusion Modelscited by 2,305
- 2015Variational Inference with Normalizing Flowscited by 2,170
24 more
- 2016Density estimation using Real NVPcited by 2,042
- 2021Image Super-Resolution via Iterative Refinementcited by 1,511
- 2021Cascaded Diffusion Models for High Fidelity Image Generationcited by 1,163
- 2020Improved Techniques for Training Score-Based Generative Modelscited by 1,043
- 2022Building Normalizing Flows with Stochastic Interpolantscited by 995
- 2016Deep Networks with Stochastic Depthcited by 979
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 939
- 2023Improving and generalizing flow-based generative models with minibatch optimal transportcited by 879
- 2021Maximum Likelihood Training of Score-Based Diffusion Modelscited by 672
- 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modelingcited by 602
- 2019Neural Spline Flowscited by 596
- 2023InstaFlow: One Step is Enough for High-Quality Diffusion-Based Text-to-Image Generationcited by 395
- 2022Rectified Flow: A Marginal Preserving Approach to Optimal Transportcited by 283
- 2023Multisample Flow Matching: Straightening Flows with Minibatch Couplingscited by 278
- 2022Cold Diffusion: Inverting Arbitrary Image Transforms Without Noisecited by 276
- 2022Diffusion probabilistic modeling of protein backbones in 3D for the motif-scaffolding problemcited by 236
- 2023Practical and Asymptotically Exact Conditional Sampling in Diffusion Modelscited by 232
- 2022Structure-based Drug Design with Equivariant Diffusion Modelscited by 217
- 2023Flow Matching in Latent Spacecited by 160
- 2023Equivariant flow matchingcited by 135
- 2022Soft Diffusion: Score Matching for General Corruptionscited by 102
- 2023Minimizing Trajectory Curvature of ODE-based Generative Modelscited by 89
- 2022Conditional Simulation Using Diffusion Schr"odinger Bridgescited by 60
- 2023Where to Diffuse, How to Diffuse, and How to Get Back: Automated Learning for Multivariate Diffusionscited by 29
Built on by
Citing papers per year: 2023: 3; 2024: 20; 2025: 40; 2026: 37.
100 papers.
Sort:
By area:
- 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 737
- 2025PaperBench: Evaluating AI's Ability to Replicate AI Researchcited by 321
- 2024Flow map matching with stochastic interpolants: A mathematical framework for consistency modelscited by 109
- 2024DepthFM: Fast Monocular Depth Estimation with Flow Matchingcited by 76
- 2025Any-Order Flexible Length Masked Diffusioncited by 51
- 2023Boosting Latent Diffusion with Flow Matchingcited by 44
- 2024Improved motif-scaffolding with SE(3) flow matchingcited by 43
- 2024Flow Matching with Gaussian Process Priors for Probabilistic Time Series Forecastingcited by 43
- 2026Meta Flow Maps enable scalable reward alignmentcited by 37
- 2024Conditional Wasserstein Distances with Applications in Bayesian OT Flow Matchingcited by 36
- 2025Much Ado About Noising: Dispelling the Myths of Generative Robotic Controlcited by 36
- 2024Posterior-Mean Rectified Flow: Towards Minimum MSE Photo-Realistic Image Restorationcited by 34
38 more
- 2024Flow matching achieves almost minimax optimal convergencecited by 29
- 2025Open Materials Generation with Stochastic Interpolantscited by 28
- 2024Flowing from Words to Pixels: A Noise-Free Framework for Cross-Modality Evolutioncited by 26
- 2024Dynamic Conditional Optimal Transport through Simulation-Free Flowscited by 25
- 2025Thermalizer: Stable autoregressive neural emulation of spatiotemporal chaoscited by 21
- 2024FrameBridge: Improving Image-to-Video Generation with Bridge Modelscited by 19
- 2023Gaussian Interpolation Flowscited by 17
- 2024EquiJump: Protein Dynamics Simulation via SO(3)-Equivariant Stochastic Interpolantscited by 16
- 2025VCT: Training Consistency Models with Variational Noise Couplingcited by 15
- 2025Flow Matching for Efficient and Scalable Data Assimilationcited by 12
- 2025All that structure matches does not glittercited by 12
- 2024Extended Flow Matching: a Method of Conditional Generation with Generalized Continuity Equationcited by 11
- 2024Conditional simulation via entropic optimal transport: Toward non-parametric estimation of conditional Brenier mapscited by 9
- 2025One-Step Offline Distillation of Diffusion-based Models via Koopman Modelingcited by 9
- 2025STFlow: Data-Coupled Flow Matching for Geometric Trajectory Simulationcited by 8
- 2025Learning Straight Flows: Variational Flow Matching for Efficient Generationcited by 8
- 2025LADB: Latent Aligned Diffusion Bridges for Semi-Supervised Domain Translationcited by 7
- 2025Stochastic Interpolants via Conditional Dependent Couplingcited by 7
- 2025Generative Modeling from Black-box Corruptions via Self-Consistent Stochastic Interpolantscited by 6
- 2025Boltzmann generators for amorphous particle systemscited by 6
- 2026Flow-DPPO: Divergence Proximal Policy Optimization for Flow Matching Modelscited by 6
- 2024Stochastic Flow Matching for Resolving Small-Scale Physicscited by 5
- 2025Good Things Come in Pairs: Paired Autoencoders for Inverse Problemscited by 5
- 2026Scalable Maximum Entropy Reinforcement Learning for Diffusion Policies via Adjoint Matchingcited by 5
- 2025Finite-Time Analysis of Discrete-Time Stochastic Interpolantscited by 4
- 2025In-Context Parametric Inference: Point or Distribution Estimators?cited by 4
- 2025Topological Schr"odinger Bridge Matchingcited by 4
- 2026Prior-Informed Flow Matching for Graph Reconstructioncited by 4
- 2026Cascaded Flow Matching for Heterogeneous Tabular Data with Mixed-Type Featurescited by 4
- 2024Exploring the Design Space of Diffusion Bridge Modelscited by 3
- 2024Diverse capability and scaling of diffusion and auto-regressive models when learning abstract rulescited by 3
- 2025On the Contractivity of Stochastic Interpolation Flowcited by 3
- 2025HazeMatching: Dehazing Light Microscopy Images with Guided Conditional Flow Matchingcited by 3
- 2025CoVAE: Consistency Training of Variational Autoencoderscited by 3
- 2025Multitask Learning with Stochastic Interpolantscited by 3
- 2025Distilling LLM Prior to Flow Model for Generalizable Agent's Imagination in Object Goal Navigationcited by 3
- 2025Incomplete Data, Complete Dynamics: A Diffusion Approachcited by 3
- 2025Split-Flows: Measure Transport and Information Loss Across Molecular Resolutionscited by 3
50 most cited of 100 All 100 (JSON)
- 2026Guidance Without Labels: Flow Matching from Coupled Gaussian Mixtures12 Oct 2026
- 2026Probabilistic Simulation of Gas-Liquid Interfaces with Coupled Flow Matching on Spherical Harmonics12 Oct 2026
- 2026Smoother Flow Matching via Contrastive Trajectory Repulsion1 Oct 2026
- 2026Graph-Spectral Flow Matching for Multivariate Time Series Anomaly Detection29 Sep 2026
- 2026What Does a Stream Model Buy You in Flow Matching?28 Sep 2026
- 2026SNaP: One-Step Posterior Sampling for Noisy Inverse Problems28 Sep 2026
- 2026BridgeCast: Bridging Ocean Wave Forecasts to Reanalysis via Flow Matching with Exogenous Variables27 Sep 2026
- 2026Posterior Flow Matching26 Sep 2026
- 2026Feasible Flow Matching for Graph Reconstruction via Within-Sampling Primal-Dual Guidance26 Sep 2026
- 2026Efficient Analog-Initialized Latent Transport for Spatially Coherent Probabilistic Downscaling from Global to Kilometer Scales15 Sep 2026
- 2026Fast and Faithful: Principled Conditional Flow Matching for Inverse Problems11 Sep 2026
- 2026There and Back Again: Bidirectional Diffusion Bridges for Multimodality Translation28 Aug 2026
38 more
- 2026Trajectory inference via Acceleration Matching4 Aug 2026
- 2026One-Sided Quantile Coupling for Flow Matching2 Aug 2026
- 2026Temporal Policy: History-Initialized Action Generation for Robotic Learning from Demonstration31 Jul 2026
- 2026STEREOFLOW: Progressive Stereo Matching with StereoDiT and Transition Flow Matching22 Jul 2026
- 2026BridgeFlow: Fast and Robust SE(2)-Equivariant Motion Planning with Flow Matching16 Jul 2026
- 2026A2BM: Alignment-Aware Bridge Matching for Image-to-Image Translation13 Jul 2026
- 2026TRIE: An Evaluation Framework for Stochastic PDE Surrogates30 Jun 2026
- 2026NullFlow: One-Step Generative Reconstruction21 Jun 2026
- 2026Scalable Maximum Entropy Reinforcement Learning for Diffusion Policies via Adjoint Matching21 Jun 2026
- 2026BrainG3N: A Dual-Purpose Tokenizer for Controllable 3D Brain MRI Generation17 Jun 2026
- 2026Flow-DPPO: Divergence Proximal Policy Optimization for Flow Matching Models9 Jun 2026
- 2026Spatiotemporal Imputation with Graph-Informed Flow Matching4 Jun 2026
- 2026A Biconvex Formulation for Stable Transport of Mixture Models with a Unique Solution1 Jun 2026
- 2026Universal Time Series Generation with Neural Controlled Differential Equations27 May 2026
- 2026Disentangling Generation and Regression in Stochastic Interpolants for Controllable Image Restoration20 May 2026
- 2026Drift Flow Matching17 May 2026
- 2026Structured Coupling for Flow Matching8 May 2026
- 2026Graph Energy Matching: Transport-Aligned Energy-Based Modeling for Graph Generation24 Mar 2026
- 2026Linearized Coupling Flow with Shortcut Constraints for One-Step Face Restoration4 Mar 2026
- 2026Transport, Don't Generate: Deterministic Geometric Flows for Combinatorial Optimization11 Feb 2026
- 2026Unlocking the Duality between Flow and Field Matching2 Feb 2026
- 2026Stochastic Interpolants in Hilbert Spaces2 Feb 2026
- 2026Cascaded Flow Matching for Heterogeneous Tabular Data with Mixed-Type Features30 Jan 2026
- 2026Prior-Informed Flow Matching for Graph Reconstruction29 Jan 2026
- 2026Meta Flow Maps enable scalable reward alignment20 Jan 2026
- 2025Boltzmann generators for amorphous particle systems18 Dec 2025
- 2025Generative Modeling from Black-box Corruptions via Self-Consistent Stochastic Interpolants11 Dec 2025
- 2025FlowEO: Generative Unsupervised Domain Adaptation for Earth Observation1 Dec 2025
- 2025Much Ado About Noising: Dispelling the Myths of Generative Robotic Control1 Dec 2025
- 2025Learning Straight Flows: Variational Flow Matching for Efficient Generation15 Nov 2025
- 2025Split-Flows: Measure Transport and Information Loss Across Molecular Resolutions3 Nov 2025
- 2025ResMatching: Noise-Resilient Computational Super-Resolution via Guided Conditional Flow Matching30 Oct 2025
- 2025Generative Point Tracking with Flow Matching23 Oct 2025
- 2025Towards General Modality Translation with Contrastive and Predictive Latent Diffusion Bridge23 Oct 2025
- 2025Diffusion Buffer for Online Generative Speech Enhancement21 Oct 2025
- 2025PAINT: Parallel-in-time Neural Twins for Dynamical System Reconstruction14 Oct 2025
- 2025Probabilistic Super-Resolution for Urban Micrometeorology via a Schr"odinger Bridge14 Oct 2025
- 2025Three Forms of Stochastic Injection for Improved Distribution-to-Distribution Generative Modeling8 Oct 2025
50 newest of 100 All 100 (JSON)
- 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 737
- 2024Flow map matching with stochastic interpolants: A mathematical framework for consistency modelscited by 109
- 2024DepthFM: Fast Monocular Depth Estimation with Flow Matchingcited by 76
- 2025Any-Order Flexible Length Masked Diffusioncited by 51
- 2023Boosting Latent Diffusion with Flow Matchingcited by 44
- 2024Flow Matching with Gaussian Process Priors for Probabilistic Time Series Forecastingcited by 43
- 2024Conditional Wasserstein Distances with Applications in Bayesian OT Flow Matchingcited by 36
- 2024Posterior-Mean Rectified Flow: Towards Minimum MSE Photo-Realistic Image Restorationcited by 34
- 2024Flow matching achieves almost minimax optimal convergencecited by 29
- 2025Open Materials Generation with Stochastic Interpolantscited by 28
- 2024Flowing from Words to Pixels: A Noise-Free Framework for Cross-Modality Evolutioncited by 26
- 2025Thermalizer: Stable autoregressive neural emulation of spatiotemporal chaoscited by 21
12 most cited of 66 All 66 (JSON)
- 2026Guidance Without Labels: Flow Matching from Coupled Gaussian Mixtures12 Oct 2026
- 2026Probabilistic Simulation of Gas-Liquid Interfaces with Coupled Flow Matching on Spherical Harmonics12 Oct 2026
- 2026Smoother Flow Matching via Contrastive Trajectory Repulsion1 Oct 2026
- 2026Graph-Spectral Flow Matching for Multivariate Time Series Anomaly Detection29 Sep 2026
- 2026What Does a Stream Model Buy You in Flow Matching?28 Sep 2026
- 2026SNaP: One-Step Posterior Sampling for Noisy Inverse Problems28 Sep 2026
- 2026BridgeCast: Bridging Ocean Wave Forecasts to Reanalysis via Flow Matching with Exogenous Variables27 Sep 2026
- 2026Posterior Flow Matching26 Sep 2026
- 2026Feasible Flow Matching for Graph Reconstruction via Within-Sampling Primal-Dual Guidance26 Sep 2026
- 2026Fast and Faithful: Principled Conditional Flow Matching for Inverse Problems11 Sep 2026
- 2026One-Sided Quantile Coupling for Flow Matching2 Aug 2026
- 2026Temporal Policy: History-Initialized Action Generation for Robotic Learning from Demonstration31 Jul 2026
12 newest of 66 All 66 (JSON)
- 2025PaperBench: Evaluating AI's Ability to Replicate AI Researchcited by 321
- 2024Improved motif-scaffolding with SE(3) flow matchingcited by 43
- 2026Meta Flow Maps enable scalable reward alignmentcited by 37
- 2025Much Ado About Noising: Dispelling the Myths of Generative Robotic Controlcited by 36
- 2024Dynamic Conditional Optimal Transport through Simulation-Free Flowscited by 25
- 2023Gaussian Interpolation Flowscited by 17
- 2024EquiJump: Protein Dynamics Simulation via SO(3)-Equivariant Stochastic Interpolantscited by 16
- 2025All that structure matches does not glittercited by 12
- 2024Conditional simulation via entropic optimal transport: Toward non-parametric estimation of conditional Brenier mapscited by 9
- 2025Stochastic Interpolants via Conditional Dependent Couplingcited by 7
- 2025Generative Modeling from Black-box Corruptions via Self-Consistent Stochastic Interpolantscited by 6
- 2025Boltzmann generators for amorphous particle systemscited by 6
12 most cited of 33 All 33 (JSON)
- 2026Efficient Analog-Initialized Latent Transport for Spatially Coherent Probabilistic Downscaling from Global to Kilometer Scales15 Sep 2026
- 2026There and Back Again: Bidirectional Diffusion Bridges for Multimodality Translation28 Aug 2026
- 2026Trajectory inference via Acceleration Matching4 Aug 2026
- 2026TRIE: An Evaluation Framework for Stochastic PDE Surrogates30 Jun 2026
- 2026NullFlow: One-Step Generative Reconstruction21 Jun 2026
- 2026A Biconvex Formulation for Stable Transport of Mixture Models with a Unique Solution1 Jun 2026
- 2026Universal Time Series Generation with Neural Controlled Differential Equations27 May 2026
- 2026Meta Flow Maps enable scalable reward alignment20 Jan 2026
- 2025Boltzmann generators for amorphous particle systems18 Dec 2025
- 2025Generative Modeling from Black-box Corruptions via Self-Consistent Stochastic Interpolants11 Dec 2025
- 2025Much Ado About Noising: Dispelling the Myths of Generative Robotic Control1 Dec 2025
- 2025Split-Flows: Measure Transport and Information Loss Across Molecular Resolutions3 Nov 2025
12 newest of 33 All 33 (JSON)