Improve Language Model and Brain Alignment via Associative Memory

Congchi Yin, Yongpeng Zhang, Xuyun Wen, Piji Li


Abstract
Associative memory engages in the integration of relevant information for comprehension in the human cognition system. In this work, we seek to improve alignment between language models and human brain while processing speech information by integrating associative memory. After verifying the alignment between language model and brain by mapping language model activations to brain activity, the original text stimuli expanded with simulated associative memory are regarded as input to computational language models. We find the alignment between language model and brain is improved in brain regions closely related to associative memory processing. We also demonstrate large language models after specific supervised fine-tuning better align with brain response, by building the Association dataset containing 1000 samples of stories, with instructions encouraging associative memory as input and associated content as output.
Anthology ID:
2025.findings-acl.55
Volume:
Findings of the Association for Computational Linguistics: ACL 2025
Month:
July
Year:
2025
Address:
Vienna, Austria
Editors:
Wanxiang Che, Joyce Nabende, Ekaterina Shutova, Mohammad Taher Pilehvar
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
986–999
Language:
URL:
https://preview.aclanthology.org/display_plenaries/2025.findings-acl.55/
DOI:
Bibkey:
Cite (ACL):
Congchi Yin, Yongpeng Zhang, Xuyun Wen, and Piji Li. 2025. Improve Language Model and Brain Alignment via Associative Memory. In Findings of the Association for Computational Linguistics: ACL 2025, pages 986–999, Vienna, Austria. Association for Computational Linguistics.
Cite (Informal):
Improve Language Model and Brain Alignment via Associative Memory (Yin et al., Findings 2025)
Copy Citation:
PDF:
https://preview.aclanthology.org/display_plenaries/2025.findings-acl.55.pdf