Paper: data schedules and learning rate schedules should be designed together

alifcoder · x · 2026-10-27

A new paper shows the optimal placement of limited math data in LLM pre-training flips with the learning rate schedule. At fixed compute and mid-training token budgets, earlier mixing yields lower MT loss when LR decays throughout training; with constant LR, placing all mid-training data at the end works better. Varying when decay begins, the best tested strategy keeps LR high after mid-training data enters and decays near the end — data and LR schedules should be co-designed.

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

Original post →

More from Research

Research channel →