Systematic Evaluation of Single-Cell Foundation Model Interpretability Reveals Attention Captures Co-Expression Rather Than Unique Regulatory Signal
Organizations: Department of Computer Science, University of Tübingen, Tübingen, Germany
Abstract
We present a systematic evaluation framework - thirty-seven analyses, 153 statistical tests, four cell types, two perturbation modalities - for assessing mechanistic interpretability in single-cell foundation models. Applying this framework to scGPT and Geneformer, we find that attention patterns encode structured biological information with layer-specific organisation - protein-protein interactions in early layers, transcriptional regulation in late layers - but this structure provides no incremental value for perturbation prediction: trivial gene-level baselines outperform both attention and correlation edges (AUROC 0.81-0.88 versus 0.70), pairwise edge scores add zero predictive contribution, and causal ablation of regulatory heads produces no degradation. These findings generalise from K562 to RPE1 cells; the attention-correlation relationship is context-dependent, but gene-level dominance is universal. Cell-State Stratified Interpretability (CSSI) addresses an attention-specific scaling failure, improving GRN recovery up to 1.85x. The framework establishes reusable quality-control standards for the field.
Figures & tables
Appendix figures & tables53 assets
Supplementary material from the paper’s appendix.
Appendix
| Analysis | Edge Type | Dataset | Key AUROC |
|---|---|---|---|
| Scaling behavior | Attention | scGPT kidney | 0.86–0.93 |
| Controlled composition | Correlation | TS kidney | 0.52–0.56 |
| Residualization | Correlation | DLPFC brain | 0.73 (resid.) |
| Degree-preserving null | Correlation | DLPFC brain | 0.69 (null) |
| Perturbation-first (corr.) | Correlation | Replogle K562 | 0.696 |
| Perturbation-first (attn.) | Attention | Replogle K562 | 0.704 |
| Dataset | Modality / Config | DE Thresh. | Notes | |
|---|---|---|---|---|
| Replogle K562 | CRISPRi (baseline) | LFC 0.1 | 44 | Mann-Whitney, HVG=1000, =500 |
| Replogle K562 | CRISPRi (primary) | LFC 0.5 | 151 | Welch , HVG=2000, =2000 |
| Replogle K562 | CRISPRi (attention) | LFC 0.5 | 280 | Welch , HVG=2000; more genes evaluable with attention tokenization |
| Adamson K562 | CRISPRa | LFC 0.5 | 77 | K562, activation modality |
| Shifrut T cells | CRISPRi | LFC 0.1 | 7 | Primary human T cells; weak DE effects require lenient threshold |
| Replogle RPE1 | CRISPRi | LFC 0.5 | 1,251 | hTERT-RPE1; perturbation genes forced into HVG |
| Tier | Jaccard (200) | Jaccard (1,000) | Spearman (200) | Spearman (1,000) |
|---|---|---|---|---|
| Small | 0.572 | 0.305 | 0.211 | -0.065 |
| Medium | 0.562 | 0.293 | 0.183 | -0.110 |
| Large | 0.536 | 0.285 | 0.083 | -0.130 |
| Top | Observed | Expected | Fold |
|---|---|---|---|
| 100 | 26 | 0.1 | 484 |
| 500 | 153 | 1.3 | 114 |
| 1,000 | 289 | 5.4 | 54 |
| 5,000 | 1,094 | 134.2 | 8.2 |
| Config | States | Pooled F1 | CSSI-max F1 | Ratio | |
|---|---|---|---|---|---|
| Small | 200 | 2 | 1.13 | ||
| Medium | 400 | 4 | 1.40 | ||
| Large | 600 | 6 | 1.85 | ||
| XLarge | 1000 | 8 | 1.76 | ||
| XXLarge | 1000 | 10 | 1.81 | ||
| Massive | 1500 | 12 | 1.79 |
| Pooled F1 | CSSI-max F1 | AUROC | AUROC | |
|---|---|---|---|---|
| 200 | 0.860 | 0.935 | ||
| 500 | 0.932 | 0.998 | ||
| 1,000 | 0.972 | 1.000 | ||
| 2,000 | 0.989 | 1.000 | ||
| 5,000 | 1.000 | 1.000 |
| Layer | Pooled | Best CSSI | AUROC | |
|---|---|---|---|---|
| 0 | 0.552 | cssi_mean | 0.566 | +0.014 |
| 1 | 0.600 | cssi_mean | 0.608 | +0.007 |
| 2 | 0.564 | cssi_mean | 0.587 | +0.022 |
| 3 | 0.573 | cssi_mean | 0.580 | +0.007 |
| 4 | 0.610 | cssi_range | 0.609 | -0.001 |
| 5 | 0.532 | cssi_mean | 0.551 | +0.019 |
| Cells | Edges | TRRUST AUROC | DoRothEA AUROC |
|---|---|---|---|
| 200 | 1.56M | 0.444 | 0.473 |
| 500 | 3.27M | 0.549 | 0.486 |
| 1000 | 5.38M | 0.522 | 0.486 |
| TRRUST AUROC | DoRothEA AUROC | |||
|---|---|---|---|---|
| Cells | scGPT | Geneformer | scGPT | Geneformer |
| 200 | 0.51 | 0.444 | 0.50 | 0.473 |
| 500 | 0.49 | 0.549 | 0.48 | 0.486 |
| 1000 | 0.46 | 0.522 | 0.47 | 0.486 |
| Section | Hypothesis | Test | Raw p | BH p | Effect | N | Sig |
|---|---|---|---|---|---|---|---|
| 1. Scaling Behavior Analysis | |||||||
| Scaling TRRUST | Cell count improves GRN | Sign test | 0.002 | 0.011 | 100% degrad. | 9 runs | Yes |
| Scaling TRRUST | Cell count improves GRN | Wilcoxon | 0.002 | 0.011 | – | 9 runs | Yes |
| Scaling DoRothEA | Cell count improves GRN | Sign test | 0.002 | 0.011 | 100% degrad. | 9 runs | Yes |
| Bootstrap CI | F1 confidence intervals | Bootstrap | – | – | – | 10k res. | – |
| Robustness | Seed stability w/ scaling | Paired test | 0.001 | 0.003 | 46–48% drop | 3 seeds | Yes |
| Headline Claim | Supporting Analysis | Key Test(s) | BH Sig |
|---|---|---|---|
| Top- scaling degradation | Scaling | Sign test (TRRUST) | Yes |
| Continuous AUROC improves | K-Sensitivity | Comparative | Descr. |
| Mediation non-additivity | Mediation Bias | Lower bound | Yes |
| Detectability theory validated | Detectability | Correlation | Yes |
| Perturbation-first AUROC 0.5 | Pert.-first | -test, all 27 cond. | Yes |
| Attn Corr on CRISPRi | Attn pert.-first | Wilcoxon | No (null) |
| Layer | STRING ≥700 | STRING ≥900 | TRRUST | Reactome | KEGG | GO BP |
|---|---|---|---|---|---|---|
| L0 | 0.640 | 0.644 | 0.558 | 0.505 | 0.530 | 0.526 |
| L1 | 0.517 | 0.501 | 0.691 | 0.501 | 0.484 | 0.501 |
| L4 | 0.546 | 0.530 | 0.686 | 0.516 | 0.525 | 0.515 |
| L8 | 0.530 | 0.519 | 0.631 | 0.509 | 0.531 | 0.534 |
| L13 | 0.574 | 0.559 | 0.708 | 0.523 | 0.548 | 0.530 |
| L15 | 0.525 | 0.500 | 0.750 | 0.521 | 0.555 | 0.541 |
| Metric | Best VW Cosine | Best Raw Attn | Correlation | Variance |
|---|---|---|---|---|
| Pert-first AUROC | 0.587 (L12) | 0.787 (L14) | 0.706 | 0.887 |
| TRRUST AUROC | 0.606 (L12) | 0.718 (L15) | 0.638 | — |
| TF | Category | Targets in HVG | AUROC | Successful |
|---|---|---|---|---|
| LEF1 | master regulator | 1 | 0.997 | Yes |
| KLF1 | master regulator | 1 | 0.995 | Yes |
| PPARG | master regulator | 10 | 0.900 | Yes |
| FOXA1 | master regulator | 2 | 0.874 | Yes |
| MAL | signal-dependent | 1 | 0.860 | Yes |
| EGR1 | signal-dependent | 9 | 0.852 | Yes |
Mechanistic Interpretability
Mechanistic interpretability studies how neural networks implement behaviors through internal features, circuits, and causal information flow.
See all 391 papers