TrajectoryNet: A Dynamic Optimal Transport Network for Modeling Cellular Dynamics
Not on the map. Why?
Cited by 32 in Diffusion models, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
It is increasingly common to encounter data from dynamic processes captured by static cross-sectional measurements over time, particularly in biomedical settings. Recent attempts to model individual trajectories from this data use optimal transport to create pairwise matchings between time points. However, these methods cannot model continuous dynamics and non-linear paths that entities can take in these systems. To address this issue, we establish a link between continuous normalizing flows and dynamic optimal transport, that allows us to model the expected paths of points over time. Continuous normalizing flows are generally under constrained, as they are allowed to take an arbitrary path from the source to the target distribution. We present TrajectoryNet, which controls the continuous paths taken between distributions to produce dynamic optimal transport. We show how this is particularly applicable for studying cellular dynamics in data from single-cell RNA sequencing (scRNA-seq) technologies, and that TrajectoryNet improves upon recently proposed static optimal transport-based models that can be used for interpolating cellular distributions.
Key facts. Cited by 155 papers in the map (data through 7 Oct 2026). First posted on arXiv 9 Feb 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 5; 2022: 13; 2023: 28; 2024: 24; 2025: 41; 2026: 44.
155 papers.
Sort:
By area:
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
- 2023Improving and generalizing flow-based generative models with minibatch optimal transportcited by 835
- 2023Multisample Flow Matching: Straightening Flows with Minibatch Couplingscited by 267
- 2023SE(3)-Stochastic Flow Matching for Protein Backbone Generationcited by 147
- 2021Solving Schr"odinger Bridges via Maximum Likelihoodcited by 140
- 2023Equivariant flow matchingcited by 131
- 2022Action Matching: Learning Stochastic Dynamics from Samplescited by 94
- 2021Proximal Optimal Transport Modeling of Population Dynamicscited by 91
- 2023Simulation-free Schr"odinger bridges via score and flow matchingcited by 81
- 2024Metric Flow Matching for Smooth Interpolations on the Data Manifoldcited by 74
- 2022Neural Conservation Laws: A Divergence-Free Perspectivecited by 67
38 more
- 2021Low-Rank Sinkhorn Factorizationcited by 66
- 2022The Schr"odinger Bridge between Gaussian Measures has a Closed Formcited by 63
- 2022Manifold Interpolating Optimal-Transport Flows for Trajectory Inferencecited by 63
- 2023A Computational Framework for Solving Wasserstein Lagrangian Flowscited by 43
- 2024Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifoldcited by 41
- 2023Deep Momentum Multi-Marginal Schr"odinger Bridgecited by 40
- 2023Aligned Diffusion Schr"odinger Bridgescited by 38
- 2023Unbalancedness in Neural Monge Maps Improves Unpaired Domain Translationcited by 32
- 2021Towards a mathematical theory of trajectory inferencecited by 31
- 2024Learning stochastic dynamics from snapshots through regularized unbalanced optimal transportcited by 30
- 2022Trajectory Inference via Mean-field Langevin in Path Spacecited by 29
- 2023DynGFN: Towards Bayesian Inference of Gene Regulatory Networks with GFlowNetscited by 27
- 2023GENOT: Entropic (Gromov) Wasserstein Flow Matching with Applications to Single-Cell Genomicscited by 27
- 2024Multi-marginal Schr"odinger Bridges with Iterative Reference Refinementcited by 26
- 2024Dynamic Conditional Optimal Transport through Simulation-Free Flowscited by 24
- 2022Riemannian Metric Learning via Optimal Transportcited by 21
- 2024Learning diffusion at lightspeedcited by 21
- 2025Joint Velocity-Growth Flow Matching for Single-Cell Dynamics Modelingcited by 20
- 2024ImageFlowNet: Forecasting Multiscale Image-Level Trajectories of Disease Progression with Irregularly-Sampled Longitudinal Medical Imagescited by 19
- 2021Diffusion Earth Mover's Distance and Distribution Embeddingscited by 18
- 2023Light Schr"odinger Bridgecited by 18
- 2025Curly Flow Matching for Learning Non-gradient Field Dynamicscited by 17
- 2022Neural Unbalanced Optimal Transport via Cycle-Consistent Semi-Couplingscited by 16
- 2024Neural Optimal Transport with Lagrangian Costscited by 16
- 2024Partially Observed Trajectory Inference using Optimal Transport and a Dynamics Priorcited by 13
- 2024Geometry-Aware Generative Autoencoders for Warped Riemannian Metric Learning and Generative Modeling on Data Manifoldscited by 13
- 2025Modeling Cell Dynamics and Interactions with Unbalanced Mean Field Schr"odinger Bridgecited by 12
- 2023Flow Factorized Representation Learningcited by 11
- 2025Multi-Marginal Stochastic Flow Matching for High-Dimensional Snapshot Data at Irregular Time Pointscited by 11
- 2022Geodesic Sinkhorn for Fast and Accurate Optimal Transport on Manifoldscited by 10
- 2025Trajectory Inference with Smooth Schr"odinger Bridgescited by 10
- 2025Variational Regularized Unbalanced Optimal Transport: Single Network, Least Actioncited by 10
- 2023Transport with Support: Data-Conditional Diffusion Bridgescited by 9
- 2023Inferring dynamic regulatory interaction graphs from time series data with perturbationscited by 9
- 2023Unbalanced Diffusion Schr"odinger Bridgecited by 9
- 2024Generalized Sobolev Transport for Probability Measures on a Graphcited by 9
- 2024Efficient Trajectory Inference in Wasserstein Space Using Consecutive Averagingcited by 9
- 2025Integrating Dynamical Systems Modeling with Spatiotemporal scRNA-seq Data Analysiscited by 9
50 most cited of 155 All 155 (JSON)
- 2026Learning Continuous Patient Trajectories from Electronic Health Records28 Sep 2026
- 2026Beyond Gradient Flow: Identifiability and Recovery from Distribution Snapshots28 Sep 2026
- 2026Learning Collective Dynamics with Differentiable Gaussian Representations23 Sep 2026
- 2026Dynamic Generalized Gromov-Wasserstein Optimal Transport17 Sep 2026
- 2026Simulation-free Unbalanced Dynamic Optimal Transport with General Growth Penalty4 Sep 2026
- 2026TracingFlow: A Simulation-Free Trajectory Inference Framework Based on Second-Order Dynamics21 Aug 2026
- 2026Pathology Transport: Optimal-Transport Explanations for Clinical Data, and When Their Heatmaps (Fail to) Localize Disease18 Aug 2026
- 2026A Unified Geometric Framework for Developmental Analysis of Spatial Transcriptomic Data15 Aug 2026
- 2026Trajectory inference via Acceleration Matching4 Aug 2026
- 2026Twisted Schr"odinger Bridge Matching18 Jul 2026
- 2026Wasserstein Residuals: Learning Gradient Flows from Population Dynamics6 Jul 2026
- 2026Topological Flow Matching14 Jun 2026
38 more
- 2026First-Order Trajectory Matching: Fast Ensemble Predictions of Chaotic, Turbulent, Stochastic Systems9 Jun 2026
- 2026Multimarginal flow matching with optimal transport potentials3 Jun 2026
- 2026Nonlocal Mean Field Schr"odinger Bridge with Learned Interactions2 Jun 2026
- 2026A Biconvex Formulation for Stable Transport of Mixture Models with a Unique Solution1 Jun 2026
- 2026CellBRIDGE: Learning Cellular Trajectories via Interaction-Aware Alignment28 May 2026
- 2026Chreode: A Cell World Model for One-Step Temporal Dynamics and Perturbation Prediction27 May 2026
- 2026Two-Parameter Flows for Learning Population Dynamics of Physical Systems25 May 2026
- 2026From Snapshots to Trajectories: Learning Single-Cell Gene Expression Dynamics via Conditional Flow Matching21 May 2026
- 2026Modeling Temporal scRNA-seq Data with Latent Gaussian Process and Optimal Transport20 May 2026
- 2026PACE: Geometry-Aware Bridge Transport for Single-Cell Trajectory Inference18 May 2026
- 2026Domain Transfer Becomes Identifiable via a Single Alignment18 May 2026
- 2026Multiscale Supervised Unbalanced Optimal Transport Flow Matching15 May 2026
- 2026scShapeBench: Discovering geometry from high dimensional scRNAseq data12 May 2026
- 2026QDSB: Quantized Diffusion Schr"odinger Bridges12 May 2026
- 2026Implicit Neural Optimal Transport via Fixed-Point Optimization11 May 2026
- 2026FLUX: Geometry-Aware Longitudinal Flow Matching with Mixture of Experts9 May 2026
- 2026A Call to Lagrangian Action: Learning Population Mechanics from Temporal Snapshots8 May 2026
- 2026Beyond Continuity: Simulation-free Reconstruction of Discrete Branching Dynamics from Single-cell Snapshots1 May 2026
- 2026Learning biophysical models of gene regulation with probability flow matching27 Apr 2026
- 2026Continuum-marginal optimal transport: a mesh-free kernel method27 Apr 2026
- 2026Relative Entropy Estimation in Function Space: Theory and Applications to Trajectory Inference22 Apr 2026
- 2026Uncertainty-quantified Pulse Signal Recovery from Facial Video using Regularized Stochastic Interpolants12 Apr 2026
- 2026On the Unique Recovery of Transport Maps and Vector Fields from Finite Measure-Valued Data9 Apr 2026
- 2026Flow matching on homogeneous spaces25 Mar 2026
- 2026MIOFlow 2.0: A unified framework for inferring cellular stochastic dynamics from single cell and spatial transcriptomics data23 Mar 2026
- 2026Learning Lineage-guided Geodesics with Finsler Geometry17 Mar 2026
- 2026COT-FM: Cluster-wise Optimal Transport Flow Matching11 Mar 2026
- 2026Hyperparameter Trajectory Inference with Conditional Lagrangian Optimal Transport2 Mar 2026
- 2026STRAND: Sequence-Conditioned Transport for Single-Cell Perturbations10 Feb 2026
- 2026Schr"odinger bridge problem via empirical risk minimization9 Feb 2026
- 2026WFR-MFM: One-Step Inference for Dynamic Unbalanced Optimal Transport28 Jan 2026
- 2026WFR-FM: Simulation-Free Dynamic Unbalanced Optimal Transport11 Jan 2026
- 2025Optimal and Diffusion Transports in Machine Learning7 Dec 2025
- 2025CellStream: Dynamical Optimal Transport Informed Embeddings for Reconstructing Cellular Trajectories from Snapshots Data16 Nov 2025
- 2025Contact Wasserstein Geodesics for Non-Conservative Schr"odinger Bridges10 Nov 2025
- 2025Modeling Microenvironment Trajectories on Spatial Transcriptomics with NicheFlow2 Nov 2025
- 2025Curly Flow Matching for Learning Non-gradient Field Dynamics30 Oct 2025
- 2025Simulation-free Structure Learning for Stochastic Population Dynamics18 Oct 2025
50 newest of 155 All 155 (JSON)
- 2023Improving and generalizing flow-based generative models with minibatch optimal transportcited by 835
- 2023Multisample Flow Matching: Straightening Flows with Minibatch Couplingscited by 267
- 2023SE(3)-Stochastic Flow Matching for Protein Backbone Generationcited by 147
- 2023Equivariant flow matchingcited by 131
- 2023Simulation-free Schr"odinger bridges via score and flow matchingcited by 81
- 2024Metric Flow Matching for Smooth Interpolations on the Data Manifoldcited by 74
- 2024Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifoldcited by 41
- 2023GENOT: Entropic (Gromov) Wasserstein Flow Matching with Applications to Single-Cell Genomicscited by 27
- 2025Joint Velocity-Growth Flow Matching for Single-Cell Dynamics Modelingcited by 20
- 2025Curly Flow Matching for Learning Non-gradient Field Dynamicscited by 17
- 2025Multi-Marginal Stochastic Flow Matching for High-Dimensional Snapshot Data at Irregular Time Pointscited by 11
- 2024Pullback Flow Matching on Data Manifoldscited by 8
12 most cited of 31 All 31 (JSON)
- 2026Learning Continuous Patient Trajectories from Electronic Health Records28 Sep 2026
- 2026Topological Flow Matching14 Jun 2026
- 2026Multimarginal flow matching with optimal transport potentials3 Jun 2026
- 2026From Snapshots to Trajectories: Learning Single-Cell Gene Expression Dynamics via Conditional Flow Matching21 May 2026
- 2026Multiscale Supervised Unbalanced Optimal Transport Flow Matching15 May 2026
- 2026FLUX: Geometry-Aware Longitudinal Flow Matching with Mixture of Experts9 May 2026
- 2026Learning biophysical models of gene regulation with probability flow matching27 Apr 2026
- 2026COT-FM: Cluster-wise Optimal Transport Flow Matching11 Mar 2026
- 2026WFR-MFM: One-Step Inference for Dynamic Unbalanced Optimal Transport28 Jan 2026
- 2026WFR-FM: Simulation-Free Dynamic Unbalanced Optimal Transport11 Jan 2026
- 2025Curly Flow Matching for Learning Non-gradient Field Dynamics30 Oct 2025
- 2025Multi-Marginal Flow Matching with Adversarially Learnt Interpolants1 Oct 2025
12 newest of 31 All 31 (JSON)
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2026Chreode: A Cell World Model for One-Step Temporal Dynamics and Perturbation Predictioncited by 0
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
- 2021Solving Schr"odinger Bridges via Maximum Likelihoodcited by 140
- 2022Action Matching: Learning Stochastic Dynamics from Samplescited by 94
- 2021Proximal Optimal Transport Modeling of Population Dynamicscited by 91
- 2022Neural Conservation Laws: A Divergence-Free Perspectivecited by 67
- 2021Low-Rank Sinkhorn Factorizationcited by 66
- 2022The Schr"odinger Bridge between Gaussian Measures has a Closed Formcited by 63
- 2022Manifold Interpolating Optimal-Transport Flows for Trajectory Inferencecited by 63
- 2023A Computational Framework for Solving Wasserstein Lagrangian Flowscited by 43
- 2023Deep Momentum Multi-Marginal Schr"odinger Bridgecited by 40
- 2023Aligned Diffusion Schr"odinger Bridgescited by 38
- 2023Unbalancedness in Neural Monge Maps Improves Unpaired Domain Translationcited by 32
12 most cited of 122 All 122 (JSON)
- 2026Beyond Gradient Flow: Identifiability and Recovery from Distribution Snapshots28 Sep 2026
- 2026Learning Collective Dynamics with Differentiable Gaussian Representations23 Sep 2026
- 2026Dynamic Generalized Gromov-Wasserstein Optimal Transport17 Sep 2026
- 2026Simulation-free Unbalanced Dynamic Optimal Transport with General Growth Penalty4 Sep 2026
- 2026TracingFlow: A Simulation-Free Trajectory Inference Framework Based on Second-Order Dynamics21 Aug 2026
- 2026Pathology Transport: Optimal-Transport Explanations for Clinical Data, and When Their Heatmaps (Fail to) Localize Disease18 Aug 2026
- 2026A Unified Geometric Framework for Developmental Analysis of Spatial Transcriptomic Data15 Aug 2026
- 2026Trajectory inference via Acceleration Matching4 Aug 2026
- 2026Twisted Schr"odinger Bridge Matching18 Jul 2026
- 2026Wasserstein Residuals: Learning Gradient Flows from Population Dynamics6 Jul 2026
- 2026First-Order Trajectory Matching: Fast Ensemble Predictions of Chaotic, Turbulent, Stochastic Systems9 Jun 2026
- 2026Nonlocal Mean Field Schr"odinger Bridge with Learned Interactions2 Jun 2026
12 newest of 122 All 122 (JSON)