中文

fMRI 中语言编码模型的缩放定律

计算与语言 2024-01-31 v4 人工智能

摘要

基于 Transformer 的单向语言模型的表征已知能有效预测大脑对自然语言的响应。然而,多数将语言模型与大脑比较的研究使用了 GPT-2 或类似规模的语言模型。在此,我们测试了更大的开源模型(如 OPT 与 LLaMA 系列)是否能更好地预测 fMRI 记录的大脑响应。呼应其他场景中的缩放结果,我们发现大脑预测性能随模型规模从 1.25 亿到 300 亿参数呈对数增长,在 3 名被试的留出测试集上相关性衡量显示编码性能提升约 15%。缩放 fMRI 训练集规模时观察到类似的对数行为。我们还刻画了使用 HuBERT、WavLM 与 Whisper 的声学编码模型的缩放规律,并发现随模型规模有可比的改进。对这些大型高性能编码模型的噪声上限分析表明,性能正接近楔前叶与高级听觉皮层等脑区的理论最大值。这些结果表明,扩大模型与数据规模将产生极其有效的脑内语言处理模型,促成更好的科学理解以及诸如解码等应用。

关键词

引用

@article{arxiv.2305.11863,
  title  = {Scaling laws for language encoding models in fMRI},
  author = {Richard Antonello and Aditya Vaidya and Alexander G. Huth},
  journal= {arXiv preprint arXiv:2305.11863},
  year   = {2024}
}

备注

Accepted to the Thirty-seventh Annual Conference on Neural Information Processing Systems (NeurIPS 2023). Please cite NeurIPS version