Characterizing the Relationship Between Training Data Mixtures and Emergent LLM Abilities
Characterize the complex, high-dimensional relationship between the composition of training datasets for large language models—specifically the proportions and properties of sources such as web text, books, code, and scientific papers—and the emergent capabilities exhibited by models trained via next-token prediction, with the aim of guiding optimal data-mixture design under resource constraints.
References
While heuristic understanding exists—for instance, that a higher proportion of code in the training data generally leads to stronger coding abilities—the complex, high-dimensional relationship between data mixture and emergent abilities is largely unknown.
ULTRA's own pretraining study points the same way from the other side: varying the pretraining mix from one to eight graphs, its authors observe performance saturating beyond three and leave a more principled choice of mix to future work.
CPT experiments are expensive, so ours cover one base model (Qwen2.5-7B) and one mixture (30B trajectories plus 20B general text). Whether the results carry over to larger models and other ratios is untested.
We leave a direct ablation of OLMo3, and the property responsible for this gap, for future work.
For decision granularity, what should the unit of decision be?
In addition, we compare only three distributions on the synthesized data, namely the baseline-level mixture, LongForge-1, and LongForge-N; they show that raising breadth and depth helps, but not how far to raise them or how much each contributes, which we leave to a more systematic mixture design.