Enhancing Linguistic Competence of Language Models through Pre-training with Language Learning Tasks
Atsuki Yamaguchi, Maggie Mi, Nikolaos Aletras
TL;DR
The paper addresses the limited linguistic competence of decoder-based language models trained with standard next-token objectives by introducing L2T, a pre-training framework that interleaves 14 language-learning tasks with raw text to produce structured input-output signals. Through experiments on 500M and 1B parameter models under Disjoint and Shared data regimes, L2T yields substantial improvements in linguistic competence on BLiMP (up to 11.3 points, average around 2.8) while largely preserving general reasoning abilities, and it accelerates the acquisition of linguistic knowledge. Analysis shows the strongest gains arise in long-distance syntactic dependencies and island constraints, with some phenomena saturating or remaining challenging, indicating the need for targeted discourse-level objectives. The findings support a data-centric view where structured linguistic supervision during pre-training complements world-knowledge learning, and they highlight important trade-offs when balancing raw text and linguistic tasks, guiding future multilingual extensions and curriculum-based scaling.
Abstract
Language models (LMs) are pre-trained on raw text datasets to generate text sequences token-by-token. While this approach facilitates the learning of world knowledge and reasoning, it does not explicitly optimize for linguistic competence. To bridge this gap, we propose L2T, a pre-training framework integrating Language Learning Tasks alongside standard next-token prediction. Inspired by human language acquisition, L2T transforms raw text into structured input-output pairs to provide explicit linguistic stimulation. Pre-training LMs on a mixture of raw text and L2T data not only improves overall performance on linguistic competence benchmarks but accelerates its acquisition, while maintaining competitive performance on general reasoning tasks.
