TL;DR
Data mixture experiments for pre-training often fail when high-quality data is limited, leading to inaccurate extrapolations. A key issue identified is repetition mismatch, where the frequency of data usage changes as the training budget increases.
✦ Why It Matters
Engineers should adjust data mixtures based on repetition rates to improve model training outcomes.
Key Takeaways
Full Summary
Pre-training models typically rely on data mixtures, but when high-quality datasets are scarce, small-scale experiments can mislead researchers about the optimal data mixture for larger training budgets. The study identifies 'repetition mismatch' as a significant factor; as the training budget grows, the rate at which high-quality data is repeated alters, affecting the mixture's effectiveness.
Researchers conducted experiments to analyze how different repetition rates impact model performance. They found that using a fixed mixture from small-scale tests often leads to suboptimal results in larger settings.
The findings suggest that adjusting the data mixture based on expected repetition rates can enhance model performance. This has implications for engineers and researchers, as it emphasizes the need for careful planning in data selection and mixture strategies.
Related