MIOFlow 2.0: A unified framework for inferring cellular stochastic dynamics from single cell and spatial transcriptomics data

• URL: http://arxiv.org/abs/2603.22564v1
• Date: Mon, 23 Mar 2026 20:49:45 GMT
• Title: MIOFlow 2.0: A unified framework for inferring cellular stochastic dynamics from single cell and spatial transcriptomics data
• Abstract summary: We introduce Interpolating Manifold-Transport Flow (MIOFlow) 2.0.<n>This framework learns biologically informed cellular trajectories by integrating manifold learning, optimal transport, and neural differential equations.<n>It is validated on synthetic datasets, embryoid body differentiation, and spatially resolved axolotl brain regeneration.
• Score: 47.97758608694957
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: Understanding cellular trajectories via time-resolved single-cell transcriptomics is vital for studying development, regeneration, and disease. A key challenge is inferring continuous trajectories from discrete snapshots. Biological complexity stems from stochastic cell fate decisions, temporal proliferation changes, and spatial environmental influences. Current methods often use deterministic interpolations treating cells in isolation, failing to capture the probabilistic branching, population shifts, and niche-dependent signaling driving real biological processes. We introduce Manifold Interpolating Optimal-Transport Flow (MIOFlow) 2.0. This framework learns biologically informed cellular trajectories by integrating manifold learning, optimal transport, and neural differential equations. It models three core processes: (1) stochasticity and branching via Neural Stochastic Differential Equations; (2) non-conservative population changes using a learned growth-rate model initialized with unbalanced optimal transport; and (3) environmental influence through a joint latent space unifying gene expression with spatial features like local cell type composition and signaling. By operating in a PHATE-distance matching autoencoder latent space, MIOFlow 2.0 ensures trajectories respect the data's intrinsic geometry. Empirical comparisons show expressive trajectory learning via neural differential equations outperforms existing generative models, including simulation-free flow matching. Validated on synthetic datasets, embryoid body differentiation, and spatially resolved axolotl brain regeneration, MIOFlow 2.0 improves trajectory accuracy and reveals hidden drivers of cellular transitions, like specific signaling niches. MIOFlow 2.0 thus bridges single-cell and spatial transcriptomics to uncover tissue-scale trajectories.

Related papers

• TracingFlow: A Simulation-Free Trajectory Inference Framework Based on Second-Order Dynamics [7.088148340428723]
  Inferring continuous system evolution from sparse temporal snapshots is a key challenge in generative modeling and single-cell omics.<n>Here, we introduce TracingFlow, a simulation-free Flow Matching framework generalizing to second-order dynamics.<n>By using neural networks to regress the acceleration field, TracingFlow provides an exact, efficient solution to the Dynamical Optimal Acceleration Transport (DOAT) problem.
  arXiv  Detail & Related papers  (2026-08-21T13:11:29Z)
• Chreode: A Cell World Model for One-Step Temporal Dynamics and Perturbation Prediction [35.689906216277606]
  Chreode is a one-step cell world model that predicts action-conditioned cell-state transitions.<n>The model is pretrained with a shared scVI and a DiT-based dynamics backbone on a 2.4M-cell mouse embryonic atlas.<n>As a transferable gene-state embedding for GEARS, the pretrained dynamics representation reduces shared-vocabulary DE20 mean squared error.
  arXiv  Detail & Related papers  (2026-05-27T08:05:35Z)
• Modeling Temporal scRNA-seq Data with Latent Gaussian Process and Optimal Transport [8.719471378063078]
  We propose a generative framework that models population trends using a latent heteroscedastic Gaussian process (GP) approximated by Hilbert space methods.<n>Our method explicitly captures biological extrapolation by incorporating cell-specific latent time and cell type conditioning to disentangle temporal asynchrony trajectories to different cell types.
  arXiv  Detail & Related papers  (2026-05-20T10:24:51Z)
• Tau-BNO: Brain Neural Operator for Tau Transport Model [16.92108740721371]
  Existing approaches typically model tau propagation as a diffusive process on the brain's structural connectome.<n>The Network Transport Model (NTM) was introduced to fill this gap, explaining how region-level progression of tau emerges from microscale biophysical processes.<n>We propose Tau-BNO, a Brain Neural Operator surrogate framework for rapidly approximating NTM dynamics.
  arXiv  Detail & Related papers  (2026-03-09T08:52:02Z)
