Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Curriculum-Guided Layer Scaling for Language Model Pretraining

About

As the cost of pretraining large language models grows, there is continued interest in strategies to improve learning efficiency during this core training stage. Motivated by cognitive development, where humans gradually build knowledge as their brains mature, we propose Curriculum-Guided Layer Scaling (CGLS), a framework for compute-efficient pretraining that synchronizes increasing data difficulty with model growth through progressive layer stacking (i.e. gradually adding layers during training). At the 100M parameter scale, using a curriculum transitioning from synthetic short stories to general web data, CGLS outperforms baseline methods on the question-answering benchmarks PIQA and ARC. Pretraining at the 1.2B scale, we stratify the DataComp-LM corpus with a DistilBERT-based classifier and progress from general text to highly technical or specialized content. Our results show that progressively increasing model depth alongside sample difficulty leads to better generalization and zero-shot performance on various downstream benchmarks. Altogether, our findings demonstrate that CGLS unlocks the potential of progressive stacking, offering a simple yet effective strategy for improving generalization on knowledge-intensive and reasoning tasks.

Karanpartap Singh, Neil Band, Ehsan Adeli• 2025

Related benchmarks

TaskDatasetResultRank
Question AnsweringPIQA
Accuracy67.36
589
Question AnsweringARC Easy
Accuracy46.97
246
Multiple-choice Question AnsweringHellaSwag
Accuracy36.8
212
Language ModelingThe Pile
Perplexity22.99
132
Language ModelingLAMBADA
Accuracy34.08
50
SummarizationTLDR9+
TLDR9+ Score3.568
6
Long-context Question AnsweringInfiniteBench long-context EN.QA
Accuracy53.9
3
Question AnsweringMC-QA
Average Accuracy42.3
3
Text RewritingOpenRewriteEval
Accuracy17.03
3
Showing 9 of 9 rows

Other info

Follow for update