Abstraction Induces the Brain Alignment of Language and Speech Models

· Source: cs.CL updates on arXiv.org · Field: Science & Research — Artificial Intelligence & Machine Learning, Life Sciences & Biology, Mathematics & Computational Sciences · Depth: Expert, extended

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

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

Topics

Code references

Best for: AI Scientist, Research Scientist, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.