Ground-truth evaluation of sparse autoencoder features on spatial, single-cell and language-model embeddings

Sparse autoencoders (SAEs) are increasingly used to extract interpretable features from the embeddings of biological foundation models. Because SAEs trained with different random seeds learn different features, reproducibility across seeds has become a common test of whether these features can be trusted. We asked whether reproducible features are also meaningful, using embeddings whose concepts are known: five spatial transcriptomics embeddings of two colorectal cancer sections and Geneformer embeddings of single-cell RNA profiles, with cell lineages as concepts; synthetic data with planted cell states; and the residual stream of a small language model, with token classes as concepts. Varimax rotations of the same embeddings were almost perfectly reproducible, yet rotation features (PCA, varimax and ICA) covered at most three of seven lineages and three of eight token classes. Single SAE fits showed the opposite profile: they covered most concepts with purer gene programs but reproduced only 27-46% of their features. A consensus over five SAE fits combined the two strengths. It was reproducible (78-98% across sources) and as pure as single fits: 11-40% of its features carried marker genes of another lineage, against 44-70% for varimax. Against the strongest baseline, varimax sparsified to the SAE's sparsity, the consensus covered as many concepts on PCA and Geneformer embeddings and more on learned spatial and text embeddings. The consensus also had costs. It covered fewer lineages than single fits and, like all SAEs, missed weak planted states that rotations recovered; neither a longer single fit nor a second SAE trained on the consensus residual recovered what it discarded. Reproducibility alone therefore ranks features poorly: SAE features should be reported with the number of known concepts they cover and the purity of their gene programs, against a rotation of the same embedding sparsified to the SAE's sparsity.

Authors

Institutions

Publication Details

Journal
bioRxiv (Cold Spring Harbor Laboratory)
Published
2026-10-29
DOI
https://doi.org/10.64898/2026.10.27.762296
Primary Topic
Topic Modeling
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

Ground-truth evaluation of sparse autoencoder features on spatial, single-cell and language-model embeddings

Shayan Forghani, Amirhossein Shahsavand, Saba Akbarzadeh, Fatemeh Mahdavi Sabet
2 links
bioRxiv (Cold Spring Harbor Laboratory)
Topic Modeling
preprint

Ground-truth evaluation of sparse autoencoder features on spatial, single-cell and language-model embeddings

Shayan Forghani, Amirhossein Shahsavand, Saba Akbarzadeh, Fatemeh Mahdavi Sabet
preprint en

Abstract

Sparse autoencoders (SAEs) are increasingly used to extract interpretable features from the embeddings of biological foundation models. Because SAEs trained with different random seeds learn different features, reproducibility across seeds has become a common test of whether these features can be trusted. We asked whether reproducible features are also meaningful, using embeddings whose concepts are known: five spatial transcriptomics embeddings of two colorectal cancer sections and Geneformer embeddings of single-cell RNA profiles, with cell lineages as concepts; synthetic data with planted cell states; and the residual stream of a small language model, with token classes as concepts. Varimax rotations of the same embeddings were almost perfectly reproducible, yet rotation features (PCA, varimax and ICA) covered at most three of seven lineages and three of eight token classes. Single SAE fits showed the opposite profile: they covered most concepts with purer gene programs but reproduced only 27-46% of their features. A consensus over five SAE fits combined the two strengths. It was reproducible (78-98% across sources) and as pure as single fits: 11-40% of its features carried marker genes of another lineage, against 44-70% for varimax. Against the strongest baseline, varimax sparsified to the SAE's sparsity, the consensus covered as many concepts on PCA and Geneformer embeddings and more on learned spatial and text embeddings. The consensus also had costs. It covered fewer lineages than single fits and, like all SAEs, missed weak planted states that rotations recovered; neither a longer single fit nor a second SAE trained on the consensus residual recovered what it discarded. Reproducibility alone therefore ranks features poorly: SAE features should be reported with the number of known concepts they cover and the purity of their gene programs, against a rotation of the same embedding sparsified to the SAE's sparsity.

bioRxiv (Cold Spring Harbor Laboratory)
Tehran University of Medical Sciences (IR)
Topic Modeling
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Laika access coming soon