How Much Can AI Virtual Cells Gain from Unified Perturbational Modeling?
- 1School of Science and Engineering, The Chinese University of Hong Kong, Shenzhen (CUHK-Shenzhen)
Abstract
Predicting how individual cells respond to perturbations is a central task in artificial intelligence virtual cell (AIVC) modeling. However, integrating perturbation datasets is challenging because they differ in measured gene sets, perturbation types and cellular contexts. Existing AIVC predictors are typically developed for specific settings, limiting their applicability across heterogeneous datasets. Here, we investigate the extent to which AIVC predictors can benefit from unified perturbational modeling (UPM), a novel paradigm designed to integrate heterogeneous Perturb-seq datasets into a single predictive model. To this end, we introduce three unification modules into a variational autoencoder (VAE)-based AIVC predictor to address these sources of heterogeneity: gene-channel alignment, shared perturbation encoding, and joint embedding of gene expression and identity. We systematically evaluate and validate the individual contributions of these modules. Building on UPM, we assemble a dataset of 110 million cells spanning over 22,000 genes, four perturbation types and 76 cellular contexts, and use 89.7 million cells from 71 contexts to train a unified AIVC predictor with more than one billion parameters. Across genetic perturbation benchmarks, UPM achieves the highest PES and gene-wise response correlation among the evaluated models. It also improves all four evaluation metrics over State on the five held-out Tahoe-100M cell lines and over Stack on the cytokine benchmark. UPM also improves combinatorial perturbation prediction relative to specialized predictors. In the HSC application, our model achieves a 40.0% improvement in accuracy on a downstream task involving the recapitulation of effects induced by the knockdown of hepatic stellate cell (HSC)-related targets. Our implementation is openly available at https://github.com/youlab2025/UPM.
Related
- Speciesformer learns conserved cellular states for cross-species generative virtual cell modeling
- BLOOM: Biologically Informed Flow Matching for Transcriptomic Virtual Embryo Modeling
- UniHAR: A unified hierarchical transformer with multi-scale temporal modeling for wearable human activity recognition
- Unified modeling of mutation-induced protein stability responses with OmegaTherm
- OmniVCBench: Benchmarking Evidence-Grounded Multimodal Reasoning Towards AI Virtual Cells
BibTeX
@misc{wei2026much,
title = {{How Much Can AI Virtual Cells Gain from Unified Perturbational Modeling?}},
author = {Z. Wei and Y. Ye and S. Cui and W. Ma and Y. You},
howpublished = {bioRxiv},
year = {2026},
month = oct,
doi = {10.64898/2026.10.25.761625},
url = {https://www.biorxiv.org/content/10.64898/2026.10.25.761625v1},
note = {CC BY-NC},
}
RIS
TY - UNPB
TI - How Much Can AI Virtual Cells Gain from Unified Perturbational Modeling?
AU - Z. Wei
AU - Y. Ye
AU - S. Cui
AU - W. Ma
AU - Y. You
PY - 2026
DA - 2026/10/25
PB - bioRxiv
DO - 10.64898/2026.10.25.761625
UR - https://www.biorxiv.org/content/10.64898/2026.10.25.761625v1
AB - Predicting how individual cells respond to perturbations is a central task in artificial intelligence virtual cell (AIVC) modeling. However, integrating perturbation datasets is challenging because they differ in measured gene sets, perturbation types and cellular contexts. Existing AIVC predictors are typically developed for specific settings, limiting their applicability across heterogeneous datasets. Here, we investigate the extent to which AIVC predictors can benefit from unified perturbational modeling (UPM), a novel paradigm designed to integrate heterogeneous Perturb-seq datasets into a single predictive model. To this end, we introduce three unification modules into a variational autoencoder (VAE)-based AIVC predictor to address these sources of heterogeneity: gene-channel alignment, shared perturbation encoding, and joint embedding of gene expression and identity. We systematically evaluate and validate the individual contributions of these modules. Building on UPM, we assemble a dataset of 110 million cells spanning over 22,000 genes, four perturbation types and 76 cellular contexts, and use 89.7 million cells from 71 contexts to train a unified AIVC predictor with more than one billion parameters. Across genetic perturbation benchmarks, UPM achieves the highest PES and gene-wise response correlation among the evaluated models. It also improves all four evaluation metrics over State on the five held-out Tahoe-100M cell lines and over Stack on the cytokine benchmark. UPM also improves combinatorial perturbation prediction relative to specialized predictors. In the HSC application, our model achieves a 40.0% improvement in accuracy on a downstream task involving the recapitulation of effects induced by the knockdown of hepatic stellate cell (HSC)-related targets. Our implementation is openly available at https://github.com/youlab2025/UPM.
ER -