HEIMDALL: A Modular Framework for Tokenization in Single-Cell Foundation Models
Haber, E.; Alam, S.; Ho, N.; Liu, R.; Trop, E.; Liang, S.; Yang, M.; Krieger, S.; Ma, J.
Show abstract
Foundation models trained on single-cell RNA-sequencing (scRNA-seq) data have rapidly become powerful tools for single-cell analysis. Their performance, however, depends critically on how cells are tokenized into model inputs - a design space that remains poorly understood. Here, we present HO_SCPLOWEIMDALLC_SCPLOW, a comprehensive framework and open-source toolkit for systematically evaluating tokenization strategies in single-cell foundation models (scFMs). HO_SCPLOWEIMDALLC_SCPLOW decomposes each scFM into modular components: a gene identity encoder (FG), an expression encoder (FE), and a "cell sentence" constructor (FC) with submodules (ORDER, SEQUENCE, and REDUCE) enabling fine-grained control and attribution. Using a transformer trained from scratch, we evaluate tokenization strategies for cell type classification across challenging transfer learning settings - cross-tissue, cross-species, and spatial gene-panel shifts - and separately assess reverse perturbation prediction. Tokenization choices show minimal impact in-distribution but are decisive under distribution shift, with FG and ORDER driving the largest gains and FE providing additional improvements. HO_SCPLOWEIMDALLC_SCPLOW further shows how existing strategies can be recombined to enhance generalization. By standardizing evaluation and providing an extensive library, HO_SCPLOWEIMDALLC_SCPLOW establishes a foundation for reproducible, systematic exploration of single-cell tokenization and accelerates the development of next-generation scFMs.
Matching journals
The top 4 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
- Learning interpretable cellular and gene signature embeddings from single-cell transcriptomic data 97%
- scPRINT: pre-training on 50 million cells allows robust gene network predictions 96%
- scDREAMER: atlas-level integration of single-cell datasets using deep generative model paired with adversarial classifier 96%
Similar papers in this journal
- scCausalVI disentangles single-cell perturbation responses with causality-aware generative model 96%
- Learning multi-cellular representations of single-cell transcriptomics data enables characterization of patient-level disease states 95%
- Ultrafast and interpretable single-cell 3D genome analysis with Fast-Higashi 95%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.