Papers
Topics
Authors
Recent
Search
2000 character limit reached

Mitigating Distribution Sharpening in Math RLVR via Distribution-Aligned Hint Synthesis and Backward Hint Annealing

Published 9 Apr 2026 in cs.AI, cs.CL, and cs.LG | (2604.07747v1)

Abstract: Reinforcement learning with verifiable rewards (RLVR) can improve low-$k$ reasoning accuracy while narrowing solution coverage on challenging math questions, and pass@1 gains do not necessarily translate into better large-$k$ performance. Existing hint-based approaches can make challenging questions trainable, but they leave two issues underexplored: teacher-student distribution mismatch and the need to reduce hint exposure to match no-hint evaluation. We address these issues through two components. Distribution-Aligned Hint Synthesis (DAHS) constructs verified teacher hints conditioned on student-style responses. Backward Hint Annealing (BHA) anneals hint exposure across difficulty buckets and uses per-question hint dropout to preserve no-hint updates throughout RL training. We evaluate the method in math RLVR under the DAPO training framework across AIME24, AIME25, and AIME26 using $\texttt{Qwen3-1.7B-Base}$ and $\texttt{Llama-3.2-1B-Instruct}$. On $\texttt{Qwen3-1.7B-Base}$, our method improves both pass@1 and pass@2048 relative to DAPO across the three AIME benchmarks. On $\texttt{Llama-3.2-1B-Instruct}$, the gains are concentrated in the large-$k$ regime. These results suggest that, in math RLVR, hint scaffolding is effective when it restores learnable updates on challenging questions early in training and is then gradually removed before no-hint evaluation.

Summary

  • The paper introduces DAHS, which synthesizes hints aligned with student-style responses to correct teacher-student distribution mismatches.
  • It details BHA, employing adaptive hint dropout and length-bucketed annealing to maintain training signals and balance scaffold exposure.
  • Experimental results on AIME benchmarks demonstrate improved pass@1 and pass@2048 metrics, indicating enhanced solution coverage with lower computational overhead.

Mitigating Distribution Sharpening in Math RLVR via Distribution-Aligned Hint Synthesis and Backward Hint Annealing

Problem Statement and Motivation

Reinforcement Learning with Verifiable Rewards (RLVR) has become a dominant approach for improving LLM-based mathematical reasoning. However, a persistent failure mode emerges: under group-relative policy updates, challenging problems fail to provide informative updates for extended periods, yielding training dynamics where easier questions dominate, and effective trajectory support narrows. This phenomenon, termed "distribution sharpening", leads to improved pass@1 but does not translate to broader solution coverage under large-k evaluation regimes.

Existing hint-based methods scaffold learning on hard questions by providing teacher prefixes, yet current implementations inadequately address two central issues:

  • Teacher-student distribution mismatch: Teacher hints may introduce unnatural continuation contexts for the student, especially when model architectures diverge.
  • Hint exposure versus evaluation distribution: Training with hints results in a distributional gap relative to no-hint evaluation, risking over-reliance on scaffolding.

DAHS and BHA: Framework Overview

To counter the above, the paper introduces two key components:

  1. Distribution-Aligned Hint Synthesis (DAHS): Verified teacher hints are synthesized by conditioning the teacher on student-style responses, aligning the hint distribution to the student’s trajectory space.
  2. Backward Hint Annealing (BHA): Hint exposure is annealed across length-bucketed difficulty bands, with per-question hint dropout preserving no-hint updates throughout RL training and gradually transitioning the policy towards the evaluation distribution.

The full framework operates as shown in (Figure 1): Figure 1

Figure 1: DAHS produces hints aligned to student-style continuations, while BHA applies length-bucketed annealing and per-question hint dropout during RL training.

Technical Contributions and Methodology

DAHS constructs a single verified teacher hint per question, using student outputs as templates to ensure contextual alignment. The teacher solution is retained only if verified correct, avoiding low-probability regions under the student policy. Empirical analysis confirms DAHS outperforms non-aligned and pre-hint CoT approaches in log-probability alignment (Figure 2): Figure 2

Figure 2: Student-policy log-probabilities for various hint sources demonstrate superior DAHS alignment versus non-aligned or raw teacher CoT hints.

