Reinforcement learning enables single-cell foundation models to learn cellular differentiation
Abstract
While single-cell foundation models excel at static representation learning and single-step perturbation prediction, their capacity to model and control dynamic, sequential cell state transitions remains underexplored. Here, we introduce Differentiation with Reinforcement Learning (DiRL), a framework that transforms foundation models into predictive environments for optimizing multi-step perturbation strategies. By formulating differentiation as a goal-conditioned sequential decision-making problem, DiRL trains agents to navigate the high-dimensional latent landscape of gene expression, learning policies that direct stem cells toward specific terminal fates. Evaluated on iPSC-derived organoid datasets, DiRL outperforms random perturbation baselines with a 69% win rate, with performance scaling systematically as the planning horizon increases. Beyond optimization, DiRL offers interpretable insights into the mechanics of cell fate transitions: the learned policies successfully recover known sequential regulators in Wnt and Hedgehog signaling pathways, while the value function from the critic model recapitulates biological pseudotime orderings comparable to established trajectory inference methods. These results demonstrate that coupling reinforcement learning with foundation models enables a paradigm shift from static embedding analysis to dynamic trajectory modeling, providing a powerful engine for discovering sequential perturbation strategies in regenerative medicine.
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 1
- Organism
- Assay
- Research Focus
- Approach
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- Benchmarking zero-shot single-cell foundation model embeddings for cellular dynamics reconstruction
Xueya Zhou, Zihan Wang, … Peijie Zhou·bioRxiv·Mar 12, 2026
- scRL: Utilizing Reinforcement Learning to Evaluate Fate Decisions in Single-Cell Data
Zeyu Fu, Chunlin Chen, … Shilei Chen·BiologyPeer-reviewed·Jun 11, 2025
- Closing the loop: Teaching single-cell foundation models to learn from perturbations
Yash Pershad, Tarak Nandi, … Alexander G. Bick·bioRxiv·Jul 8, 2025
- Multimodal Benchmarking of Foundation Model Representations for Cellular Perturbation Response Prediction
Euxhen Hasanaj, Elijah Cole, … Eric P. Xing·bioRxiv·Jun 26, 2025
- Predicting how perturbations reshape cellular trajectories with PerturbGen
Kevin Chi Hao Ly, Adib Miraki Feriz, … Mohammad Lotfollahi·bioRxiv·Mar 5, 2026
- Task-dependent Performance of Single-cell Foundation Models under Low Supervision
Yongjie Xu, Yiming Li, … Zelin Zang·bioRxiv·Aug 21, 2026