Search papers, labs, and topics across Lattice.
This study formalizes and quantifies the concept of data synergy in language model pretraining, revealing that the interaction between datasets from different domains can significantly influence model performance. By analyzing open-weight LLMs with various pretraining mixtures, the authors estimate both direct domain-to-benchmark synergy and second-order domain-domain synergy, demonstrating that the combined effect of certain datasets can either enhance or hinder performance. The findings indicate that predictive accuracy can be improved by considering these synergies, surpassing traditional domain-agnostic scaling laws.
Data synergy can either amplify or diminish model performance, revealing that the right dataset combinations are crucial for optimal language model training.
Machine learning progress is often attributed to scaling model size and dataset volume, yet the composition of data can be just as consequential. Empirical findings repeatedly show that combining datasets from different domains yields nontrivial interactions. For instance, adding code improves mathematical reasoning, while certain mixtures introduce interference that reduces model performance. We refer to these effects collectively as data synergy, where the contribution of multiple domains exceeds or falls short of the sum of their isolated contributions. In this work, we formalize and quantify data synergy in language model pretraining. Leveraging observational variation across open-weight LLMs with diverse pretraining mixtures, we estimate both direct domain-to-benchmark synergy (how one domain contributes to performance on another) and a second-order domain-domain synergy (capabilities that require co-occurrence of multiple domains). Our framework improves predictive accuracy over domain-agnostic scaling laws and recovers stable synergy estimates. We validate these estimates by training models on predicted optimal and predicted anti-optimal mixtures and confirm that our synergy estimates correctly predict performance rankings.