Finsler Flow Matching: Dynamics-Aware Geodesic Interpolation for Single-Snapshot Trajectory Inference
cs.AI
Submitted: 2026-10-08
Updated: 2026-10-08
Code: https://github.com/mlbelab/FinslerFlowMatching
License: http://creativecommons.org/licenses/by/4.0/
The gist: Single-cell snapshot data can resolve a continuum of cellular states but do not uniquely determine the dynamics governing transitions between them.
Terminology
Abstract
Single-cell snapshot data can resolve a continuum of cellular states but do not uniquely determine the dynamics governing transitions between them. However, additional dynamical information can often be encoded in a cell-cell Markov transition kernel. Existing generative approaches for single cell trajectory inference either infer transport only from population marginals, impose a symmetric geometry on the state space, or incorporate directionality through a single velocity vector at each observed state. We introduce Finsler Flow Matching (FFM), a framework for learning continuous stochastic dynamics from discrete Markov transition graphs. We use the first and second local moments to construct a Finsler structure motivated by the Freidlin--Wentzell action, where the second moment determines anisotropic accessibility and the first moment introduces a preferred direction of motion. We learn neural approximations of the resulting directed geodesics, use their Finsler cost to construct source-target couplings, and define geometry-aware stochastic conditional paths that can be distilled into a continuous generative process through simulation-free score and flow matching. Across synthetic and single-cell trajectory inference benchmarks, FFM improves recovery of withheld intermediate populations, particularly when the transition dynamics are strongly directional or anisotropic. Our results provide a principled route from discrete transition probabilities to continuous generative dynamics while retaining both directional and diffusive structure.
Sources
- Building Normalizing Flows with Stochastic Interpolants
- A Locally Adaptive Normal Distribution
- Local Kernels and the Geometric Structure of Data
- Wasserstein Flow Matching: Generative modeling over families of distributions
- A survey of the Schr\"odinger problem and some of its connections with optimal transport
- Curly Flow Matching for Learning Non-gradient Field Dynamics
- An Analysis of the Convergence of Graph Laplacians
- Improving and generalizing flow-based generative models with minibatch optimal transport
- Learning stochastic dynamics from snapshots through regularized unbalanced optimal transport
- Learning Lineage-guided Geodesics with Finsler Geometry
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems