Training Foundation Models with a Single Algorithm
Foundation models are typically trained in stages: pretraining, midtraining, supervised fine-tuning (SFT), and reinforcement learning (RL), with changes in both data and learning algorithms. We show that this algorithmic separation is not fundamental. By deriving suitable RL objectives as maximum-likelihood fitting to model-generated target distributions, we place all of these stages under a single likelihood-based training algorithm, unified training. This perspective turns the foundation-model training lifecycle into a problem of choosing which data to train on, in what proportions, and when, with conventional staged training as a special case. We instantiate this framework in a unified training system that mixes static data and RL generations within a single trainer. Across controlled and real-world experiments, unified training improves the accuracy-compute tradeoff and task coverage while preserving broader capabilities, and provides a stronger starting point for continual learning. This work lays the groundwork for a unified science of foundation-model training, connecting mathematical principles, training infrastructure, and the development of model capabilities.
Authors
- Daniel Khashabi (ORCID: https://orcid.org/0009-0009-7664-2230)
- Fahim Tajwar (ORCID: https://orcid.org/0000-0001-9257-6282)
- Jeff Schneider (ORCID: https://orcid.org/0000-0002-5080-9073)
- Andrea Zanette
- Ruslan Salakhutdinov (ORCID: https://orcid.org/0000-0002-3752-2756)
- Daman Arora
- James Andrew Bagnell
- Alvin Zhang
- Yuda Song
- Aarti Singh
Publication Details
- Journal
- arXiv (Cornell University)
- Published
- 2026-10-29
- DOI
- https://doi.org/10.48550/arxiv.2610.40564
- Primary Topic
- Domain Adaptation and Few-Shot Learning
- Type
- preprint