Title: BEYOND IMITATION: RECOVERING DENSE REWARDS FROM DEMONSTRATIONS
Primary area: foundation or frontier models, including LLMs
Keywords: LLM, Reinforcement Learning, SFT, Inverse Reinforcement Learning

Abstract: 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.

TL;DR: We prove that SFT is equivalent to IRL, which allows us to extract a dense, token-level reward signal directly from an SFT model and use it to significantly improve the model's performance through further fine-tuning.