New paper: LLM data schedules and learning rate schedules should be co-designed

jasondeanlee · x · 2026-10-27

A new paper, "Data Schedules and Learning Rate Schedules Should Be Designed Together," treats pretraining + midtraining as one run with a data schedule and shows the optimal timing for midtraining data (e.g. math) depends on the LR schedule. At fixed compute and midtraining token budget, earlier mixing yields lower midtraining loss when LR decays throughout; with constant LR, placing all midtraining data at the end works better. Varying when decay begins, the best tested strategy keeps LR high after midtraining data enters, then decays near the end.

Related event: New Paper Argues Data Schedules and Learning Rate Schedules Should Be Co-Designed(3 posts)→

Original post →

More from Research

Research channel →