HarmonyCell: Automating Single-Cell Perturbation Modeling under Semantic and Distribution Shifts
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.
Keywords
Cite
@article{arxiv.2603.01396,
title = {HarmonyCell: Automating Single-Cell Perturbation Modeling under Semantic and Distribution Shifts},
author = {Wenxuan Huang and Mingyu Tsoi and Yanhao Huang and Xinjie Mao and Xue Xia and Hao Wu and Jiaqi Wei and Yuejin Yang and Lang Yu and Cheng Tan and Xiang Zhang and Zhangyang Gao and Siqi Sun},
journal= {arXiv preprint arXiv:2603.01396},
year = {2026}
}
Comments
18 pages total (8 pages main text + appendix), 6 figures