当专家反馈有噪声时,在线蒸馏比离线模仿更优,可实现多项式样本复杂度。
Behavior Cloning is Not All You Need: The Optimality of On-Policy Distillation for Noisy Expert Feedback
- 提出噪声专家模型解释在线蒸馏优于离线模仿的机制
- 离线学习需指数级样本量,而在线方法仅需多项式依赖于时序长度
- 理论支持语言模型训练中使用在线蒸馏而非监督微调
模仿学习是序列决策系统中的自然范式,已成为理解大语言模型训练的主要框架。一个核心难题是:理论上离线模仿学习可实现无时域依赖且最优,但实践中在线方法如在线策略蒸馏(OPD)常优于离线方法如监督微调(SFT)。本文通过引入噪声专家模型解释该差距——学习者仅能访问专家策略的噪声版本,却希望达到干净专家的奖励表现。在此设定下,我们揭示离线与在线模仿学习的尖锐分离:离线学习从噪声轨迹中获取信息本质困难,为竞争干净专家,样本复杂度必须指数增长;而在线交互通过新型OPD变体,可实现对时域长度的多项式依赖。进一步证明,在专家噪声分布满足自然条件下,可获得无时域依赖的样本复杂度保证,尽管算法在策略类大小上损失统计效率。分析还导出一种常用的大语言模型训练损失函数。我们提供算法与下界,并将结果扩展至清洁专家确定性但未知扰动的现实场景,为语言模型训练中OPD优于SFT提供了理论基础。
原文摘要 · Abstract (English)
Imitation Learning is a natural framework for learning in sequential decision-making systems and has emerged as the dominant paradigm through which we understand language model training. A central puzzle is that, while in theory offline IL can be horizon-free and optimal, in practice online methods such as on-policy distillation often outperform offline methods such as supervised fine-tuning. We propose a noisy expert model to explain this gap, in which the learner only has access to a noisy version of the expert's policy, but wishes to compete against the reward achieved by a clean expert, motivated by the fact that in many applications, e.g. training language models to perform long chains of thought, the expert is often imperfect. In this setting, we show a sharp separation between offline and online IL. Offline learning from noisy trajectories is fundamentally hard: to compete with the clean expert, the sample complexity must grow exponentially, in contradistinction to the clean expert setting where no explicit horizon dependence exists. In contrast, we prove that online interaction with the noisy expert via a novel variant of OPD enables polynomial dependence on the horizon in general. We further show that, under a natural condition on the expert noise distribution, which we show to be necessary for any horizon-free sample complexity, one can obtain such a guarantee, although our proposed algorithm sacrifices statistical efficiency in its dependence on the size of the policy class. Our analysis leads to an alternative loss function that is commonly considered empirically for LM training. We further provide algorithms and lower bounds, and extend our results to the more realistic setting of unknown corruption when the clean expert is deterministic, thereby providing a theoretical foundation for why OPD can outperform SFT when training language models from imperfect teachers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。