Exploring the Limit of Outcome Reward for Learning Mathematical Reasoning¶
Source: converted from arXiv LaTeX source source/InternThinker-1/neurips_2024.tex.
$^*$ Equal contribution $\dag$ Corresponding author
-0.6cm
Abstract¶
Reasoning abilities, especially those for solving complex math problems, are crucial components of general intelligence. Recent advances by proprietary companies, such as o-series models of OpenAI, have made remarkable progress on reasoning tasks. However, the complete technical details remain unrevealed, and the techniques that are believed certainly to be adopted are only reinforcement learning (RL) and the long chain of thoughts. This paper proposes a new RL framework, termed , to pursue the performance limit that can be achieved through Outcome REwArd-based reinforcement Learning for mathematical reasoning tasks, where only binary outcome rewards are easily accessible. We theoretically prove that behavior cloning on positive trajectories from best-of-N (BoN) sampling is sufficient to learn the KL-regularized optimal policy in binary feedback environments. This formulation further implies that the rewards of negative samples should be further reshaped to ensure the gradient consistency between positive and negative samples. To alleviate the long-existing difficulties brought by sparse rewards in RL, which are even exacerbated by the partial correctness of the long chain of thought for reasoning tasks, we further apply a token-level reward model to sample important tokens in reasoning trajectories for learning. With , for the first time, a 7B model can obtain 94.0 pass@1 accuracy on MATH-500 through RL, being on par with 32B models. -32B also surpasses previous 32B models trained by distillation with 95.0 pass@1 accuracy on MATH-500. Our investigation also indicates the importance of initial policy models and training queries for RL. Code, models, and data will be released to benefit future researchhttps://github.com/InternLM/.
Introduction¶
Solving complex problems with reasoning capability forms one of the cornerstones of human cognition - a cognitive ability that artificial general intelligence must ultimately master [xu2025towards, zhong2024evaluation]. Among various problem domains, the mathematical problem emerges as a particularly compelling experimental paradigm for AI research [liu2023mathematical, matzakos2023learning, ying2024internlm, shao2024deepseekmath], owing to its relatively well-defined structure and availability of precise binary correctness feedback based on the verifiable final answers.
Recent advances in large language models (LLMs) have achieved remarkable progress in mathematical reasoning by the chain-of-thought technics [wei2022chain, wang2022self, kojima2022large], in which the LLMs are elicited to produce a series of intermediate reasoning steps before providing the final answers to the problem. However, as most of the capable models (, the o-series models by OpenAI [openai2024learning]) are developed by proprietary companies, there is no clear pathway to develop state-of-the-art reasoning models. Some recent work shows that distillation [huang2024o1, Slow_Thinking_with_LLMs_2] is sufficient to obtain high performance given the accessibility to existing best or near best AI models, reinforcement learning (RL) is believed to be a more fundamental approach and has exhibited potential [deepseekr1] to advance beyond the intelligence boundary of current AI models, using the most capable open-source foundation models (DeepSeek-V3-base [liu2024deepseek], inter alia).
However, fundamental challenges of sparse reward in RL persist and are even exacerbated in mathematical reasoning tasks that mainly rely on the chain of thought technics with LLMs [wei2022chain]: the evaluation of intermediate reasoning steps is labor intensive [lightman2023let] and its accurate automation approach is still under-explored, thus, the only reliable reward is based on the outcome (correctness of final answer), which is inherently binary and sparse when faced with more than 2000 tokens in the long reasoning trajectories [deepseekr1, team2025kimi]. Existing approaches have attempted to estimate the advantages or values of reasoning steps by search [wang2024math, kazemnejad2024vineppo] or value function-based credit assignment [schulman2017proximal, cui2025process], yet, their performance remains unsatisfactory in comparison with the distilled models [deepseekr1].
This paper aims to conquer the above challenges and proposes a simple framework, termed , to push the limit of Outcome REwArd-based reinforcement Learning for mathematical reasoning tasks. is grounded in the unique characteristics of mathematical reasoning tasks that binary outcome feedback creates an environment where all positive trajectories are equally valid. We first establish that behavior cloning on BoN-sampled positive trajectories is sufficient to achieve KL-regularized optimality, which emerges from the analysis that the positive trajectory from BoN sampling converges to a distribution independent of the sample number. For learning on negative samples, reveals the necessity of reward shaping to maintain consistent gradient estimation between sampling and target distributions. Such a mechanism compensates for BoN's under-sampling of negative gradients, and enables difficulty-adaptive optimization over both successful and failed trajectories.
Another intrinsic property of mathematical reasoning tasks is the partial correctness in long reasoning chains, which further imposes the learning difficulty of sparse rewards when only a binary outcome reward is available at each iteration of RL training. Thus, adopts a lightweight credit assignment scheme through a token-level reward model trained using outcome rewards. This mechanism automatically estimates step-wise importance weights by decomposing trajectory advantages, enabling focused learning of critical reasoning steps or errors. The integration of these components yields a theoretically sound framework that effectively bridges the gap between sparse binary feedback and dense policy optimization requirements for mathematical reasoning tasks.
Figure fig: main_fig: Overall performance between -32B and some competitive baselines.
Extensive experimental results show that effectively improves the mathematical reasoning capability of LLMs. At the 7B parameter scale, to the best of our knowledge, -7B is the first to obtain the pass@1 accuracy on MATH-500 [hendrycks2021measuring] to 91.0 using RL instead of distillation, which even exceeds QwQ-32B-Preview [qwq-32b-preview] and o1-mini [openai2024learning]. OREAL also improves DeepSeek-R1-Distilled-Qwen-7B from 92.8 to 94.0 pass@1 accuracy, being on par with the previous best 32B models. For the 32B model, -32B outperforms all previous models (Figure fig: main_fig), both distilled and RL-based, obtaining new state-of-the-art results with 95.0 pass@1 accuracy on MATH-500.
Methods¶
To obtain a deeper understanding of the challenges when applying reinforcement learning (RL) for solving math word problems, we first analyze the formulation of RL and the intrinsic properties of underlying binary feedback environments (:preliminary), and establish a theoretical foundation for our optimization framework about how to learn from positive samples (:learn_from_positive) and failure trials (:learn_from_negative). To further conquer the learning ambiguity brought by outcome rewards to the partially correct long reasoning chains, we adopt a new strategy to estimate the importance of tokens for learning (:tokenlevel_rm).
Preliminary¶
When adopting a large language model (LLM) for mathematic reasoning, the input to the LLM policy is a textual math problem that prompts the LLM to output a multi-step reasoning trajectory consisting of multiple tokens as actions. During RL training, common practices [shao2024deepseekmath,yuan2023scaling] conduct sampling on the LLM to produce multiple reasoning trajectories, assign binary feedback (0/1 reward) based solely on the correctness of their final answer, and perform corresponding policy optimization using the sampled trajectories with reward.
Policy Optimization. Consider a Markov Decision Process (MDP) defined by the tuple (( S, A, P, r, )), where (S) is a finite state space (, contextual steps in mathematical reasoning), (A) is the action space ( the token space of LLMs), (P(s'|s, a)) specifies the state transition dynamics, (r: S A R) is the reward function, and ( [0,1)) denotes the discount factor.
In this section, we focus on KL-regularized policy optimization, which maximizes the expected cumulative returns while regularizing the policy (_(|s)) toward a reference policy (_0(|s)). The objective function is formulated as:
$$ \label{eq: klpg} J(\theta) \triangleq \mathbb{E}{s \sim \rho_0, a \sim \pi\theta(\cdot|s)}\left[ Q^{\pi_\theta}(s, a) \right] - \alpha \cdot \mathbb{E}{s \sim \rho_0}\left[ D\left(\pi_\theta(\cdot|s) | \pi_0(\cdot|s)\right)\right] $$}
with the state-action value function (Q^(s,a) = E_) under vanilla policy (). This objective admits a closed-form solution for optimal policy (^*):
$$ \label{eq: solution} \pi^*(a|s) = \frac{\pi_0(a|s) \exp\left( Q^{\pi}(s,a)/\alpha \right)}{Z(s)}, $$
where (Z(s) = E_a _0(|s)) is the partition function that ensures normalization.
Best-of-N (BoN) Sampling. As a common and efficient strategy to sample multiple reasoning trajectories from LLMs, Best-of-(N) sampling selects the trajectory with maximal reward among (n) independent rollouts from (_0) to enhance policy performance. Formally, given candidate actions (\a^(i)_i=1^n _0(|s)), the chosen action is (a^* = _a^(i) Q(s,a^(i))). This strategy effectively leverages the exploration-exploitation trade-off through parallel sampling [gao2023scaling, go2023compositional].
Binary Feedback under Outcome Supervision. Though a reasoning trajectory usually contains multiple reasoning steps with thousands of tokens, there lacks an efficient approach to automatically label the correctness of each token or reasoning step in math reasoning tasks. Thus, a practical way is to parse the final answer from the reasoning trajectory [deepseekr1, lambert2024t], evaluate its correctness based on rules or models, and then provide an outcome reward at the end of the trajectory as below:
$$ \label{eq:orm} R(s_t) = \begin{cases} 1 & \text{if (t) is the end step and the answer is correct} \ 0 & \text{otherwise}, \end{cases} $$
which intrinsically treats the correct trajectory equally for learning. Moreover, the reward signal is severely sparse when compared to the thousands of tokens and does not provide any signal of progress or correctness for intermediate steps. The resulting reward distribution of trajectories is also different from that of the dense reward function constructed through preference pairs in traditional RL for large language models [ouyang2022training], which induces a more appropriate optimization framework for mathematical reasoning tasks, discussed in the next section.
Learning from Positive Samples¶
Building upon the reward equivalence principle stated in Eq. eq:orm, we first formalize a key probabilistic characteristic of BoN sampling:
lemma Let $\pi(\theta, s)$ be a distribution over parameters $\theta$ and trajectory $s$ , where each $s$ is associated with a binary reward $R(s) \in {0, 1}$ . Define $p \triangleq \mathbb{E}{s \sim \pi(\theta,\cdot)}[R(s) = 1] > 0$ . Consider the BoN sampling: $n = n_0 \to \infty$ and sample ${s_1, s_2, \dots, s_n}$ i.i.d. from $\pi\theta$ . BoN selects $s^$ uniformly from the subset with $R(s_i) = 1$ . We have that, The probability of selecting $s^$ is converge to $\frac{\pi(\theta, s)}{p}$ , which is independent of $n$ . lemma
The proof follows directly from the union law of BoN sampling ( $\text{BoN}_{n+m} = \text{BoN}_2(\text{BoN}_m, \text{BoN}_n)$ ) and the trivial distinguishability of $0-1$ rewards. This result reveals that for problems with attainable positive responses, we are using a BoN generator with an arbitrary sampling budget to construct the positive training samples.
To quantify the distributional divergence induced by BoN sampling, prior work [hilton2022measuring, scheurer2023training, coste2023reward] has analyzed the KL divergence between the BoN distribution $\pi_{\text{BoN}}$ and the original policy $\pi$ . For continuous trajectory spaces $\mathcal{S}$ , the BoN distribution admits the explicit form:
$$ \label{eq: pibon} \pi_{\text{BoN}}(s) = n \cdot \left[P(s)\right]^{n-1} \cdot \pi(s), $$
where $P(s)$ denotes the cumulative distribution function (CDF) associated with $\pi(s)$ . The corresponding KL divergence is given by [ KL(_BoN ) = n - n-1n. ]
The KL divergence expression possesses two crucial properties, as $n \to \infty$ , the $\text{KL}(\pi_{\text{BoN}}\parallel \pi)$ is strictly increasing in $n$ for $n \geq 1$ , and converge to $\log n \to \infty$ , covering the entire positive real axis. This implies that for any prescribed KL divergence constraint $\epsilon > 0$ , there exists a BoN parameterization for approximating the optimal policy, and can be simply sampled with the minimal $n(\epsilon)$ satisfying that [ n() = _n E_s _BoN[-R(s)]. ]
BoNBoN [gui2024bonbon] empirically shows that BoN sampling achieves the optimal win rate under fixed KL constraint by exhaustive search over the positive support. Therefore, the behavior cloning on BoN-selected positive samples directly learns the analytic solution to Eq. eq: klpg. Intuitively, since every correct answer is preferred identically in the outcome-supervised sense, we only need to sample until we get a positive example, whose generating probability distribution will be the same as randomly picking from arbitrarily large numbers of samples.
Based on the theoretical understanding established, we formulate the first component of the learning objective in by incorporating KL-constrained max-likelihood-objective over positive examples obtained through sampling:
[ L_1() = E_s D^+ Positive example alignment + KL( _old)_Policy constraint, ] where $\mathcal{D}^+$ denotes the set of positive trajectories selected via BoN sampling from RL rollouts.
Learning from Negative Samples¶
As established in Section subsec:learn_from_positive, direct behavioral cloning on positive responses can effectively recover the policy distribution. BOND [sessa2024bond] proposes estimating Jeffreys divergence [jeffreys1946invariant] for the BoN strategy to train with both positive and negative samples, and demonstrates that signals from unsuccessful trajectories provide critical information about decision boundaries and failure modes.
In this section, we will discuss the relationship between the BoN (Best-of-N) distribution and the optimization objective defined in Eq. eq: klpg, then elucidate the necessity of reward reshaping when training with negative samples. Notably, while Eq. eq: pibon shares structural similarities with Eq. eq: solution, its application to mathematical reasoning tasks with binary feedback requires reformulation. Specifically, the transformed BoN distribution can be expressed as
$$ \label{eq. pbon} \pi_{\text{bon}}(s) = \pi(s) \left[ R(s) \cdot \frac{1 - \left(1 - p\right)^n}{p} + \left(1 - R(s)\right) \cdot \left(1 - p\right)^{n-1} \right], $$
which reveals fundamental differences between the BoN distribution and the original sampling distribution. Consider a scenario where two correct and two incorrect solutions are sampled, yielding an empirical accuracy of 50%. However, the probability of selecting negative samples under Best-of-4 becomes $(0.5)^4 = 6.25\%$ , significantly lower than the original distribution. This discrepancy necessitates reward shaping to maintain consistency between our optimization target and the expected return under the BoN distribution.
Building on BoN-RLB's [chow2024inference] application of the log-likelihood trick for BoN-aware policy gradients, we analyze the reward shaping technic for negative samples to maintain gradient consistency with Section subsec:learn_from_positive. With expectation return $p$ follow the definition in Lemma lemma:3.1. The policy gradient under the BoN distribution can be derived as
$$ \begin{aligned} \nabla_\theta J_{\text{bon}} &= \mathbb{E}{s \sim \pi(s)\right] \ &= \mathbb{E}}}(\cdot)}\left[R(s) \nabla_\theta \log \pi_{\text{bon}{s \sim \pi \right)\right], \end{aligned} $$}}(\cdot)}\left[R(s) \nabla_\theta\left( I_{D_+}(s)\log \pi(s)\frac{1 - (1-p)^n}{p} + I_{D_-}(s)\log \pi(s)(1-p)^{n-1
where $I_{D_+}(s)$ and $I_{D_-}(s)$ denote indicator functions for positive and negative sample sets respectively. Notably, these indicators are independent of policy parameters $\theta$ . Given $\mathbb{E}{s \sim \pi(s)] = 1-(1-p)^n$ , we derive the gradient components as}}}[I_{D_+
$$ &\mathbb{E}{s \sim \pi}}}\left[\nabla_\theta \left( I_{D_+}(s) \log \pi(s)\frac{1 - (1-p)^n}{p}\right)\right] = n(1-p)^{n-1} \mathbb{E{s \sim \pi, s\in D+}\left[\nabla_\theta \log \pi(s)\right]. $$
Similarly, we also have
$$ \mathbb{E}{s \sim \pi}}}\left[\nabla_\theta \left( I_{D_-}(s) \log \pi(s)(1-p)^{n-1}\right)\right] = n(1-p)^n \mathbb{E{s \sim \pi, s\in D-}\left[\nabla_\theta \log \pi(s)\right]. $$
This derivation reveals that when assigning unit reward ( $R(s)=1$ ) to positive samples, gradient consistency requires reshaping negative sample rewards to $R^\star(s) \triangleq (1-p)R(s)$ . Based on this reward shaping, we can construct policy optimization both on positive and negative samples for optimal policy.
To obtain the parameter $1-p$ which can be linked to the Monte-Carlo (MC) advantage estimation, we can simply estimate that probability by calculating the expected accuracy on the sample space by counting a small number of responses. In this paper we apply a similar setting to RLOO [fukunaga1989leave], namely $R_{RLOO}(s) = R(s) - \frac{1}{N-1} \sum_{s^\star \neq s} R(s^\star)$ for unbiased mean reward and train with policy gradient. The second part of our OREAL objective is then formulated as below: [ L_2() = E_s S_- + KL( _old), ] where $p = \mathbb{P}$ in the recent GRPO [shao2024deepseekmath] algorithm, where $mean({r_i...r_n}) \to p$ when $n \to \infty$ .}[R(\theta) = 1]$ , $S_-$ is the failed subset generated by policy model, and $F$ represents the preprocessing for advantage scores to serve as a generalized form, for example, $F(1-p) \triangleq \frac{r_i-mean({r_i...r_n})}{std({r_i...r_n})
Dealing with Long Reasoning Chains¶
In the previous discussion, we introduced the adaptation of binary reward training in response space. However, since the outcome supervision only provides feedback at the sequence level, this modeling essentially reduces to a contextual bandit without internal reward modeling within the MDP. A common counterexample is PPO, which utilizes a separate critic model to estimate the value function. However, such a solution appears to be expensive and complex, which has induced numerous explorations on how to stabilize the PPO training.
Things become slightly different in mathematical reasoning, where the model can spontaneously revise omissions in intermediate steps to obtain the correct final answer. Therefore, outcome supervision is preferred, and the value function is more meant to be a simple credit assignment to determine how much the process step contributes to the outcome reward. With efficiency and performance trade-off considerations, we choose to use some low-cost alternatives for sequence-level reweighting.
Taking into account the deterministic dynamics in mathematical reasoning ((s_t+1 = f(s_t, a_t))), the state-action function $Q^\pi(s_{<t}, \pi(s_t))$ simplifies to the cumulative discounted reward of policy ():
$$ \label{eq: vandq} Q^\pi(s_{<t}, \pi(s_t)) = V^\pi(s_{\leq t}) = \sum_{k=0}^{T-t} \gamma^k r(s_{t+k} | s_{<t}). $$
Since intermediate rewards are not provided in mathematical reasoning tasks, we define an advantage function based solely on outcome feedback:
$$ A(s_{\leq t}) = V^\pi(s_{\leq t+1}) - V^\pi(s_{\leq t}). $$
This formulation treats (A(s_ t)) as a token-wise credit assignment mechanism, estimating each token's contribution toward the final outcome.
For a pair of responses $y_1$ and $y_2$ to the same query, their initial values coincide $V_0^1 = V_0^2$ . The win rate between them then satisfies:
$$ \begin{aligned}\label{eq: winrate} p(y_1>y_2)&= \sigma(r(y_1) - r(y_2)) \&= \sigma\left( \left(V_0^1 + \sum_{t=0}^T \gamma^t A_{y_1}^t\right) - \left(V_0^2 + \sum_{t=0}^T \gamma^t A_{y_2}^t\right) \right) \ &= \sigma\left( \sum_{t=0}^T \gamma^t \left(A_{y_1}^t - A_{y_2}^t\right) \right). \end{aligned} $$
Equation eq: winrate indicates that for any function family $\mathcal{A} = {A(s_{\leq t})}$ , a cumulative reward function through sequence aggregation can be constructed to model rewards:
[ r^*(s) t=0^T ^t A(s t), ]
which is trainable via preference pairs ${(y_w, y_l)}$ by fitting the outcome feedback. The learned $A(s_{\leq t})$ serves as a weighting function for credit assignment, which is used to reweight the original training loss, emphasizing critical reasoning steps or errors. An analogous implementations is $ r 2 Q^* $ [rafailov2024r, xia2024inverse] by defining $A = \log \frac{\pi(y_i)}{\pi_{\text{ref}}(y_i)}$ , PRIME [cui2025process] then apply this formulation to improve performance of RLOO. In our work, following the practice from [cobbe2021training], we directly train a token-level reward function $w(s_{\leq t})$ satisfying [ 1T_t=0^T w(s_ t) = r(s), ] without constraining KL-divergence to reference model in reward model training. These sequential rewards can serve as a proxy for the contribution of thinking steps to the result accuracy. Assuming a pair of prefix-consistent correct and incorrect samples, due to the causal inference nature of the token-level reward model, the preference optimization for these samples will only function on the steps that have different contents, which induces higher credits on the core reasoning step that affects the final result. We further discuss the training details of this model and analyze the visualization of its token-wise scoring effects later in Section subsec: rl and Appendix sec: vis_token.
In practice, we decompose the output weight $w(s)$ for positive and negative samples and clip on the positive axis to prevent reversing the direction of the optimized gradient, denoted as $\omega^+$ and $\omega^-$ :
$$ \omega^+ = \max(2 \sigma(w) - 1, 0), \omega^- = \max(1 - 2\sigma(w), 0). $$
Giving input query $d$ , the overall loss is as follows:
$$ \begin{aligned}\label{eq: overall loss} \mathcal{L}{\text{total}}(d) \triangleq &\mathbb{E}(s) + \eta \ \omega_{s\leq t}^-\log \frac{\pi_\theta(s_{\leq t}|d)}{\pi_{old}(s_{\leq t}|d)}I_{D_-}(s) \right) \right] \ &+ \beta \text{KL}(\pi_\theta(\cdot|d) \parallel \pi_{\text{old}}(\cdot|d)), \end{aligned} $$} \left[ \sum_{t=0}^T \left( -\omega_{s\leq t}^+\log \pi_{\theta}(s_{\leq t}|d)I_{D_+
where $\eta$ represents the balancing weights for positive and negative losses.
Implementation¶
Policy Initialization¶
We utilize Qwen2.5-7B and Qwen2.5-32B [yang2024qwen2] as the base model. Initially, we fine-tune the base models using long chain-of-thought data obtained through rejection sampling [yuan2023scaling]. This rejection sampling fine-tuned (RFT) [yuan2023scaling] models then serve as the initialization for the policy model in our RL framework. We also explore to use of DeepSeek-R1-Distill-Qwen-7B [deepseekr1] as the initial policy model and perform on it and discuss the influence of different initial policy models in Section subsec: policy model analysis. The training data for the RFT models consists of in-house datasets supported by OpenDataLab [opendatalab] and open-source datasets including Numina [li2024numinamath] and the training set of MATH [hendrycks2021measuring].
Reinforcement Learning¶
Data Preparation.* During the on-policy RL process, we utilize questions from Numina, MATH training sets, and historical AMC/AIME (without AIME2024) competitions. For each question, we independently sample 16 trajectories from the RFT models. The correctness of each trajectory is then averaged to estimate the correctness rate of each query. To increase the difficulty of training queries, only questions with correctness rates between 0 and 0.8 are retained for further training.
Outcome Reward Signal.* We employ the Qwen2.5-72B-Instruct [yang2024qwen2] as a generative verifier, in conjunction with a rule-based verifier, to evaluate the correctness of the model's outputs and provide binary rewards. This combination enhances the robustness of correctness assessment, mitigating issues related to the false negative of the rule-based verifier.
Training Token-level Reward Model.* For the token-level reward model, we directly use the binary outcome rewards provided by the verifier and optimize using the cross-entropy loss:
$$ \mathcal{L}{\text{CE}} = - \mathbb{E} \left[ r \log p(s) + (1 - r) \log (1 - p(s)) \right], \label{eq: ce} $$}
where $s$ represents the sampled trajectory, $ r \in {0,1}$ is the binary outcome reward from the verifier, and $p(s) = \sigma(\frac{1}{T}\sum_{t}^{T} w(s_{t}))$ denotes the predicted probability of correctness by the token-level reward model $w$ .
To further analyze the behavior of the token-level reward model, we visualize its output distribution $w(s_{t})$ during the on-policy RL training process (see Appendix sec: vis_token). In this training paradigm, $w(s_{t})$ assigns token-wise importance scores across the chain-of-thought reasoning process, capturing each token's contribution to the final correctness of the generated response. Consequently, this allows us to leverage $w(s_{t})$ for importance sampling during the optimization process, enabling a more principled selection of informative tokens.
Training Algorithm.* The loss function for the policy model follows the formulation described in Section sec: methods. The complete RL training procedure is described in Algorithm alg:oreal.
algo/code
Hyperparameters.* The policy model is initialized from the RFT model. Similarly, the token-level reward model is also initialized with the same weights, but its output layer is replaced with a linear layer that produces a one-dimensional scalar. The weights of this layer are initialized to zero to ensure unbiased importance sampling weight at the start of training.
During training iterations, each batch consists of 64 questions, with 16 rollouts per question. The max length of each rollout trajectory is set to 16384 tokens. Then the correctness of each response is averaged to calculate the pass rate, and questions with an overall pass rate of 0 or 1 are discarded. For the remaining trajectories, we retain only one correct response and one incorrect response per question, ensuring a balanced distribution of positive and negative samples for token-level reward model training.
For optimization, the policy model is trained with a learning rate of $5e{-7}$ , while the token-level reward model is trained with a learning rate of $2e{-6}$ . The latter undergoes a 10-step warm-up phase before training begins. Both models employ a cosine annealing learning rate schedule, decaying to $1/5$ of the initial learning rate over time. We optimize both models using the AdamW optimizer. The total number of training steps is 80, with evaluation conducted every 10 steps. The KL coefficient $\beta$ is set to 0.01. We select the best-performing model determined by evaluation metrics.
Skill-based Enhancement¶
During the RL training procedure, we observe that the model consistently struggles with certain types of questions, particularly those involving specific knowledge and skill areas, such as trigonometric constant transformations, probability statistics, series transformations, . We believe this is caused by the insufficient learning of the base model on these concepts in the Pre-training or RFT stages.
To address this problem, we implement a skill-based enhancement approach, using the MATH dataset to reduce the high cost of skill annotation. Specifically, we annotate each question in the training set with its corresponding core skill. For questions that the model repeatedly fails to answer correctly during the RL phase, we perform data augmentation by including similar questions from the training set that share the same skill. These augmented questions are then added to the training data during the RFT stage to help the model better internalize these skills.
Experiment¶
Evaluation Setup¶
Baseline. We conduct evaluations against several baselines, including GPT-4o-0513 [gpt4o], Claude-Sonnet-3.5-1022 [claude35sonnet], OpenAI-o1-mini, OpenAI-o1-preview [openai2024learning], Qwen2.5-Instrust-7B, Qwen2.5-Math-Instrust-7B, Qwen2.5-Instrust-32B [yang2024qwen2], QwQ-32B-Preview [qwq-32b-preview], DeepSeek-R1-Distill-Qwen-7B, DeepSeek-R1-Distill-Qwen-32B [deepseekr1], SimpleRL [zeng2025simplerl], PRIME [cui2025process], rStarMath [guan2025rstar]. For part of the baseline, we directly use the results from their report, which we mark with .
Benchmark.* We use some well-established mathematical datasets for evaluation, including MATH-500 [hendrycks2021measuring], AIME2024 [AIME2024], AIME2025 (Part1) [AIME2024], LiveMathBench [liu2024your], and OlympiadBench [he2024olympiadbench].
Metrics.* We use pass@1 as the metric for evaluation under the zero-shot chain-of-thought setting and use greedy decoding for each sample to assess correctness using OpenCompass [2023opencompass].
./tables/main_result
Overall Results¶
Tabel tab: main_res shows the results of the comprehensive evaluation, highlighting the performance of our proposed models across different parameter scales. Notably, at the 7B scale, -7B achieves a remarkable pass@1 accuracy of 91.0 on the MATH-500 and 59.9 on OlympiadBench. To the best of our knowledge, this is the first time a model of this size has reached such a high level of accuracy using RL instead of distillation. This performance not only establishes a new milestone for RL-based methods but also surpasses significantly larger models, including QwQ-32B-Preview and OpenAI-o1-mini, demonstrating the effectiveness of our approach. Furthermore, after applying on the previous best 7B model, DeepSeek-R1-Distill-Qwen-7B, the resulting model, OREAL-DSR1-Distill-Qwen-7B, obtains 94.0 and 66.1 pass@1 accuracy on MATH-500 and OlympiadBench, respectively, setting new records among all 7B models. This result verifies the effectiveness of even when faced with strong initial policies.
For 32B models, -32B achieves a groundbreaking pass@1 accuracy of 95.0 on MATH-500, 46.7 on AIME2025-I, 74.8 on LiveMathBench, and 72.4 on OlympiadBench, setting a new state-of-the-art among all previously reported models. These results underscore the advantages of our methodology, including its scalability for training superior mathematical reasoning models across different model sizes.
Compared to the most competitive baseline, DeepSeek-R1-Distill-Qwen series, -32B demonstrates a clear advantage, whereas -7B lags slightly behind than the distilled 7B model, despite being trained on the same dataset as -32B. We attribute this discrepancy to the different affinities of the base models for the post-training data. Qwen-7B and Qwen-32B may exhibit varying degrees of knowledge gaps due to model sizes and pre-training settings. Our training data appears to better complement the existing knowledge of Qwen-32B, while it may be less effective in bridging gaps for Qwen-7B.
In addition, -DSR1-Distill-Qwen-7B improves the MATH-500 score from 92.8 to 94.0 and also achieves gains on LiveMathBench and OlympiadBench. However, its performance on the AIME benchmark series is comparatively weaker. We observe the same disadvantages of -32B and -7B, whose AIME2024 scores are relatively lower than the best scores. Since the overall performance verifies the effectiveness of the algorithm, we attribute the reason to the deficiency (, in response quality, query difficulty, and quantity) of RFT data and RL training queries for obtaining high performance in the domain of AIME and leave it for the future work.
Ablation Study¶
./tables/ablation
To verify the effectiveness of each component described in Section sec: methods, we progressively add the proposed component based on the 7B model and compare the evaluation results on MATH-500, starting from REINFORCE [sutton1999policy] as baseline.
As shown in Tabel tab: ablation, we add each component step by step, where Reward Shaping'' represents $L_2$ introduced in Section subsec:learn_from_negative,Behavior Cloning'' represents $L_1$ introduced in Section subsec:learn_from_positive, ``Importance Shaping'' represents $L_{total}$ introduced in Section subsec:tokenlevel_rm. The gradual addition of the modules steadily increases the Pass@1 scores of the 7B model on MATH-500, proving the effectiveness of our method. Ultimately, the policy model is raised from an initial score of 84.8 to 91.0.
We also report average pass@1 accuracy across all benchmarks during the training process with different RL settings. As shown in Figure fig: test acc, the REINFORCE training process is unstable, which can be mitigated by Reward Shaping''.Behavioral Cloning'' for positive samples can speed up convergence and show better performance early in training. Although the performance growth of ``Importance Sampling'' is relatively slow in the early stage of training, it ultimately obtains the best results.
Analysis of Initial Policy Models¶
./tables/analysis_policy
We further analyze by adopting it to several different initial policy models, as shown in Table tab: base_model. consistently improves the performance of each initial policy model, including our own trained model and the strong distilled model [deepseekr1], on MATH-500, LiveMathBench, and OlympiadBench, except slight fluctuations on AIME2024 and AIME2025 part1 when the performance of initial policy models are already high (, DeepSeek-R1-Distill-Qwen-7B), which demonstrates the generality of .
After adding skill-based enhancement data (introduced in Section subsec:skill-enhance), there is a significant rise in MATH-500 scores for the initial policy model (row 1 and row 3) and the corresponding RL-trained model (row 2 and row 4). Since our enhancement is performed primarily for the MATH-500, this verifies the effectiveness of the skill-based enhancement approach. In addition, the performance of the model after RL is strongly correlated with the capabilities of the initial policy model itself. The stronger the initial policy model, the higher the performance that RL can deliver, indicating the importance of policy initialization.
Related Work¶
Stimulate Reasoning using Chain of Thought.* In mathematical reasoning tasks, Chain of Thought (CoT) [wei2022chain] is recognized as a crucial technique to enhance the reasoning ability of large language models (LLMs), which can be implemented through few-shot examples [wei2022chain] or zero-shot prompt engineering [kojima2022large]. Self-consistency (SC) [wang2022self] is further proposed to generate and voting through multiple CoTs. In addition to simple CoTs, various search methods have been explored that simultaneously consider multiple potential CoTs, such as Tree-of-Thought (ToT) [yao2024tree] and Graph-of-Thought (GoT) [besta2024graph], which extend the idea to tree or graph structure, offering more flexibility in developing CoTs and backtracking. However, these methods mainly stimulate the reasoning capability of LLMs by prompts without parameter updates, these inference-time techniques do not fundamentally improve the underlying ability of LLMs.
Reasoning Enhancement by Supervised Fine-tuning.* To let the LLMs essentially acquire the reasoning abilities, many studies ying2024internlm, yu2023metamath, liu2024augmenting, li2023query, liu2023goat, yue2023mammoth have explored synthesizing high-quality data to conduct supervised fine-tuning (SFT) on LLMs. But this method heavily relies on high-quality training data and a existing high-performing model [lightman2023let]. As a result, many existing works [huang2024o1, Slow_Thinking_with_LLMs_2] have turned to distilling knowledge from powerful, large-scale models to synthesize data, yielding good results. However, distilled-based methods receive the limitations of the teacher model. One criticism of SFT is its limited generalization capability [chu2025sft]. Some studies argue that SFT merely transforms the model into a knowledge retriever, rather than an actual reasoner [kambhampati2024can].
Reinforcement Learning for LLM.* Compared to SFT, reinforcement learning (RL) offers better generalization and is therefore considered a more fundamental training aproach [chu2025sft]. Previous attempts applying RL for LLMs mainly target aligning the LLM to human preferences [ouyang2022training]. Later, some works [ying2024internlm, shao2024deepseekmath, luo2023wizardmath, lightman2023let] has attempted to use it to enhance the model's reasoning and obtained promising results. Recently, the advent of the o1 family of models [openai2024learning] and a series of o1-like works [deepseekr1, zeng2025simplerl, cui2025process, guan2025rstar] make the importance of large-scale RL for inference became more apparent. Currently, the mainstream approach to RL involves using outcome reward signals [deepseekr1, zeng2025simplerl, kazemnejad2024vineppo] and there are different views in the community on how to use that reward signal. ReST $^{EM}$ [singh2023beyond] and RFT [yuan2023scaling] simply select the positive samples based on the binary signal and only use them for behavior cloning. GRPO [shao2024deepseekmath], RLOO [fukunaga1989leave, ahmadian2024back], REINFORCE [sutton1999policy], use both positive and negative samples for policy updating, but facing the challenges of sparse reward in long sequence. PPO [schulman2017proximal] makes the preference modeling on sequence-level. Different from them, to explore the limit of outcome reward, presents a unified framework, grounded in the unique characteristics of mathematical reasoning tasks.
Conclusion and Future Work¶
This paper aims to explore the limit of Outcome REwArd-based reinforcement Learning for mathematical reasoning tasks, and proposes a unified policy optimization framework, termed , grounded in three key insights: 1) Behavior cloning on positive trajectories from Best-of-n (BoN) sampling is both necessary and sufficient for optimal policy learning under binary feedback; 2) Accordingly, a reward-shaping mechanism should be further introduced to transform the reward function derived from the optimal policy; 3) An efficient token-level credit assignment scheme can be achieved through trajectory advantage decomposition without relying on additional value networks. Together, these components form a theoretically grounded, general, and scalable approach for mathematical reasoning tasks. With , we are the first to improve the performance of a 7B model on the MATH-500 accuracy to 91 using the RL method instead of distillation, which even surpasses OpenAI-o1-mini and QwQ-32B-Preview. Even when taking the previous best 7B model, DeepSeek-R1-Distill-Qwen-7B, as initial policy, can improve it to 94 pass@1 accuracy on MATH-500, being even on par with the previous best 32B models. -32B also obtains new state-of-the-art results among the 32B model on MATH-500, LiveMathBench, and OlympiadBench.
Along with the experimental observations presented in this paper, we also find two factors that are crucial for the success of scalable RL for mathematical reasoning tasks, which become the primary focus of our future work. First, the initial policy model should be as free of knowledge deficiencies as possible, as this serves as the foundation for further improvement during the RL stage. A strong starting point ensures that RL can effectively and efficiently incentivize the underlying capability of LLMs obtained through pre-training or supervised fine-tuning. Towards this goal, it is a practical way to conduct distillation or data synthesis with DeepSeek-R1 or DeepSeek-V3, which is not explored in this work as it is orthogonal to our investigation. Second, the quality of the data used in the RL phase must be diverse and sufficient in terms of difficulty, quantity, and scope. A well-balanced dataset enables the model to reach its full potential by exposing it to a broad range of challenges and learning opportunities. Thus, we believe it is still valuable to make efforts in the pre-training and post-training data construction process.
table0 figure0 equation0 Atable Afigure Aequation
Token Level Reward Model Score Visualization¶
Figure fig:token_score_correct and fig:token_score_incorrect show the token-level reward model scores across responses. The values are normalized to [0, 1]. Cooler colors indicate higher reward scores, while warmer colors denote lower scores. For correct responses, the overall REWARDS are high, especially at the end, although there are a few lower sections in the middle. For incorrect responses, the distribution of rewards is reversed, and the closer to the end the lower the rewards. This indicates that not all tokens contribute to the response equally and it is important to assign token-level credits to the sequences.

Figure fig:token_score_correct: Token-level reward model score visualization for a correct response.
Prompt¶
Figure fig: verifier prompt is the system prompt of the verifier model, which is used during RL training to provide the binary outcome reward for a response. Figure fig: sys prompt is the system prompt we use for fine-tuning and RL training as well as the evaluation.
Figure fig: verifier prompt: Prompts for the model-based generative verifier.
Figure fig: sys prompt: System prompts for long CoT reasoning.