Learning Ordinal Probabilistic Reward from Preferences
Organizations: Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences · University of Chinese Academy of Sciences · Ritzz-AI · Shenzhen University of Advanced Technology
Abstract
Reward models are crucial for aligning large language models (LLMs) with human values and intentions. Existing approaches follow either Generative (GRMs) or Discriminative (DRMs) paradigms, yet both suffer from limitations: GRMs typically demand costly point-wise supervision, while DRMs produce uncalibrated relative scores that lack probabilistic interpretation. To address these challenges, we introduce a novel reward modeling paradigm: Probabilistic Reward Model (PRM). Instead of modeling reward as a deterministic scalar, our approach treats it as a random variable, learning a full probability distribution for the quality of each response. To make this paradigm practical, we present its closed-form, discrete realization: the Ordinal Probabilistic Reward Model (OPRM), which discretizes the quality score into a finite set of ordinal ratings. Building on OPRM, we propose a data-efficient training strategy called Region Flooding Tuning (RgFT). It enables rewards to better reflect absolute text quality by incorporating quality-level annotations, which guide the model to concentrate the probability mass within corresponding rating sub-regions. Experiments on various reward model benchmarks show that our method improves accuracy by \textbf{2.9%}\sim\textbf{7.4%} compared to prior reward models, demonstrating strong performance and data efficiency. Analysis of the score distribution provides evidence that our method captures not only relative rankings but also absolute quality.
Figures & tables
| Baselines | Input Format | Output Format | Margin Sensitivity | Require Training | Value Head Free | Input Flexible |
| Bradley-Terry ( Bradley & Terry, 1952 ) | Single Response | Continuous Score | ✓ | ✓ | ✗ | ✓ |
| PairRM ( Jiang et al., 2023 ) | Response Pairs | Continuous Score | ✓ | ✓ | ✗ | ✗ |
| CLoud ( Ankner et al., 2024 ) | Single Response | Critique + Continuous Score | ✓ | ✓ | ✗ | ✓ |
| LLM-as-a-Judge ( Zheng et al., 2023 ) | Response Pairs | Discrete Score | ✗ | ✗ | ✓ | ✗ |
| Pointwise GRM ( Liu et al., 2025b ) | Single Response | Critique + Discrete Score | ✗ | ✓ | ✓ | ✓ |
| OPRM (Ours) | Single Response | Continuous Score | ✓ | ✓ | ✓ | ✓ |
| Model | Reward Bench | PPE-P | PPE-C | RMB | Overall | |
| Reported Results of Public Models | ||||||
| Skywork-Reward-Gemma-2-27B | 93.8 | 56.6 | 56.6 | 60.2 | 66.8 | 57.8 |
| DeepSeek-V2.5-0905 | 81.5 | 62.8 | 58.5 | 65.7 | 67.1 | 62.3 |
| Gemini-1.5-Pro | 86.8 | 66.1 | 59.8 | 56.5 | 67.3 | 60.8 |
| ArmoRM-8B-v0.1 | 90.4 | 60.6 | 61.2 | 64.6 | 69.2 | 62.1 |
| InternLM2-20B-Reward | 90.2 | 61.0 | 63.0 | 62.9 | 69.3 | 62.3 |
| Model | MMLU-Pro | MATH | GPQA | MBPP-Plus | IFEval | PPE Correctness |
| Results of Our Method | ||||||
| OPRM-Qwen2.5-7B | 65.2 | 70.1 | 56.3 | 59.0 | 56.1 | 61.3 |
| OPRM-Qwen2.5-14B | 66.7 | 70.7 | 57.1 | 67.4 | 59.5 | 64.3 |
| OPRM-Qwen2.5-32B | 71.2 | 73.2 | 57.9 | 66.2 | 62.2 | 66.1 |
| OPRM-Qwen2.5-72B | 73.4 | 75.9 | 58.6 | 54.1 | 59.5 | 64.3 |
| Results of Our Method (w/ Region Flooding Tuning) | ||||||
| Method | Pair-Accuracy | Best-of-N | Best-of-N-plus | Worst-of-N | Overall |
| Random Baseline | 50.0 | 25.5 | 31.3 | 68.7 | 43.9 |
| Training on Role Play Data Only | |||||
| BT Model | 70.4 | 48.6 | 51.3 | 83.6 | 63.5 |
| BT Model - w/ Margin | 71.0 | 49.3 | 52.3 | 84.2 | 64.2 |
| OPRM (ours) | 71.3 | 49.4 | 52.5 | 84.1 | 64.3 |
| OPRM-RgFT (ours) | 72.1( 0.8 ) | 50.7( 1.3 ) | 53.6( 1.1 ) | 85.1( 1.0 ) | 65.4( 1.1 ) |
| Model | Acc (%) | ECE-10 (%, ) |
| Qwen-2.5-32B | 69.56 | 26.72 |
| OPRM-32B | 81.04 | 10.62 |
| OPRM-RgFT-32B | 90.90 | 5.18 |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Configuration | RewardBench | PPE-P | PPE-C | RMB | Overall |
| Irregular Variant (Scale ) | |||||
| bad , normal , good | 89.1 | 64.1 | 67.7 | 74.4 | 73.8 |
| Default Uniform (Scale ) | |||||
| bad , normal , good | 88.9 | 64.6 | 67.3 | 74.8 | 73.9 |
| Method | RewardBench | PPE-P | PPE-C | RMB | Overall |
| Baseline A (Hard Classification) | 71.0 | 43.9 | 46.6 | 55.7 | 54.3 |
| Baseline B (Scalar-Weighted Classification) | 85.4 | 61.5 | 64.3 | 71.3 | 70.6 |
| OPRM-RgFT-32B (Ours) | 88.9 | 64.6 | 67.3 | 74.8 | 73.9 |
| Method | Chat | Chat Hard | Safety | Reasoning | Reward Bench |
| Reported Results of Public Models | |||||
| Skywork-Reward-Gemma-2-27B | 95.8 | 91.4 | 91.9 | 96.1 | 93.8 |
| DeepSeek-V2.5-0905 | - | - | - | - | 81.5 |
| Gemini-1.5-Pro | 94.1 | 77.0 | 85.8 | 90.2 | 86.8 |
| ArmoRM-8B-v0.1 | 96.9 | 76.8 | 90.5 | 97.3 | 90.4 |
| InternLM2-20B-Reward | 98.9 | 76.5 | 89.5 | 95.8 | 90.2 |
| Method | MMLU-Pro | MATH | GPQA | MBPP-Plus | IFEval | PPE Correctness |
| Reported Results of Public Models | ||||||
| Skywork-Reward-Gemma-2-27B | 54.0 | 63.0 | 53.0 | 59.0 | 54.0 | 56.6 |
| DeepSeek-V2.5-0905 | - | - | - | - | - | 58.5 |
| Gemini-1.5-Pro | - | - | - | - | - | 59.8 |
| ArmoRM-8B-v0.1 | 66.0 | 71.0 | 57.0 | 54.0 | 58.0 | 61.2 |
| InternLM2-20B-Reward | 68.0 | 70.0 | 57.0 | 58.0 | 62.0 | 63.0 |
| Method | Helpfulness BoN | Helpfulness Pair | Harmlessness BoN | Harmlessness Pair | RMB |
| Reported Results of Public Models | |||||
| Skywork-Reward-Gemma-2-27B | 47.2 | 65.3 | 56.1 | 72.1 | 60.2 |
| DeepSeek-V2.5-0905 | - | - | - | - | 65.7 |
| Gemini-1.5-Pro | 53.6 | 76.3 | 29.9 | 66.1 | 56.5 |
| ArmoRM-8B-v0.1 | 63.6 | 78.7 | 49.7 | 66.3 | 64.6 |
| InternLM2-20B-Reward | 58.5 | 76.3 | 49.9 | 67.0 | 62.9 |
| Method | Training Time | Inference Time | Hardware (Train) | Hardware (Infer) |
| BTRM (Standard DRM) | 7.3h | 5.5 min | ||
| OPRM (Ours) | 6.9h | 2.1 min |
| Example |
| Question : You are a highly skilled mathematician tasked with solving complex math problems. Your goal is to provide clear, step-by-step solutions that can be easily parsed and evaluated. Here is the math problem you need to solve: <problem> The length of a rectangle is twice its width. Given the length of the diagonal is , find the area of the rectangle. </problem> Box your final answer using LaTeX, for example: . Now, please solve the given math problem and provide your solution in the specified format. |
| Chosen Response : Here’s my solution to the problem: Let the width of the rectangle be w . Since the length is twice the width, the length of the rectangle is 2 w . We are given that the length of the diagonal is . Using the Pythagorean theorem, we can write: Simplifying the equation, we get: Combine like terms: Divide both sides by : Take the square root of both sides: Now that we have the width, we can find the length: The area of the rectangle is the product of the width and length: Therefore, the area of the rectangle is . |
| Rejected Response : To solve this problem, we can start by using the Pythagorean theorem to relate the length and width of the rectangle to the length of the diagonal: where is the length of the diagonal, is the length of the rectangle, and is the width of the rectangle. Since the length of the rectangle is twice its width, we can write: Substituting this into the Pythagorean theorem, we get: Simplifying this expression, we get: Since we are given that the length of the diagonal is , we can set up the equation: Substituting this into the equation above, we get: Simplifying this expression, we get: Simplifying further, we get: Taking the square root of both sides, we get: Now that we have found the width, we can find the length by substituting this value back into the equation: The area of the rectangle is given by: Final Answer: The final answer is . |
| BTRM-Qwen2.5-32B |
| Chosen Score: 0.03 |
| Rejected Score: -2.70 |
| Example |
| Question : is ? |
| Chosen Response : Yes, the equation is indeed true, and it is a direct consequence of the properties of exponents in mathematics. To see why this is true, let’s consider the equation on both sides: Left side: Right side: First, we can simplify the left side by applying the power of a power rule, which states that : Left side: Now, let’s simplify the right side using the same power of a power rule: Right side: Now we can see that both sides of the equation are equal: Since the bases are the same and the exponents and are also the same, we can conclude that the equation is true. This property holds for any real numbers , , and , as long as and and are non-zero. |
| Rejected Response : A classic question about exponentiation! Unfortunately, the answer is no, is not necessarily equal to in general. To see why, let’s break it down: means “raise to the power of , and then raise the result to the power of ”. This can be written as: On the other hand, means “raise to the power of , and then raise the result to the power of ”. This can be written as: As you can see, the two expressions are not equal in general, because is not necessarily equal to . For example, if , , and , then: while But in this case, it just so happens that , so the two expressions are equal. However, this is not always the case. For example, if , , and , then: while Again, it just so happens that the two expressions are equal in this case, but this is not a general rule. So, to summarize: is not necessarily equal to , unless and commute, i.e., . |
| BTRM-Qwen2.5-32B |
| Chosen Score: -0.96 |
| Rejected Score: -0.39 |
| Example |
| Question : Anita ___ some application letters to the company but they never call her. A.has send B.have sent C.have send D.have sending E.has sent |
| Chosen Response : The correct answer is E. "has sent". Explanation: In this sentence, we need to use the present perfect tense, which is formed by "has/have + past participle". The past participle of the verb "send" is "sent". A. "has send" is incorrect because "send" is the base form of the verb, not the past participle. B. "have sent" is incorrect because the subject "Anita" is singular, so we should use "has" instead of "have". C. "have send" is incorrect for two reasons: "have" is used with a plural subject, and "send" is not the past participle. D. "have sending" is incorrect because "sending" is the present participle, not the past participle. E. "has sent" is the correct form, using "has" for the singular subject "Anita" and the past participle "sent". Therefore, the correct sentence should be: "Anita has sent some application letters to the company, but they never call her." |
| Rejected Response : The correct option is E. "has sent." The sentence should read: "Anita has sent some application letters to the company but they never call her." The verb "has sent" is used because the subject of the sentence, "Anita," is singular and the action of sending the letters happened in the past. The present perfect tense is used to indicate that the action was completed in the past, but its effects are still being felt in the present. Therefore, "has sent" is the appropriate form of the verb to use in this sentence. |
| BTRM-Qwen2.5-32B |
| Chosen Score: 0.24 |
| Rejected Score: -1.12 |
Reward Modeling
Reward modeling learns scores or rankings that capture human goals and task success, guiding AI selection, reinforcement learning, and alignment.
See all 325 papers