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.

Explainable autoencoder-based representation learning for gene expression data

Learning interpretable latent autoencoder representations with annotations of feature sets

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

Developing explainable models for lncRNA-Targeted drug discovery using graph autoencoders

A deep auto-encoder model for gene expression prediction

Discovering genetic biomarkers for targeted cancer therapeutics with eXplainable AI

Deep autoencoder for interpretable tissue-adaptive deconvolution and cell-type-specific gene analysis

AE-TPGG: a novel autoencoder-based approach for single-cell RNA-seq data imputation and dimensionality reduction

CODE-AE: A Coherent De-confounding Autoencoder for Predicting Patient-Specific Drug Response From Cell Line Transcriptomics

Principled feature attribution for unsupervised gene expression analysis

Single-cell gene regulatory network prediction by explainable AI

Evaluating deep variational autoencoders trained on pan-cancer gene expression

Studying Limits of Explainability by Integrated Gradients for Gene Expression Models

Extracting a biologically relevant latent space from cancer transcriptomes with variational autoencoders

DeepGSEA: explainable deep gene set enrichment analysis for single-cell transcriptomic data

XOmiVAE: an interpretable deep learning model for cancer classification using high-dimensional omics data

GSAE: an autoencoder with embedded gene-set nodes for genomics functional characterization

SurvIAE: Survival prediction with Interpretable Autoencoders from Diffuse Large B-Cells Lymphoma gene expression data

Autoencoder-transformed transcriptome improves genotype-phenotype association studies

Unsupervised feature construction and knowledge extraction from genome-wide assays of breast cancer with denoising autoencoders

RVAgene: generative modeling of gene expression time series data