Scaling laws for language encoding models in fMRI
Representations from transformer-based unidirectional language models are
known to be effective at predicting brain responses to natural language.
However, most studies comparing language models to brains have used GPT-2 or
similarly sized language models. Here we tested whether larger open-source
models such as those from the OPT and LLaMA families are better at predicting
brain responses recorded using fMRI. Mirroring scaling results from other
contexts, we found that brain prediction performance scales log-linearly with
model size from 125M to 30B parameter models, with ~15% increased encoding
performance as measured by correlation with a held-out test set across 3
subjects. Similar log-linear behavior was observed when scaling the size of the
fMRI training set. We also characterized scaling for acoustic encoding models
that use HuBERT, WavLM, and Whisper, and we found comparable improvements with
model size. A noise ceiling analysis of these large, high-performance encoding
models showed that performance is nearing the theoretical maximum for brain
areas such as the precuneus and higher auditory cortex. These results suggest
that increasing scale in both models and data will yield incredibly effective
models of language processing in the brain, enabling better scientific
understanding as well as applications such as decoding.
646af6d51556443f24b9ef13
Richard Antonello
extracted_confirmed
2023-05-22T09:35:01.363Z
646b354e36505117e229b773
/avatars/3a8c51bcb969585502667faf7a7b6496.svg
Richard Antonello
user
RAntonello1
646af6d51556443f24b9ef14
Aditya Vaidya
extracted_confirmed
2023-05-23T19:46:08.812Z
6204d06212ccba127e39aaa3
/avatars/a35c69528ae1470593d3eb1a4cdacb88.svg
Aditya Vaidya
user
avaidya
646af6d51556443f24b9ef15
Alexander G. Huth
claimed_verified
2023-05-22T18:01:13.171Z
6303d6633926de1f7ec2fc42
/avatars/4ce51df91ed381fd0b1f4d8f46b1e65a.svg
Alex Huth
user
alexhuth
Scaling laws for language encoding models in fMRI
Representations from transformer-based unidirectional language models are
known to be effective at predicting brain responses to natural language.
However, most studies comparing language models to brains have used GPT-2 or
similarly sized language models. Here we tested whether larger open-source
models such as those from the OPT and LLaMA families are better at predicting
brain responses recorded using fMRI. Mirroring scaling results from other
contexts, we found that brain prediction performance scales log-linearly with
model size from 125M to 30B parameter models, with ~15% increased encoding
performance as measured by correlation with a held-out test set across 3
subjects. Similar log-linear behavior was observed when scaling the size of the
fMRI training set. We also characterized scaling for acoustic encoding models
that use HuBERT, WavLM, and Whisper, and we found comparable improvements with
model size. A noise ceiling analysis of these large, high-performance encoding
models showed that performance is nearing the theoretical maximum for brain
areas such as the precuneus and higher auditory cortex. These results suggest
that increasing scale in both models and data will yield incredibly effective
models of language processing in the brain, enabling better scientific
understanding as well as applications such as decoding.
Larger transformer-based language models like OPT and LLaMA improve brain response prediction accuracy compared to smaller models, scaling log-linearly with both model size and fMRI training set size.
transformer-based unidirectional language models brain responses fMRI OPT LLaMA GPT-2 model size fMRI training set HuBERT WavLM Whisper acoustic encoding models noise ceiling precuneus higher auditory cortex