跳到正文
HuggingFace Daily Papers·· 2 天前

用单一算法训练基础模型:统一训练框架将预训练、SFT 与 RL 合并

Training Foundation Models with a Single Algorithm

AI 导读

研究者提出“统一训练”框架,通过将 RL 目标推导为对模型生成目标分布的最大似然拟合,把预训练、midtraining、SFT 和 RL 全部纳入同一个基于似然的训练算法。该框架在单一训练器中混合静态数据与 RL 生成,在受控和真实实验中改善了精度-算力权衡与任务覆盖,同时保留更广泛能力,并为持续学习提供更好的起点。论文共 67 页、31 图、22 表。

正文 · 原文

View PDF HTML (experimental)

Abstract: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.
Comments: 67 pages, 31 figures, 22 tables
Subjects: Machine Learning (cs.LG)
Cite as: arXiv:2610.40564 [cs.LG]
  (or arXiv:2610.40564v1 [cs.LG] for this version)
  https://doi.org/10.48550/arXiv.2610.40564

arXiv-issued DOI via DataCite

Submission history

From: Yuda Song [view email]
[v1] Thu, 29 Oct 2026 17:42:05 UTC (836 KB)

来源:HuggingFace Daily Papers · arxiv.org