Can sparse autoencoders make sense of latent representations?

๐Ÿ“… 2024-10-15
๐Ÿ›๏ธ arXiv.org
๐Ÿ“ˆ Citations: 0
โœจ Influential: 0
๐Ÿ“„ PDF
๐Ÿค– AI Summary
This study addresses the challenges of latent variable identifiability and poor interpretability in high-dimensional single-cell multi-omics data. Methodologically, we systematically investigate the capacity of sparse autoencoders (SAEs) to disentangle latent variables and uncover underlying biological mechanisms. We first validate SAEsโ€™ ability to recover ground-truth generative latents using controllable synthetic data; then perform end-to-end modeling and interpretability analysis on real single-cell multi-omics datasets. Our key contributions are: (1) the first empirical demonstration that SAEs, in a fully unsupervised setting, automatically identify biologically meaningful processesโ€”such as COโ‚‚ transport and ion homeostasis; (2) strong biological consistency in discovering erythroid differentiation and immune regulatory pathways; and (3) a theoretical advance overcoming the identifiability limitation of overparameterized models. Collectively, this work establishes a novel paradigm for interpretable representation learning in multi-omics.

Technology Category

Machine Learning: Deep Generative Models & AutoencodersSearch and Optimization: Metareasoning and MetaheuristicsNatural Language Processing: Interpretability, Analysis, and Evaluation of NLP Models

Application Category

Semantics and Knowledge: Methods to enhance, augment, integrate or synergize semantic models such as knowledge graphs and LLMsGraph Algorithms and Modeling for the Web: Graph embeddings and representation learning for Web-related graphsSearch and Retrieval-Augmented AI: Web query analysis, representation and understanding
๐Ÿ“ Abstract
Sparse autoencoders (SAEs) have lately been used to uncover interpretable latent features in large language models. Here, we explore their potential for decomposing latent representations in complex and high-dimensional biological data, where the underlying variables are often unknown. On simulated data we show that generative hidden variables can be captured in learned representations in the form of superpositions. The degree to which they are learned depends on the completeness of the representations. Superpositions, however, are not identifiable if these generative variables are unknown. SAEs can to some extent recover these variables, yielding interpretable features. Applied to single-cell multi-omics data, we show that an SAE can uncover key biological processes such as carbon dioxide transport and ion homeostasis, which are crucial for red blood cell differentiation and immune function. Our findings highlight how SAEs can be used in advancing interpretability in biological and other scientific domains.
Problem

Research questions and friction points this paper is trying to address.

Sparse Autoencoders
Complex Biological Data
Single-cell Multimodal Analysis
Innovation

Methods, ideas, or system contributions that make the work stand out.

Sparse Autoencoders
Biological Data Analysis
Single-cell Expression Models
University of Copenhagen
V
Viktoria Schuster
Department of Computer Science, University of Copenhagen