arXiv AI

TempoBridge: Source-Conditioned Flow Matching with Optimal Transport Couplings for Single-Cell Population Transitions

arXiv Machine Learning
Oct 5

Generalizable single-cell perturbation response prediction using energy-guided flow matching

The paper introduces scEGFlow, an energy‑guided flow matching framework that predicts how single cells respond to perturbations by modeling continuous transitions from control to perturbed states. It uses conditional flow matching and condition‑specific energy gradients to dynamically adjust predictions without retraining, achieving superior performance on imaging and transcriptomic benchmarks, even for unseen perturbations or limited data. The method preserves cellular manifold geometry, heterogeneity, and accurately recapitulates gene‑expression regulatory patterns.

By Jianan Wei, Jiajun Hong, Guikun Chen, Ning Yang, Lifeng Fan, Wenguan Wang
arXiv Machine Learning
Sep 29

Distribution-Conditioned Transport

The paper introduces Distribution‑Conditioned Transport (DCT), a framework that learns transport maps conditioned on embeddings of source and target distributions, allowing generalization to unseen distribution pairs. DCT supports semi‑supervised learning for distributional forecasting by leveraging distributions observed at only one condition. It is agnostic to the transport mechanism and is demonstrated on synthetic benchmarks and four biological applications, including batch effect transfer in single‑cell genomics and modeling T‑cell receptor sequence evolution.

By Nic Fishman, Gokul Gowri, Paolo L. B. Fischer, Marinka Zitnik, Omar Abudayyeh, Jonathan Gootenberg
arXiv AI
Aug 10

Control-Anchored Residual Flow Matching Conditioned on Gene Geometry for Virtual Cell Perturbation Modeling

arXiv:2608. 06824v1 Announce Type: cross Abstract: A central task in virtual cell modeling is predicting single-cell transcriptional responses to unseen genetic perturbations and drug combinations, and biological networks provide valuable priors on gene relationships.

By Quanquan Li, Yihe Chi, Liuyang Song, Hongbo Zhang, Jingyu Li, Xidong Xi, Conghua Wei, Yijie Sun, Yu Chen, Xin Liu, Qi Hu, Jing Ke, Guitao Cao
arXiv AI
Sep 7

Simulation-free Unbalanced Dynamic Optimal Transport with General Growth Penalty

The paper introduces SUDO, a simulation‑free framework for unbalanced dynamic optimal transport (UDOT) that supports general convex growth penalties beyond the quadratic Wasserstein‑Fisher‑Rao case. By showing that concave penalties lead to degenerate solutions, the authors focus on convex penalties, learning conditional paths and transport costs to solve a semi‑coupling problem and then applying unbalanced flow matching. On benchmark datasets, SUDO matches the accuracy of analytical WFR solvers while being faster than simulation‑based methods, and it also handles asymmetric penalties that better reflect proliferation‑dominant biological priors.

By Junda Ying, Yuxuan Wang, Bowen Yang, Peijie Zhou, Lei Zhang
arXiv Statistics ML
Sep 4

Distributional Treatment Effect Transportability across Heterogeneous Sites

The paper investigates how to transport the full distribution of treatment effects from a source site to a target site in a cross‑site, one‑armed design where only control units are observed in the target. It proposes learning a transformation between sites via optimal transport on control samples, then applying this transformation to the source treated sample to generate a synthetic target treated distribution. The authors prove convergence of this synthetic distribution to the true target distribution and validate the method with simulations and a patient‑derived xenograft data application.

By Borna Bateni, Yubai Yuan, Qi Xu, Annie Qu