Causal Circuit Tracing Reveals Distinct Computational Architectures in Single-Cell Foundation Models: Inhibitory Dominance, Biological Coherence, and Cross-Model Convergence
Organizations: Department of Computer Science, University of Tübingen, Tübingen, Germany
Abstract
Motivation: Sparse autoencoders (SAEs) decompose foundation model activations into interpretable features, but causal feature-to-feature interactions across network depth remain unknown for biological foundation models. Results: We introduce causal circuit tracing by ablating SAE features and measuring downstream responses, and apply it to Geneformer V2-316M and scGPT whole-human across four conditions (96,892 edges, 80,191 forward passes). Both models show approximately 53 percent biological coherence and 65 to 89 percent inhibitory dominance, invariant to architecture and cell type. scGPT produces stronger effects (mean absolute d = 1.40 vs. 1.05) with more balanced dynamics. Cross-model consensus yields 1,142 conserved domain pairs (10.6x enrichment, p < 0.001). Disease-associated domains are 3.59x more likely to be consensus. Gene-level CRISPRi validation shows 56.4 percent directional accuracy, confirming co-expression rather than causal encoding.
Figures & tables
| Metric | K562/K562 | K562/Multi | TS/Multi (GF) | TS/Multi (scGPT) |
|---|---|---|---|---|
| Model | Geneformer | Geneformer | Geneformer | scGPT |
| Layers with SAEs | 18 (all) | 4 (subset) | 4 (subset) | 12 (all) |
| Features/layer | 4,608 | 4,608 | 4,608 | 2,048 |
| Source features | 120 | 90 | 90 | 90 |
| Input cells | 200 K562 | 200 K562 | 200 TS | 200 TS |
| Forward passes | 24,776 | 18,465 | 18,455 | 18,495 |
| Source | Downstream | Passes | Time | Total Edges | Avg/Feature | Range |
|---|---|---|---|---|---|---|
| L0 | 17 (L1–L17) | 6,176 | 150 min | 73,769 | 2,459 | 382–8,028 |
| L5 | 12 (L6–L17) | 6,200 | 140 min | 41,684 | 1,389 | 533–4,703 |
| L11 | 6 (L12–L17) | 6,200 | 86 min | 30,981 | 1,033 | 321–3,536 |
| L15 | 2 (L16–L17) | 6,200 | 71 min | 18,438 | 615 | 79–3,257 |
| Feature | Biology | Out-Deg | Feature | In-Deg |
|---|---|---|---|---|
| L0_F2905 | Golgi Organization | 8,028 | L16_F2818 | 93 |
| L0_F2982 | RNA Methylation | 6,921 | L16_F1691 | 89 |
| L0_F1568 | Growth Factor Response | 6,006 | L16_F4354 | 89 |
| L0_F3402 | Cholesterol Biosynthesis | 5,096 | L16_F1375 | 88 |
| L0_F4201 | RNA Splicing | 4,782 | L16_F1057 | 88 |
| Source | Target | Shared | Biological Theme | |
| DNA damage response | ||||
| L0 DNA Repair | L1 DNA Damage Resp. | 113 | Damage detection response | |
| L0 DNA Repair | L6 Kinetochore | — | Damage mitotic checkpoint | |
| L5 DNA Metabolic | L7 G2/M Transition | — | Replication division | |
| L5 DNA Metabolic | L11 Kinetochore | — | Replication spindle | |
| Mitotic apparatus | ||||
| Layer Pair | PMI Targets | Causal Targets | Target Overlap |
|---|---|---|---|
| L0 L5 | 4,101 | 1,113 | 90.6% |
| L5 L11 | 4,369 | 996 | 94.8% |
| L11 L17 | 4,205 | 881 | 91.7% |
| Source | Downstream | Passes | Time | Total Edges | Avg/Feature |
|---|---|---|---|---|---|
| L0 | 11 (L1–L11) | 6,194 | 17.0 min | 49,777 | 1,659 |
| L4 | 7 (L5–L11) | 6,187 | 14.2 min | 41,409 | 1,380 |
| L8 | 3 (L9–L11) | 6,114 | 5.9 min | 11,361 | 379 |
| Feature | Biology | Out-Degree |
|---|---|---|
| L4_F446 | Endonucleolytic Cleavage (rRNA) | 6,494 |
| L4_F1643 | Aerobic Electron Transport Chain | 6,050 |
| L0_F552 | NADH Dehydrogenase Complex Assembly | 4,785 |
| L0_F590 | NADH Dehydrogenase Complex Assembly | 3,849 |
| L0_F233 | Aerobic Electron Transport Chain | 3,420 |
| L0_F880 | Golgi Organization | 3,133 |
| Source | Target | Shared | Biological Theme | |
| Stress response cascade | ||||
| L0 Protein Catabolism | L1 Chromatin Org. | 161 | -catenin degrad. | |
| L0 Protein Catabolism | L2 Chemical Stress | 152 | Stress detection | |
| L0 Protein Catabolism | L4 DNA Metabolism | 158 | ER-phagosome | |
| L4 DNA Metabolism | L6 Macromol. Biosyn. | 153 | Biosynthetic recovery | |
| Ubiquitin-proteasome persistence | ||||
| Metric | Value | Significance |
| Total causal edges | 96,892 | Across 4 conditions |
| Both-annotated edges | 37,088 | 38.3% annotation rate |
| Unique domain pairs | 16,067 | 1,126 unique domains |
| Cross-model consensus | ||
| Consensus pairs (GF scGPT) | 1,142 | 10.6 enrichment |
| High-confidence ( both) | 303 | (perm.) |
| Disease Category | Domains | Circuit Edges | Consensus | Mean |
|---|---|---|---|---|
| Transcription regulation | 739 | 28,155 | 7,046 | 1.11 |
| Immune response | 660 | 27,071 | 7,305 | 1.10 |
| Apoptosis | 631 | 25,622 | 6,614 | 1.11 |
| Cell cycle (cancer) | 577 | 25,180 | 6,610 | 1.12 |
| Protein quality control | 576 | 21,896 | 6,256 | 1.10 |
| DNA damage/repair | 532 | 22,233 | 6,010 | 1.12 |
Mechanistic Interpretability
Mechanistic interpretability studies how neural networks implement behaviors through internal features, circuits, and causal information flow.
See all 367 papers