BHA solves the challenge of transferring from hint-augmented training to no-hint evaluation. Hint prefixes are constructed from the full teacher hint and truncated per length bucket by an annealed ratio τb\tau_b, which is adaptively decreased as bucket-level accuracy thresholds are met. Per-question hint dropout is applied, facilitating continual no-hint updates and enabling Go-Explore–like accumulation of policy-discovered correct trajectories.

The training loop employs suffix-only policy updates: only student-generated tokens after the hint prefix are updated, avoiding SFT-style losses and off-policy weighting.

Experimental Evaluation and Ablation Analysis

Comprehensive evaluation on AIME24/25/26 benchmarks, using Qwen3-1.7B-Base and Llama-3.2-1B-Instruct, is performed under the DAPO-Math-17k dataset. DAHS+BHA consistently improves both pass@1 and pass@2048 metrics relative to DAPO, with the most pronounced gains for Qwen3-1.7B-Base (Figure 3): Figure 3

Figure 3: Pass@kk curves for Qwen3-1.7B-Base, showing substantial DAHS+BHA gains across all kk.

BHA's length-bucketed annealing avoids premature removal of hint scaffolds for harder questions, preserving training signals (Figure 4): Figure 4

Figure 4: Panel C demonstrates BHA bucketed hint-ratio annealing pacing per question length, while Panel A shows improved transfer at no-hint evaluation.

Ablation studies confirm:

  • DAHS yields significant large-kk improvements (pass@2048 up by 6.6 points on multiple benchmarks).
  • BHA outperforms per-prompt search, attaining similar/better coverage with much less online rollout overhead.
  • Per-question hint dropout balances early hint-scaffolded learning and no-hint transfer, with intermediate dropout probability (p0=0.75p_0=0.75) being optimal (Figure 5). Figure 5

    Figure 5: Per-question hint-dropout ablation pass@kk curves further validate the trade-off between hint-exposure and transfer.

On Llama-3.2-1B-Instruct, gains are concentrated in the large-kk regime (Figure 6), corroborating effectiveness under smaller model capacity. Figure 6

Figure 6: Pass@kk curves for Llama-3.2-1B-Instruct demonstrating improved coverage primarily at high kk (large sampling budgets).

The work sits within a growing literature on RLVR and hint-scaffolding in LLMs, extending prior frameworks like BREAD and SCAF-GRPO. By focusing on suffix-only guidance, DAHS+BHA avoids pitfalls of distribution mismatch and over-dependence on teacher hints. The length-bucketed annealing draws explicit inspiration from curriculum learning and backward algorithm methodologies, merging automated context alignment and gradual reduction of scaffolds.

The empirical results directly align with analyses of distribution sharpening, reinforcing the principle that low-kk gains alone cannot guarantee broad solution coverage [yue2025does, he2025rewarding]. The explicit policy-level alignment and backward hint annealing offered here provide a practical pathway to maintaining exploration diversity and effective RLVR learning signals.

Implications and Future Directions

Practically, the DAHS+BHA recipe yields reliably improved mathematical reasoning in base LLMs and scalable RLVR pipelines, without the computational overhead of per-prompt search. Theoretically, it demonstrates that aligned context design and scaffold reduction are joint necessary conditions for high-coverage RLVR optimization. The framework supports broader solution coverage and restores learnability on previously hard, untrainable questions.

Potential future developments include extending DAHS+BHA to larger models and additional domains, integrating more sophisticated alignment measures, and further automating the curriculum by dynamic difficulty estimation. The promise of RLVR training that maintains diversity and adaptability across training and evaluation distributions is likely to stimulate additional research into exploration and scaffold management in scalable RL systems.

Conclusion

The paper rigorously addresses a concrete failure mode in math RLVR—distribution sharpening and stalling on hard questions—by introducing Distribution-Aligned Hint Synthesis and Backward Hint Annealing. The combined approach improves solution coverage throughout the sampling regime, with ablation analyses confirming the necessity of distribution alignment, the practical utility of length-bucketed schedules, and the value of per-question hint dropout. These findings establish DAHS+BHA as an effective strategy for countering distribution sharpening and restoring learning signals on challenging mathematical reasoning tasks.

Paper to Video (Beta)

No one has generated a video about this paper yet.

Whiteboard

No one has generated a whiteboard explanation for this paper yet.

Open Problems

We haven't generated a list of open problems mentioned in this paper yet.

Collections

Sign up for free to add this paper to one or more collections.