• Modeling Microenvironment Trajectories on Spatial Transcriptomics with NicheFlow [80.00833033784079]
  Understanding of cellular microenvironment is essential for deciphering tissue development and disease data.<n>NicheFlow is a flow-based generative model that infers the temporal trajectory of cellular microenvironments across spatial slides.
  arXiv  Detail & Related papers  (2025-11-02T15:41:38Z)
• Learning noisy tissue dynamics across time scales [0.0]
  We introduce a biomimetic machine learning framework capable of inferring noisy multicellular dynamics directly from experimental movies.<n>This generative model combines graph neural networks, normalizing flows and WaveNet algorithms to represent tissues as neural differential equations.<n>We show that our model not only captures cell motion but also predicts the evolution of cell states in their division cycle.
  arXiv  Detail & Related papers  (2025-10-21T21:33:54Z)
• Langevin Flows for Modeling Neural Latent Dynamics [81.81271685018284]
  We introduce LangevinFlow, a sequential Variational Auto-Encoder where the time evolution of latent variables is governed by the underdamped Langevin equation.<n>Our approach incorporates physical priors -- such as inertia, damping, a learned potential function, and forces -- to represent both autonomous and non-autonomous processes in neural systems.<n>Our method outperforms state-of-the-art baselines on synthetic neural populations generated by a Lorenz attractor.
  arXiv  Detail & Related papers  (2025-07-15T17:57:48Z)
• Efficient Federated Learning with Heterogeneous Data and Adaptive Dropout [62.73150122809138]
  Federated Learning (FL) is a promising distributed machine learning approach that enables collaborative training of a global model using multiple edge devices.<n>We propose the FedDHAD FL framework, which comes with two novel methods: Dynamic Heterogeneous model aggregation (FedDH) and Adaptive Dropout (FedAD)<n>The combination of these two methods makes FedDHAD significantly outperform state-of-the-art solutions in terms of accuracy (up to 6.7% higher), efficiency (up to 2.02 times faster), and cost (up to 15.0% smaller)
  arXiv  Detail & Related papers  (2025-07-14T16:19:00Z)
• Modeling Cell Dynamics and Interactions with Unbalanced Mean Field Schrödinger Bridge [18.459855419368726]
  We develop a framework to model unbalanced interaction dynamics from snapshot data.<n>We then propose CytoBridge, a deep learning algorithm designed to approximate the UMFSB problem.<n>The effectiveness of our method has been extensively validated using both synthetic gene regulatory data and real scRNA-seq datasets.
  arXiv  Detail & Related papers  (2025-05-16T12:55:13Z)
• CellFlux: Simulating Cellular Morphology Changes via Flow Matching [44.8979602893102]
  We introduce CellFlux, an image-generative model that simulates cellular morphology changes induced by chemical and genetic perturbations.<n>CellFlux generates biologically meaningful cell images that faithfully capture perturbation-specific morphological changes.<n>These capabilities mark a significant step toward realizing virtual cell modeling for biomedical research.
  arXiv  Detail & Related papers  (2025-02-13T21:10:00Z)
• Synthetic location trajectory generation using categorical diffusion models [50.809683239937584]
  Diffusion models (DPMs) have rapidly evolved to be one of the predominant generative models for the simulation of synthetic data. We propose using DPMs for the generation of synthetic individual location trajectories (ILTs) which are sequences of variables representing physical locations visited by individuals.
  arXiv  Detail & Related papers  (2024-02-19T15:57:39Z)
• Manifold Interpolating Optimal-Transport Flows for Trajectory Inference [64.94020639760026]
  We present a method called Manifold Interpolating Optimal-Transport Flow (MIOFlow) MIOFlow learns, continuous population dynamics from static snapshot samples taken at sporadic timepoints. We evaluate our method on simulated data with bifurcations and merges, as well as scRNA-seq data from embryoid body differentiation, and acute myeloid leukemia treatment.
  arXiv  Detail & Related papers  (2022-06-29T22:19:03Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.