HarmonyCell: Automating Single-Cell Perturbation Modeling under Semantic and Distribution Shifts
Organizations: Fudan University · Shanghai Artificial Intelligence Laboratory · The Hong Kong University of Science and Technology · Xi’an Jiaotong-Liverpool University ·
Abstract
Single-cell perturbation studies face dual heterogeneity bottlenecks: (i) semantic heterogeneity--identical biological concepts encoded under incompatible metadata schemas across datasets; and (ii) statistical heterogeneity--distribution shifts from biological variation demanding dataset-specific inductive biases. We propose HarmonyCell, an end-to-end agent framework resolving each challenge through a dedicated mechanism: an LLM-driven Semantic Unifier autonomously maps disparate metadata into a canonical interface without manual intervention; and an adaptive Monte Carlo Tree Search engine operates over a hierarchical action space to synthesize architectures with optimal statistical inductive biases for distribution shifts. Evaluated across diverse perturbation tasks under both semantic and distribution shifts, HarmonyCell achieves a 95% valid execution rate on heterogeneous input datasets (versus 0% for general agents) while matching or even exceeding expert-designed baselines in rigorous out-of-distribution evaluations. This dual-track orchestration enables scalable automatic virtual cell modeling without dataset-specific engineering.
Figures & tables
| Abilities | Heterogeneity Data Unification | Biological Prior | Model Exploration | Collaborative Coding |
|---|---|---|---|---|
| General-Purpose Agents | ✓ | |||
| Specialized Cell Scientists | ✓ | ✓ | ✓ | |
| HarmonyCell | ✓ | ✓ | ✓ | ✓ |
| Agents | Error Type | Valid Execution Rate | ||
|---|---|---|---|---|
| Preprocess Error | Model Error | Hallucinated Success | ||
| AIDE ( Jiang et al., 2025 ) | 35% | 50% | 15% | 0% |
| R&D Agent ( Yang et al., 2025 ) | 45% | 30% | 25% | 0% |
| HarmonyCell | 0% | 5% | 0% | 95% |
| CosLogFC | RMSE | DeltaPCC | CosLogFC | RMSE | DeltaPCC | |
|---|---|---|---|---|---|---|
| Gene Perturbation | Norman ( Norman et al., 2019 ) | Adamson ( Adamson et al., 2016 ) | ||||
| Biolord ( Piran et al., 2024 ) | 0.57 | 0.05 | 0.42 | 0.52 | 0.02 | 0.54 |
| Sams VAE ( Bereket and Karaletsos, 2023 ) | 0.58 | 0.05 | 0.44 | 0.51 | 0.02 | 0.55 |
| CPA ( Lotfollahi et al., 2021 ) | 0.55 | 0.05 | 0.39 | 0.41 | 0.03 | 0.32 |
| HarmonyCell | 0.61 | 0.09 | 0.62 | 0.32 | 0.09 | 0.49 |
| Drug Perturbation | Srivatsan-Sciplex2 ( Srivatsan et al., 2020 ) | Srivatsan-Sciplex3 ( Srivatsan et al., 2020 ) | ||||
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Component | Symbol | Value | Description |
|---|---|---|---|
| Retrieval (RAG) | 0.3 | Threshold for filtering irrelevant historical tasks | |
| 3 | Number of top neighbors retrieved | ||
| MCTS Selection | 0.7 | Weight for Optimistic Value ( ) | |
| 1.0 | Exploration constant in UCT formula | ||
| 32 | Number of MCTS simulation iterations | ||
| Reward Function | 0.8 | Weight for validation performance ( PCC) |
Single-Cell Perturbation Modeling
Single-cell perturbation modeling predicts how genetic or chemical interventions change cellular states across cells and contexts.
See all 61 papers