Preprint
Reinforcement Learning

The Value of RL in Fine-Tuning

Gokul Swamy, Sanjiban Choudhury, Wen Sun, Zhiwei Wu, J. Bagnell
March 3, 2025arXiv.org62 citations

62

Citations

2

Influential Citations

arXiv.org

Venue

2025

Year

Abstract

From a first-principles perspective, it may seem odd that the strongest results in foundation model fine-tuning (FT) are achieved via a relatively complex, two-stage training procedure. Specifically, one first trains a reward model (RM) on some dataset (e.g., human preferences) before using it to provide online feedback as part of a downstream reinforcement learning (RL) procedure, rather than directly optimizing the policy parameters on said dataset via offline maximum likelihood estimation. In fact, from an information-theoretic perspective, we can only lose information via passing through a reward model and cannot create any new information via on-policy sampling. To explain this discrepancy, we scrutinize several hypotheses on the value of RL in FT through both theoretical and empirical lenses. Of the hypotheses considered, we find the most support for the explanation that on problems with a generation-verification gap, (1) it is relatively easy to learn the relatively simple RM (verifier) from the preference data. Then, (2) the downstream RL procedure only returns policies (generators) that are optimal for such relatively simple verifiers. Thus, end-to-end, two-stage online FT only has to search over a reduced subset of the full space of policies, requiring less data than offline FT.

Analysis

Why This Paper Matters

This paper addresses a fundamental puzzle in modern AI: why does the complex two-stage RL fine-tuning pipeline outperform simpler offline maximum likelihood estimation (MLE) when information theory suggests the latter should be at least as good? The authors challenge the conventional wisdom by dissecting the pipeline from a first-principles perspective, offering a compelling explanation that could reshape how practitioners approach fine-tuning.

The significance lies in its potential to demystify a widely used but poorly understood process. By identifying the generation-verification gap as the key factor, the paper provides a theoretical foundation for why RL is not just a heuristic but a principled approach. This insight could lead to more efficient training methods, reducing the computational and data costs associated with fine-tuning large models.

Technical Contributions

  • Information-Theoretic Analysis: The paper rigorously analyzes the information flow in the two-stage pipeline, showing that reward modeling can only lose information and on-policy sampling cannot create new information.
  • Hypothesis Testing: It systematically evaluates several hypotheses for RL's value, including the generation-verification gap, and provides evidence for the most plausible one.
  • Reduced Policy Space: The key contribution is the formalization that RL only searches over policies that are optimal for the learned verifier, which is a simpler function than the full policy space, thus requiring less data.
  • Theoretical and Empirical Blend: The authors combine theoretical arguments with empirical experiments to validate their claims, strengthening the credibility of their findings.

Results

The paper does not report specific numerical metrics but rather presents a qualitative and theoretical result. The main finding is that the generation-verification gap hypothesis is the most supported among those considered. The authors demonstrate that when there is a gap between generating and verifying outputs, learning a simple verifier from preference data is easy, and the subsequent RL optimization only needs to find policies that are optimal for that verifier. This reduces the search space and data requirements compared to offline FT, which must optimize the full policy space directly.

While no concrete numbers are given, the theoretical argument is backed by empirical observations from the literature, suggesting that RL fine-tuning is particularly effective in tasks like code generation and mathematical reasoning where verification is easier than generation.

Significance

This paper has broad implications for the AI community. It provides a clear rationale for the success of RL fine-tuning, which could lead to more targeted improvements in training pipelines. For instance, if the generation-verification gap is the key, then designing verifiers that are even simpler or more aligned with the task could further enhance efficiency.

Moreover, the work encourages a deeper understanding of why certain training paradigms work, moving beyond empirical success to theoretical grounding. This could inspire new algorithms that explicitly exploit the reduced policy space, potentially leading to faster convergence and better generalization. Ultimately, this paper contributes to a more principled foundation for fine-tuning large language models and other foundation models.