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.

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

Universal Cell Embeddings: A Foundation Model for Cell Biology

Comprehensive Integration of Single-Cell Data

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

Sctab: Scaling Cross-Tissue Single-Cell Annotation Models

Mapping Single-Cell Data to Reference Atlases by Transfer Learning

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

Integration of millions of transcriptomes using batch-aware triplet neural networks

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

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

Adversarial learning enables unbiased organism-wide cross-species alignment of single-cell RNA data at scale

Embryo-scale, single-cell spatial transcriptomics

STEP: Spatial Transcriptomics Embedding Procedure for Multi-scale Biological Heterogeneities Revelation in Multiple Samples

Transfer Learning Of Gene Expression Using Reactome

Cell type matching across species using protein embeddings and transfer learning

Cell Type Identification from Single-Cell Transcriptomic Data Via Gene Embedding.

CACIMAR: cross-species analysis of cell identities, markers, regulations, and interactions using single-cell RNA sequencing data

Unraveling Neuronal Identities Using SIMS: A Deep Learning Label Transfer Tool for Single-Cell RNA Sequencing Analysis

Partially Shared Multi-Modal Embedding Learns Holistic Representation of Cell State

STEP: Deciphering Spatial Atlas at Single-Cell Level with Whole-Transcriptome Coverage