Instability in Downstream Task Performance During LLM Pretraining

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Nishida, Yuto, Isonuma, Masaru, Oda, Yusuke
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916991361613824
author Nishida, Yuto
Isonuma, Masaru
Oda, Yusuke
author_facet Nishida, Yuto
Isonuma, Masaru
Oda, Yusuke
contents When training large language models (LLMs), it is common practice to track downstream task performance throughout the training process and select the checkpoint with the highest validation score. However, downstream metrics often exhibit substantial fluctuations, making it difficult to identify the checkpoint that truly represents the best-performing model. In this study, we empirically analyze the stability of downstream task performance in an LLM trained on diverse web-scale corpora. We find that task scores frequently fluctuate throughout training, both at the aggregate and example levels. To address this instability, we investigate two post-hoc checkpoint integration methods: checkpoint averaging and ensemble, motivated by the hypothesis that aggregating neighboring checkpoints can reduce performance volatility. We demonstrate both empirically and theoretically that these methods improve downstream performance stability without requiring any changes to the training procedure.
format Preprint
id arxiv_https___arxiv_org_abs_2510_04848
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Instability in Downstream Task Performance During LLM Pretraining
Nishida, Yuto
Isonuma, Masaru
Oda, Yusuke
Computation and Language
When training large language models (LLMs), it is common practice to track downstream task performance throughout the training process and select the checkpoint with the highest validation score. However, downstream metrics often exhibit substantial fluctuations, making it difficult to identify the checkpoint that truly represents the best-performing model. In this study, we empirically analyze the stability of downstream task performance in an LLM trained on diverse web-scale corpora. We find that task scores frequently fluctuate throughout training, both at the aggregate and example levels. To address this instability, we investigate two post-hoc checkpoint integration methods: checkpoint averaging and ensemble, motivated by the hypothesis that aggregating neighboring checkpoints can reduce performance volatility. We demonstrate both empirically and theoretically that these methods improve downstream performance stability without requiring any changes to the training procedure.
title Instability in Downstream Task Performance During LLM Pretraining
topic Computation and Language
url https://arxiv.org/abs/2510.04848