论文精选73°

大语言模型同策略蒸馏研究:单训练示例效果显著

Rethinking On-Policy Distillation of Large Language Models II: One Training Example

精选理由

OpenAI新研究揭示OPD只需少量数据就能接近全数据效果,算法效率比数据量更重要。

AI 摘要

该研究探讨了同策略蒸馏(OPD)中训练数据的作用,发现在仅使用单个查询的情况下,OPD仍能持续数百步改进,并恢复全数据OPD的大部分增益。单个查询已达到71.5%的状态覆盖率,其中大部分在前100步内完成。添加语义不同的查询可提高覆盖率和验证准确率,16个查询即可达到98.9%并匹配全数据训练。

原文 · arXiv cs.AI

Rethinking On-Policy Distillation of Large Language Models II: One Training Example

On-policy distillation (OPD) combines student-generated rollouts with dense token-level supervision from a teacher. Existing work has mainly studied its algorithmic behavior, leaving the role of training data unclear. We examine this role at the data-minimal limit by training on a single query. One-shot OPD keeps improving for hundreds of steps and recovers most of full-data OPD's gain across task domains and model families. We explain this result through the states visited during training and the rate at which the student aligns with the teacher. We measure \emph{state coverage}, the fraction of the states full-data OPD visits that a query set's rollouts reach. A single query already reaches \(71.5\%\), most of it within the first 100 steps. Adding semantically distinct queries raises coverage and validation accuracy together, until 16 queries reach \(98.9\%\) and match full-data training. Yet alignment slows at a similar pace whether OPD trains on one query or the whole dataset, and even a fixed set of states takes hundreds of steps to absorb. OPD is therefore data-overfed but algorithm-starved. Its rollouts quickly expose broad supervision, while the student absorbs that supervision increasingly slowly. The state-coverage result extends to multi-teacher OPD, where 16 semantically diverse queries per domain match full-data MOPD. As a further stress test, content-light templates and off-domain WildChat queries also approach the real-query baseline. Task content and induced state coverage can therefore come apart. We hope these findings direct future work toward the step efficiency of OPD, and prompt a re-examination of the data and the mechanisms behind its recent successes in frontier post-training.