Working Memory Constraints Scaffold Learning in Transformers under Data Scarcity

Pranava Madhyastha, Dagmar Adamcová


Abstract
We investigate the integration of human-like working memory constraints into the Transformer architecture and implement several cognitively inspired attention variants, including fixed-width windows based and temporal decay based attention mechanisms. Our modified GPT-2 models are trained from scratch on developmentally plausible datasets (10M and 100M words). Performance is evaluated on grammatical judgment tasks (BLiMP) and alignment with human reading time data. Our results indicate that these cognitively-inspired constraints, particularly fixed-width attention, can significantly improve grammatical accuracy especially when training data is scarce. These constrained models also tend to show a stronger alignment with human processing metrics. The findings suggest that such constraints may serve as a beneficial inductive bias, guiding models towards more robust linguistic representations, especially in data-limited settings.
Anthology ID:
2026.findings-acl.2133
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
43022–43038
Language:
URL:
https://preview.aclanthology.org/ingest-acl-workshops/2026.findings-acl.2133/
DOI:
Bibkey:
Cite (ACL):
Pranava Madhyastha and Dagmar Adamcová. 2026. Working Memory Constraints Scaffold Learning in Transformers under Data Scarcity. In Findings of the Association for Computational Linguistics: ACL 2026, pages 43022–43038, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
Working Memory Constraints Scaffold Learning in Transformers under Data Scarcity (Madhyastha & Adamcová, Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl-workshops/2026.findings-acl.2133.pdf
Checklist:
 2026.findings-acl.2133.checklist.pdf