SFT不仅能模仿专家行为,还能从中恢复出细粒度的奖励信号。
Beyond Imitation: Recovering Dense Rewards from Demonstrations
- 将SFT视为逆Q学习的特例,揭示其隐含的奖励建模能力
- 通过基线相对奖励函数,直接从SFT模型中提取逐标记奖励
- 用恢复的奖励改进策略,在指令遵循任务上超越原始SFT
传统上,监督微调(SFT)被视为仅在示范数据集上模仿专家行为的简化版模仿学习。本文挑战这一观点,证明SFT目标是逆Q学习的一个特例,表明SFT不仅学习策略,还隐式地学习一个密集的、基于标记级别的奖励模型来解释专家示范。我们进一步提出一种基线相对奖励函数,可直接从SFT模型中恢复该密集奖励信号。该奖励模型提供每个生成标记的细粒度信用分配。我们以恢复的奖励为基础,应用强化学习进一步优化策略,提出Dense-Path REINFORCE方法,在指令遵循基准测试中持续优于原始SFT模型。本工作重新定义SFT不仅是策略模仿,更是一种强大的奖励学习机制,为利用专家示范开辟了新路径。
原文摘要 · Abstract (English)
Conventionally, supervised fine-tuning (SFT) is treated as a simple imitation learning process that only trains a policy to imitate expert behavior on demonstration datasets. In this work, we challenge this view by establishing a fundamental equivalence between SFT and Inverse Reinforcement Learning. We prove that the SFT objective is a special case of Inverse Q-Learning, which implies that the SFT process does not just learn a policy, but also an implicit, dense, token-level reward model that explains the expert demonstrations. We then show how to recover this dense reward signal directly from the SFT model by formulating a baseline-relative reward function. The availability of such a dense reward model offers numerous benefits, providing granular credit assignment for each token generated. We demonstrate one key application by using these recovered rewards to further improve the policy with reinforcement learning. Our method, Dense-Path REINFORCE, consistently outperforms the original SFT models on instruction-following benchmarks. This work reframes SFT not merely as policy imitation but as a powerful reward learning mechanism, opening new possibilities for leveraging expert demonstrations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。