mlsanityData to 7 Oct 2026

Sparse Autoencoder Decomposition of Clinical Sequence Model Representations: Feature Complexity, Task Specialisation, and Mortality Prediction

arXiv 2605.04072 ↗First posted 13 Apr 2026cs.LG cs.AI cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Sparse autoencoders (SAEs) have been applied to large language models and protein language models, but not systematically to electronic health record (EHR) foundation models. We train TopK SAEs on FlatASCEND, a 14.5-million-parameter autoregressive clinical sequence model, at all 10 residual stream extraction points on INSPECT (outpatient) and MIMIC-IV (ICU). SAE decomposition reveals progressive abstraction across transformer depth: layer-0 features are near-perfect token detectors (45.7% singleton), while layer-6 features span approximately 30 token types across multiple clinical categories (0.5% singleton). Under full-sequence simple linear probes, SAE features outperform dense representations for discrete event prediction (mortality) while dense representations outperform for continuous magnitude prediction (length of stay) - a probe-level representational phenomenon that does not extend to clinically relevant leakage-safe windows, where dense representations match or exceed SAE features across all tested settings (eICU-CRD 48-hour AUC: SAE 0.871 versus dense 0.880; base model zero-shot, SAE dictionaries trained on eICU activations; MIMIC-IV: 0.836 versus 0.914; INSPECT 1-year/3-year: 0.697 versus 0.800). A delta-mode intervention method reduces SAE perturbation noise by 86x, enabling cleaner feature-level experiments, though the resulting perturbation effects are larger than random controls in 3 of 4 conditions but not formally significant. Feature reproducibility across random seeds is 21%, and individual features should be interpreted as illustrative rather than stable.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 11 of its 22 references that are in the map. First posted on arXiv 13 Apr 2026.

Builds on

11 of 22 references.

References · 11

  1. 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
  2. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
  3. 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
  4. 2024Scaling and evaluating sparse autoencoderscited by 530
  5. 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 368
  6. 2019BEHRT: Transformer for Electronic Health Recordscited by 258
  7. 2025Sparse Autoencoders Trained on the Same Data Learn Different Featurescited by 86
  8. 2023MOTOR: A Time-To-Event Foundation Model For Structured Medical Recordscited by 75
  9. 2024Learning biologically relevant features in a pathology foundation model using sparse autoencoderscited by 14
  10. 2025Building the EHR Foundation Model via Next Event Predictioncited by 5
  11. 2026Scaling Recurrence-aware Foundation Models for Clinical Records via Next-Visit Predictioncited by 3

Built on by

Not cited yet.