Abstract:The increasing availability of large-scale single-cell datasets has enabled the detailed description of cell states across multiple biological conditions and perturbations. In parallel, recent advances in unsupervised machine learning, particularly in transfer learning, have enabled fast and scalable mapping of these new single-cell datasets onto reference atlases. The resulting large-scale machine learning models however often have millions of parameters, rendering interpretation of the newly mapped datasets challenging. Here, we propose expiMap, a deep learning model that enables interpretable reference mapping using biologically understandable entities, such as curated sets of genes and gene programs. The key concept is the substitution of the uninterpretable nodes in an autoencoder’s bottleneck by labeled nodes mapping to interpretable lists of genes, such as gene ontologies, biological pathways, or curated gene sets, for which activities are learned as constraints during reconstruction. This is enabled by the incorporation of predefined gene programs into the reference model, and at the same time allowing the model to learn de novo new programs and refine existing programs during reference mapping. We show that the model retains similar integration performance as existing methods while providing a biologically interpretable framework for understanding cellular behavior. We demonstrate the capabilities of expiMap by applying it to 15 datasets encompassing five different tissues and species. The interpretable nature of the mapping revealed unreported associations between interferon signaling via the RIG-I/MDA5 and GPCRs pathways, with differential behavior in CD8+ T cells and CD14+ monocytes in severe COVID-19, as well as the role of annexins in the cellular communications between lymphoid and myeloid compartments for explaining patient response to the applied drugs. Finally, expiMap enabled the direct comparison of a diverse set of pancreatic beta cells from multiple studies where we observed a strong, previously unreported correlation between the unfolded protein response and asparagine N-linked glycosylation. Altogether, expiMap enables the interpretable mapping of single cell transcriptome data sets across cohorts, disease states and other perturbations. ### Competing Interest Statement Fabian J. Theis consults for Immunai Inc., Singularity Bio B.V., CytoReason Ltd, and Omniscope Ltd, and has ownership interest in Dermagnostix GmbH and Cellarity.

Universal Cell Embeddings: A Foundation Model for Cell Biology

Mapping cells to gene programs

Single-cell Reference Mapping to Construct and Extend Cell-Type Hierarchies

Building a learnable universal coordinate system for single-cell atlas with a joint-VAE model

CELLama: Foundation Model for Single Cell and Spatial Transcriptomics by Cell Embedding Leveraging Language Model Abilities

SubCell: Vision foundation models for microscopy capture single-cell biology

A General Framework for Representing and Annotating Multifaceted Cell Heterogeneity in Human Cell Atlas

Toward universal cell embeddings: integrating single-cell RNA-seq datasets across species with SATURN

Cell-Graph Compass: Modeling Single Cells with Graph Structure Foundation Model

A Foundation Model for Cell Segmentation

Construction of continuously expandable single-cell atlases through integration of heterogeneous datasets in a generalized cell-embedding space

Biologically Informed Deep Learning to Infer Gene Program Activity in Single Cells

Sctab: Scaling Cross-Tissue Single-Cell Annotation Models

A sandbox for prediction and integration of DNA, RNA, and proteins in single cells

Contrastive Learning for Robust Cell Annotation and Representation from Single-Cell Transcriptomics

CellFM: a large-scale foundation model pre-trained on transcriptomics of 100 million human cells

Insights, opportunities and challenges provided by large cell atlases

hECA: The cell-centric assembly of a cell atlas

Cell type discovery using single-cell transcriptomics: implications for ontological representation

Multi-omics Single-Cell Data Integration and Regulatory Inference with Graph-Linked Embedding

scEMB: Learning context representation of genes based on large-scale single-cell transcriptomics