DEVOURED
Scaling Data Repetition for LLMs
High-quality data can be repeated more frequently as models grow, allowing smaller proxy models to accurately predict the ideal data mixture for massive training runs.
What: Researchers found that the optimal repetition of domain-specific training data correlates strongly with the model's eventual validation loss, rather than the amount of unique data available. This allows developers to use smaller models to optimize expensive, large-scale training recipes.
Why it matters: This finding provides a practical strategy to manage data scarcity, enabling teams to get more value out of limited, high-quality expert datasets.
Decoder
- Tokens-per-parameter ratio (TPP): A measure used to define the scale of a model relative to the amount of data it is trained on; maintaining this balance is crucial for model performance.
Original article
Scaling Domain Data Repetition in LLM Pretraining
As large language models scale, their training-token budgets must also increase to maintain an appropriate tokens-per-parameter ratio (TPP). However, high-quality domain data is much harder to scale than general web data. As model size and the training-token budget increase, its fraction in the training mixture tends to decrease. Repeating the available high-quality data provides an effective way to counteract this dilution, but excessive repetition may lead to overfitting. We study this trade-off under practical LLM scaling, where the training-token budget grows proportionally with model size. For a fixed domain, we first find that, surprisingly at a fixed TPP, the optimal repetition count mildly increases with model size. Across different domains, we find that the optimal repetition count is strongly negatively correlated with the final validation loss of a domain: domains with lower loss can generally benefit from more repetitions. In contrast, the amount of unique domain data is only weakly related to the optimal repetition count. These findings suggest that repetition counts tuned on smaller proxy models with the same TPP can provide a practical estimate for larger models.
Explore all 43 articles in today's digest
Fresh dev news, AI-digested daily. Never miss a trend at devoured.fyi