PTPP-Aware Adaptation Scaling Laws: Predicting Domain-Adaptation Performance at Unseen Pre-Training Budgets

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Goffinet, Etienne, Bergsma, Shane, Sheinin, Avraham, Vassilieva, Natalia, Muhammad, Shaheer, Nakov, Preslav, Gosal, Gurpreet
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914116308828160
author Goffinet, Etienne
Bergsma, Shane
Sheinin, Avraham
Vassilieva, Natalia
Muhammad, Shaheer
Nakov, Preslav
Gosal, Gurpreet
author_facet Goffinet, Etienne
Bergsma, Shane
Sheinin, Avraham
Vassilieva, Natalia
Muhammad, Shaheer
Nakov, Preslav
Gosal, Gurpreet
contents Continual pre-training (CPT) for domain adaptation must balance target-domain gains with stability on the base domain. Existing CPT scaling laws typically assume a fixed pre-training budget, which limits their ability to forecast adaptation outcomes for models trained at different tokens-per-parameter (PTPP). We present \emph{PTPP-aware} adaptation scaling laws that make the pre-training budget an explicit variable, enabling accurate \emph{prediction} of adaptation loss at unseen \ptpp. On a multilingual setup (English/Arabic $\rightarrow$ French), PTPP-aware formulations trained on early stages (\ptpp{}=\{15,31\}) predict target loss at \ptpp{}=279 and outperform a PTPP-agnostic \dcpt{} transfer baseline on metrics (Huber-on-log, MAE$_\mathrm{rel}$, calibration slope); full diagnostics (RMSE, MAPE) are in the appendix. Beyond forecasting, we show a practical use case: planning replay ratios and adaptation token budgets that satisfy target and forgetting constraints under compute limits.
format Preprint
id arxiv_https___arxiv_org_abs_2510_23198
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle PTPP-Aware Adaptation Scaling Laws: Predicting Domain-Adaptation Performance at Unseen Pre-Training Budgets
Goffinet, Etienne
Bergsma, Shane
Sheinin, Avraham
Vassilieva, Natalia
Muhammad, Shaheer
Nakov, Preslav
Gosal, Gurpreet
Machine Learning
Artificial Intelligence
Computation and Language
Continual pre-training (CPT) for domain adaptation must balance target-domain gains with stability on the base domain. Existing CPT scaling laws typically assume a fixed pre-training budget, which limits their ability to forecast adaptation outcomes for models trained at different tokens-per-parameter (PTPP). We present \emph{PTPP-aware} adaptation scaling laws that make the pre-training budget an explicit variable, enabling accurate \emph{prediction} of adaptation loss at unseen \ptpp. On a multilingual setup (English/Arabic $\rightarrow$ French), PTPP-aware formulations trained on early stages (\ptpp{}=\{15,31\}) predict target loss at \ptpp{}=279 and outperform a PTPP-agnostic \dcpt{} transfer baseline on metrics (Huber-on-log, MAE$_\mathrm{rel}$, calibration slope); full diagnostics (RMSE, MAPE) are in the appendix. Beyond forecasting, we show a practical use case: planning replay ratios and adaptation token budgets that satisfy target and forgetting constraints under compute limits.
title PTPP-Aware Adaptation Scaling Laws: Predicting Domain-Adaptation Performance at Unseen Pre-Training Budgets
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2510.23198