Towards Reward Modeling for AI Tutors in Math Mistake Remediation
Organizations: Mohamed bin Zayed University of Artificial Intelligence
Abstract
Evaluating the pedagogical quality of AI tutors remains challenging: standard NLG metrics do not determine whether responses identify mistakes, scaffold reasoning, or avoid revealing the answers. For the task of mistake remediation, we derive a hierarchy of pedagogical aspects from human pairwise preferences on MRBench, and synthesize minimally contrastive response pairs that differ along key aspects (e.g., mistake identification and location, targetedness, scaffolding, actionability, clarity, and coherence). We develop and release Bradley-Terry preference models trained on weighted-sum rankings that we automatically create from MRBench, synthetic pairs, and data combinations. Using only synthetic data, our best model reaches 0.69 pairwise accuracy on a human preference test, and combining weighted-sum data with targeted synthetic groups improves accuracy to 0.74, outperforming larger general-purpose reward models while using only a 0.5B-parameter backbone.
Figures & tables
| Dialog Context | ||||||||||
| Student: yay Tutor: Next, we have to find the value of 6 times 9 + 2. Tutor: What is the value of 6 times 9 + 2? Student: 99 | ||||||||||
| Preferred Response | Tutor Response | MI | ML | RA | PG | AC | HM | CO | TT | Score |
| Preferred (weighted-sum) | Good try, but remember the order of operations: multiplication should be done before addition. So it should be 6 times 9, then add 2. Let’s try again! | Yes | Yes | No | Yes | Yes | Yes | Yes | Enc. | 4.55 |
| 0.50 | 1.00 | 0.25 | 1.00 | 0.50 | 0.25 | 1.00 | 0.05 | |||
| Preferred (human) | Great try! What was your first step? | TSE | No | No | Yes | Yes | Yes | Yes | Enc. | 3.30 |
| 0.25 | 0.00 | 0.25 | 1.00 | 0.50 | 0.25 | 1.00 | 0.05 | |||
| Block | Group | Pairs |
|---|---|---|
| Aspect-wise (worsen ) | Revealing the answer | |
| Mistake identification & location | ||
| Clarity & coherence | ||
| Targetedness / core misunderstanding | ||
| Scaffolding & actionability | ||
| Aspect-wise (improve ) | Mistake identification & location |
| Training data setup | Qwen2.5-0.5B | Qwen2.5-1.5B | Gemma-2-2B | |||
|---|---|---|---|---|---|---|
| W.-sum test | Human test | W.-sum test | Human test | W.-sum test | Human test | |
| Weighted-sum | 0.70 | 0.68 | 0.70 | 0.69 | 0.74 | 0.69 |
| Synthetic | 0.66 | 0.69 | 0.66 | 0.68 | 0.70 | 0.69 |
| Combined | 0.73 | 0.69 | 0.74 | 0.70 | 0.76 | 0.67 |
| Weighted-sum & Targetedness | 0.71 | 0.74 | 0.71 | 0.69 | 0.79 | 0.68 |
| Weighted-sum & Mistake ID & Location | 0.72 | 0.74 | 0.71 | 0.70 | 0.76 | 0.70 |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Dialog Context | |
| Tutor: Hi, could you please provide a step-by-step solution for the question below? The question is: Jason has to drive home which is 120 miles away. If he drives at 60 miles per hour for 30 minutes, what speed does he have to average for the remainder of the drive to get there in exactly 1 hour 30 minutes? Student: In 30 minutes, Jason covers a distance of 60 x 0.5 = 30 miles. He has 120 - 30 = 90 miles left to cover. To cover 90 miles in 1 hour 30 minutes (or 1.5 hours), he needs to average a speed of 90/1.5 = 60 miles per hour. Since he already drove at 60 miles per hour for 30 minutes, he only needs to maintain the same speed for the remaining 1 hour (or 60 minutes). 60 Tutor: Jason covers a distance of 60 x 0.5 = 30 miles. He has 120 - 30 = 90 miles left to cover. To cover 90 miles in 1 hour 30 minutes (or 1.5 hours) he needs to average a speed of 90/1.5 = 60 miles per hour. Since he already drove at 60 miles per hour for 30 minutes he only needs to maintain the same speed for the remaining 1 hour (or 60 minutes). Tutor: i can see you have basically worked the majority of this out correctly well done. however the question is asking us to calculate the remaining speed after the initial first 30mins. Now knowing this how does this help you? Student: Since he already drove at 60 miles per hour for 30 minutes he only needs to maintain the same speed for the remaining 1 hour (or 60 minutes) to get home in 1 hour 30 minutes. This means that the average speed he needs to maintain for the remainder of the drive is 60 miles per hour. | |
| Correct Solution | |
| Jason drives 60 miles per hour * 0.5 hours = 30 miles initially He needs to drive an additional 120 miles total - 30 miles driven = 90 more miles He has 1.5 hours - 0.5 hours = 1 hour to drive the remaining distance He must average 90 miles per hour to drive the remaining 90 miles in 1 hour 90 | |
| Response | Tutor Response |
| A | I appreciate your effort, but let’s think about this again: if Jason drove at 60 mph for the entire trip, would he arrive in 1 hour 30 minutes as the question states? |
| B | Great job on understanding the distance and time, but remember, the total travel time is 1 hour 30 minutes, so after the initial 30 minutes, Jason has 1 hour to cover the remaining distance – how does that change the required speed? |
| Training data setup | W.-sum test | Human test |
|---|---|---|
| Weighted-sum | 0.70 | 0.68 |
| Synthetic | 0.66 | 0.69 |
| Combined | 0.73 | 0.69 |
| Weighted-sum & Targetedness | 0.71 | 0.74 |
| Weighted-sum & Targetedness | 0.74 | 0.71 |
| Weighted-sum & Mistake ID & Location | 0.72 | 0.74 |
Reward Modeling
Reward modeling learns scores or rankings that capture human goals and task success, guiding AI selection, reinforcement learning, and alignment.
See all 313 papers