Revisiting the Scaling Properties of Downstream Metrics in Large Language Model Training

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Krajewski, Jakub, Shidani, Amitis, Busbridge, Dan, Wiseman, Sam, Ramapuram, Jason
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915664174776320
author Krajewski, Jakub
Shidani, Amitis
Busbridge, Dan
Wiseman, Sam
Ramapuram, Jason
author_facet Krajewski, Jakub
Shidani, Amitis
Busbridge, Dan
Wiseman, Sam
Ramapuram, Jason
contents While scaling laws for Large Language Models (LLMs) traditionally focus on proxy metrics like pretraining loss, predicting downstream task performance has been considered unreliable. This paper challenges that view by proposing a direct framework to model the scaling of benchmark performance from the training budget. We find that for a fixed token-to-parameter ratio, a simple power law can accurately describe the scaling behavior of log accuracy on multiple popular downstream tasks. Our results show that the direct approach extrapolates better than the previously proposed two-stage procedure, which is prone to compounding errors. Furthermore, we introduce functional forms that predict accuracy across token-to-parameter ratios and account for inference compute under repeated sampling. We validate our findings on models with up to 17B parameters trained on up to 350B tokens across two dataset mixtures. To support reproducibility and encourage future research, we release the complete set of pretraining losses and downstream evaluation results.
format Preprint
id arxiv_https___arxiv_org_abs_2512_08894
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Revisiting the Scaling Properties of Downstream Metrics in Large Language Model Training
Krajewski, Jakub
Shidani, Amitis
Busbridge, Dan
Wiseman, Sam
Ramapuram, Jason
Machine Learning
Artificial Intelligence
Computation and Language
While scaling laws for Large Language Models (LLMs) traditionally focus on proxy metrics like pretraining loss, predicting downstream task performance has been considered unreliable. This paper challenges that view by proposing a direct framework to model the scaling of benchmark performance from the training budget. We find that for a fixed token-to-parameter ratio, a simple power law can accurately describe the scaling behavior of log accuracy on multiple popular downstream tasks. Our results show that the direct approach extrapolates better than the previously proposed two-stage procedure, which is prone to compounding errors. Furthermore, we introduce functional forms that predict accuracy across token-to-parameter ratios and account for inference compute under repeated sampling. We validate our findings on models with up to 17B parameters trained on up to 350B tokens across two dataset mixtures. To support reproducibility and encourage future research, we release the complete set of pretraining losses and downstream evaluation results.
title Revisiting the Scaling Properties of Downstream Metrics in Large Language Model Training
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2512.08894