Chreode: A Cell World Model for One-Step Temporal Dynamics and Perturbation Prediction
On the map: World models → Medical and surgical world models
Abstract (arXiv)
Predicting how a cell will change its transcriptional state under a developmental signal or a genetic perturbation is the computational core of in-silico biology and the AI Virtual Cell program. Existing approaches either fit static control-to-treated maps that discard time, or solve multi-step ODE / Schr"odinger-bridge problems on each dataset independently. We introduce Chreode, a one-step cell world model that predicts action-conditioned cell-state transitions through a structured residual transition operator. It shifts distributional evolution from inference time to training time, enabling single-pass generation while preserving a Waddington-inspired decomposition into downhill landscape flow, rotational in-tangent dynamics, and stochastic spread. The model is pretrained with a shared scVI encoder and a DiT-based dynamics backbone on a 2.4M-cell mouse embryonic atlas spanning 7 datasets. As a fine-tuning initialization, Chreode improves per-target Sinkhorn distance on Weinreb hematopoiesis and Veres islet differentiation over matched scratch models, PI-SDE, and PRESCIENT. As a transferable gene-state embedding for GEARS, the pretrained dynamics representation reduces shared-vocabulary DE20 mean squared error on Norman Perturb-seq from 0.2121 to 0.1858, a 12.4% relative improvement, without changing the GEARS training procedure. We interpret this transfer to perturbation prediction as evidence that pretrained developmental-trajectory dynamics encode differentiation primitives transferable to CRISPR-induced state shifts, since both involve cell-state transitions in a shared latent geometry. The pretrained backbone additionally produces zero-shot clonal fate scores on Weinreb that are competitive with strong dynamic-OT baselines.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Medical and surgical world models, one of 11 directions in World models. Builds on 11 of its 31 references that are in the map. First posted on arXiv 27 May 2026.
Builds on
11 of 31 references.
References · 11
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2018World Modelscited by 1,444
- 2023Improving and generalizing flow-based generative models with minibatch optimal transportcited by 835
- 2019Time2Vec: Learning a Vector Representation of Timecited by 208
- 2020TrajectoryNet: A Dynamic Optimal Transport Network for Modeling Cellular Dynamicscited by 155
- 2026Generative Modeling via Driftingcited by 98
- 2024How to Build the Virtual Cell with Artificial Intelligence: Priorities and Opportunitiescited by 50
- 2025Branched Schr"odinger Bridge Matchingcited by 4
- 2025CellStream: Dynamical Optimal Transport Informed Embeddings for Reconstructing Cellular Trajectories from Snapshots Datacited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026VCLMU: Mechanism-Centric Virtual Cell World Modeling for Perturbation Response
- 2026PerturbDiff: Functional Diffusion for Single-Cell Perturbation Modeling
- 2026Learning biophysical models of gene regulation with probability flow matching
- 2026D2R2: Discrete Diffusion with Regulation Reinforcement for Single-Cell Perturbation Prediction
- 2026Generalizable single-cell perturbation response prediction using energy-guided flow matching
- 2025VCWorld: A Biological World Model for Virtual Cell Simulation
- 2026KoopCell: Koopman-Based Generative Model for Learning Single-Cell Dynamics from Distribution Snapshots
- 2025CellFlux: Simulating Cellular Morphology Changes via Flow Matching