Sparse autoencoders reveal organized biological knowledge but minimal regulatory logic in single-cell foundation models: a comparative atlas of Geneformer and scGPT
Organizations: Department of Computer Science, University of Tübingen, Tübingen, Germany
Abstract
Background: Single-cell foundation models such as Geneformer and scGPT encode rich biological information, but whether this includes causal regulatory logic rather than statistical co-expression remains unclear. Sparse autoencoders (SAEs) can resolve superposition in neural networks by decomposing dense activations into interpretable features, yet they have not been systematically applied to biological foundation models. Results: We trained TopK SAEs on residual stream activations from all layers of Geneformer V2-316M (18 layers, d=1152) and scGPT whole-human (12 layers, d=512), producing atlases of 82525 and 24527 features, respectively. Both atlases confirm massive superposition, with 99.8 percent of features invisible to SVD. Systematic characterization reveals rich biological organization: 29 to 59 percent of features annotate to Gene Ontology, KEGG, Reactome, STRING, or TRRUST, with U-shaped layer profiles reflecting hierarchical abstraction. Features organize into co-activation modules (141 in Geneformer, 76 in scGPT), exhibit causal specificity (median 2.36x), and form cross-layer information highways (63 to 99.8 percent). When tested against genome-scale CRISPRi perturbation data, only 3 of 48 transcription factors (6.2 percent) show regulatory-target-specific feature responses. A multi-tissue control yields marginal improvement (10.4 percent, 5 of 48 TFs), establishing model representations as the bottleneck. Conclusions: These models have internalized organized biological knowledge, including pathway membership, protein interactions, functional modules, and hierarchical abstraction, yet they encode minimal causal regulatory logic. We release both feature atlases as interactive web platforms enabling exploration of more than 107000 features across 30 layers of two leading single-cell foundation models.
Figures & tables
| Layer | VarExpl | Alive | Dead | SVD | Novel | MeanCos | Ann% | Enrichments |
|---|---|---|---|---|---|---|---|---|
| 0 | 83.9% | 4,608 | 0 | 41 | 4,567 | 0.033 | 58.6% | 23,959 |
| 1 | 84.6% | 4,606 | 2 | 27 | 4,579 | 0.033 | 57.4% | 23,131 |
| 2 | 84.8% | 4,601 | 7 | 29 | 4,572 | 0.034 | 55.5% | 23,383 |
| 3 | 85.2% | 4,595 | 13 | 12 | 4,583 | 0.035 | 56.4% | 21,922 |
| 4 | 85.3% | 4,583 | 25 | 13 | 4,570 | 0.037 | 53.9% | 19,910 |
| 5 | 84.6% | 4,576 | 32 | 8 | 4,568 | 0.036 | 52.1% | 17,865 |
| Layer | VarExpl | Alive | Dead | MeanCos | Ann% | Modules |
|---|---|---|---|---|---|---|
| 0 | 92.0% | 2,027 | 21 | 0.038 | 32.7% | 6 |
| 1 | 93.0% | 2,035 | 13 | 0.041 | 30.8% | 6 |
| 2 | 93.2% | 2,038 | 10 | 0.040 | 28.7% | 5 |
| 3 | 93.2% | 2,045 | 3 | 0.041 | 31.6% | 7 |
| 4 | 93.5% | 2,048 | 0 | 0.041 | 30.4% | 6 |
| 5 | 92.1% | 2,048 | 0 | 0.042 | 29.6% | 7 |
| Metric | Geneformer | scGPT |
|---|---|---|
| Architecture | 18 layers, | 12 layers, |
| Training data | 30M cells | 33M cells |
| Input encoding | Rank-value tokens | Continuous expression |
| Training objective | Next-token prediction | Masked gene prediction |
| Features per layer | 4,608 | 2,048 |
| Total alive features | 82,525 | 24,527 |
| Feature | Top Genes | Biological Identity | Databases |
| Layer 0 — Molecular Programs | |||
| F3717 | CDK1, CDC20, DLGAP5, PBK | Cell cycle (G2/M) | GO, KEGG, React., STRING |
| F3607 | RRM1, E2F1, MCM4, MCM6, RAD51 | DNA replication / repair | GO, KEGG, React., STRING, TRRUST |
| F1116 | ARL6IP4, SQSTM1, JAK1 | B cell activation | GO, KEGG, Reactome |
| F4536 | DYNC1H1, TLN1, MYH9, SPTAN1 | Cytoskeleton / focal adhesion | GO, KEGG, Reactome |
| F2829 | TGFB1, PKN1, GADD45A, ZBTB7A | MAPK/TGF signaling | GO, KEGG, React., STRING, TRRUST |
| Metric | Value |
|---|---|
| Features tested | 50 |
| Mean specificity ratio | 8.98 |
| Median specificity ratio | 2.36 |
| Specific ( 2 ) | 30/50 (60%) |
| Highly specific ( 10 ) | 6/50 (12%) |
| Mean target logit diff | 0.116 |
| Metric | Value |
|---|---|
| Perturbation targets tested | 100 |
| Targets causing feature changes | 92/100 (92%) |
| TRRUST TFs tested | 48 |
| TFs with specific response | 3/48 (6.2%) |
| Mean responding features per target | 2.54 |
| Mean specific features per target | 0.03 |
| SAE Training Data | Layer | TFs Specific | Rate |
|---|---|---|---|
| K562-only | 11 | 3/48 | 6.2% |
| K562 + Tabula Sapiens | 0 | 0/48 | 0.0% |
| K562 + Tabula Sapiens | 5 | 4/48 | 8.3% |
| K562 + Tabula Sapiens | 11 | 5/48 | 10.4% |
| K562 + Tabula Sapiens | 17 | 1/48 | 2.1% |
| Criterion | Target | Result | Status |
| Modules match pathways | 20 total | 141 (6–12/layer) | Exceeded |
| Causal specificity | Majority 2 | 60% 2 , median 2.36 | Exceeded |
| Perturbation specificity | 30% | 6.2% (3/48 TFs) | Not met |
| Unannotated structure | 10% in clusters | 95–98.5% in modules ∗ | Partial |
| Cross-layer tracking | 50% | 97–99.8% highways | Exceeded |
| ∗ Co-activation module membership, not standalone clusters (2–3%). | |||
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| Layer | GO BP | KEGG | Reactome | STRING | TRRUST TF | TRRUST Edges |
|---|---|---|---|---|---|---|
| 0 | 10,153 | 2,650 | 11,001 | 302 | 155 | 42 |
| 1 | 10,022 | 2,433 | 10,512 | 258 | 164 | 48 |
| 2 | 9,948 | 2,495 | 10,790 | 283 | 150 | 32 |
| 3 | 9,726 | 2,514 | 9,525 | 273 | 157 | 37 |
| 4 | 8,537 | 2,045 | 9,195 | 248 | 133 | 30 |
| 5 | 7,695 | 1,845 | 8,189 | 216 | 136 | 24 |
| L0 Target | Matches (cos ) | Rate |
|---|---|---|
| L0 L1 | 114 | 2.5% |
| L0 L2 | 93 | 2.0% |
| L0 L4 | 67 | 1.5% |
| L0 L6 | 25 | 0.5% |
| L0 L8 | 10 | 0.2% |
| L0 L10 | 1 | 0% |
| Layer | Modules | Feats in Modules | PMI Edges | Coverage |
|---|---|---|---|---|
| 0 | 6 | 4,577 | 446,324 | 99.3% |
| 1 | 8 | 4,562 | 440,681 | 99.0% |
| 2 | 7 | 4,536 | 404,403 | 98.6% |
| 3 | 8 | 4,518 | 393,574 | 98.3% |
| 4 | 9 | 4,502 | 393,194 | 98.2% |
| 5 | 12 | 4,472 | 390,845 | 97.7% |
| Feature | Annotation | Specificity | Target | Other |
|---|---|---|---|---|
| F2035 | Cell Differentiation (neg. reg.) | 114.5 | 0.208 | 0.002 |
| F3692 | ERAD Pathway | 108.1 | 0.129 | 0.001 |
| F3933 | Intracellular Signaling (neg. reg.) | 55.7 | 0.196 | 0.004 |
| F157 | Golgi Vesicle Transport | 25.4 | 0.056 | 0.002 |
| F3532 | Protein Metabolic Process (pos. reg.) | 11.2 | 0.127 | 0.011 |
| F4516 | Mitotic Spindle Microtubules | 10.6 | 0.672 | 0.063 |
| Layer Pair | Feats w/ Deps | Mean Max PMI | Median Max PMI | Max PMI | Highways |
|---|---|---|---|---|---|
| L0 L5 | 4,604 | 6.61 | 6.72 | 11.10 | 4,530 (98.4%) |
| L5 L11 | 4,518 | 6.63 | 6.71 | 10.87 | 4,401 (97.4%) |
| L11 L17 | 4,555 | 6.79 | 6.86 | 10.66 | 4,544 (99.8%) |
| Layer Pair | PMI Edges | Upstream | Downstream | Max PMI |
|---|---|---|---|---|
| L0 L4 | 75,305 | 1,935/2,027 (95.5%) | 1,960/2,048 (95.7%) | 9.15 |
| L4 L8 | 61,263 | 1,955/2,048 (95.5%) | 1,723/2,048 (84.1%) | 9.26 |
| L8 L11 | 45,258 | 1,773/2,048 (86.6%) | 1,289/2,048 (62.9%) | 10.78 |
| Pair | Source Feature | Target Feature | Biological Logic | PMI |
|---|---|---|---|---|
| L0 L5 | Protein Processing in ER | unlabeled | ER stress cascade | 11.10 |
| L0 L5 | mTORC1 Regulation | Autophagy | mTORC1 autophagy | 9.55 |
| L0 L5 | Wnt Signaling | unlabeled | Wnt pathway processing | 9.48 |
| L5 L11 | Protein Polyubiq. | unlabeled | Protein quality control | 10.87 |
| L5 L11 | Translation | unlabeled | Translational regulation | 10.35 |
| L5 L11 | RNA Splicing (neg. reg.) | unlabeled | Post-transcriptional | 10.21 |
| TF | K562-SAE Specific | MT-SAE Specific | Change |
| ATF5 | 0 | 1 | gained |
| BRCA1 | 0 | 1 | gained |
| GATA1 | 0 | 1 | gained |
| RBMX | 0 | 1 | gained |
| NFRKB | 0 | 1 | gained |
| MAX | 1 | 0 | lost |
| SAE | Features with TFs in top genes | TF-dominant ( 3 TFs) |
|---|---|---|
| K562-only L11 | 2,967/4,598 (64.5%) | 424 |
| Multi-tissue L0 | 2,796/4,608 (60.7%) | 452 |
| Multi-tissue L5 | 2,777/4,568 (60.8%) | 337 |
| Multi-tissue L11 | 2,782/4,601 (60.5%) | 343 |
| Multi-tissue L17 | 2,680/4,603 (58.2%) | 346 |
| Layer | Annotated | Unannotated | Clusters | Co-activate w/ Annotated | Isolated |
|---|---|---|---|---|---|
| 0 | 2,702 | 1,906 | 15 (48 feats) | 1,876 (98.4%) | 30 (1.6%) |
| 5 | 2,383 | 2,193 | 19 (69 feats) | 2,090 (95.3%) | 103 (4.7%) |
| 11 | 2,583 | 2,015 | 11 (47 feats) | 1,984 (98.5%) | 28 (1.4%) |
| 17 | 2,154 | 2,426 | 12 (58 feats) | 2,334 (96.2%) | 75 (3.1%) |