Abstraction Induces the Brain Alignment of Language and Speech Models
Summary
This research demonstrates that the alignment between large language models (LLMs), speech audio models, and human brain activity stems from shared meaning abstraction, not next-word prediction. The study found that intermediate hidden layers, rather than output layers, are most effective at predicting fMRI and ECoG signals. This effectiveness correlates strongly with a peak in the layerwise "intrinsic dimension" (I_d), a measure of feature complexity. Models like OPT (125m, 1.3b, 13b), Pythia (160m, 410m, 6.9b), WavLM, and Whisper exhibit this I_d peak in their middle layers, coinciding with the construction of higher-order linguistic features. Finetuning models to better predict brain activity causally increases both their I_d and semantic content, reinforcing that rich meaning abstraction is the primary driver of model-brain similarity.
Key takeaway
For AI Scientists developing language or speech models, recognize that optimizing solely for next-token prediction may not yield the most brain-aligned or semantically rich representations. You should focus on architectural designs and training objectives that foster high intrinsic dimension and meaning abstraction in intermediate layers. Consider finetuning models directly on neural response data to enhance their semantic content and improve their correspondence with human brain processing, moving beyond autoregressive loss limitations.
Key insights
Brain-model alignment in language and speech processing is driven by meaning abstraction, not next-token prediction.
Principles
- Intermediate model layers exhibit peak intrinsic dimension (I_d).
- High I_d correlates with rich semantic feature building.
- Next-token prediction layers are less brain-aligned.
Method
The study measured model-to-brain encoding performance, computed layerwise intrinsic dimension (I_d) using GRIDE, performed layerwise linguistic probing, and calculated next-token surprisal.
In practice
- Prioritize intermediate layers for brain-aligned representations.
- Monitor intrinsic dimension (I_d) to identify abstraction phases.
- Finetune models on neural data to enhance semantic richness.
Topics
- Language Models
- Speech Models
- Brain-Computer Interfaces
- Intrinsic Dimension
- Neural Encoding Models
- Semantic Abstraction
Code references
- sissa-data-science/DADApy
- aalok-sathe/surprisal
- HuthLab/encoding-model-scaling-laws
- facebookresearch/SentEval
Best for: AI Scientist, Research Scientist, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.