Reasoning over mathematical objects: on-policy reward modeling and test time aggregation
Organizations: FAIR at Meta · CMU · UCLA
Abstract
The ability to precisely derive mathematical objects is a core requirement for downstream STEM applications, including mathematics, physics, and chemistry, where reasoning must culminate in formally structured expressions. Yet, current LM evaluations of mathematical and scientific reasoning rely heavily on simplified answer formats such as numerical values or multiple choice options due to the convenience of automated assessment. In this paper we provide three contributions for improving reasoning over mathematical objects: (i) we build and release training data and benchmarks for deriving mathematical objects, the Principia suite; (ii) we provide training recipes with strong LLM-judges and verifiers, where we show that on-policy judge training boosts performance; (iii) we show how on-policy training can also be used to scale test-time compute via aggregation. We find that strong LMs such as Qwen3-235B and o3 struggle on Principia, while our training recipes can bring significant improvements over different LLM backbones, while simultaneously improving results on existing numerical and MCQA tasks, demonstrating cross-format generalization of reasoning abilities.
Figures & tables
| Category | Content |
| Subject Entity | Statistical mechanics, structure of matter Equilibrium statistical mechanics Classical equilibrium statistical mechanics (general) |
| Strategy Description | Evaluating free-energy differences by exploiting convexity and pair-correlation functions: Applying the convexity of the Helmholtz free energy with respect to an interpolating interaction parameter, using Jensen’s inequality and the definition of the two-particle density (or radial distribution function) to derive rigorous upper and lower bounds on for systems with ordered pair potentials. |
| Problem Statement | 1. Identify a suitable statistical-mechanical inequality — monotonicity of with respect to . |
| Proposal CoT | 2. Define the Hamiltonians and show implies . |
| (Summary) | 3. Convert to free energies: . |
| 4. Require a single final inequality as the answer. |
| Model | Training Data | PrincipiaBench (Math Obj.) | MCQA | Numerical | Total Avg. | ||||||
| RealMath | Physics | ARB | SuperGPQA ∗ | Avg. | SuperGPQA ∗ | GPQA-Diamond | AIME-24 | AIME-25 | |||
| Qwen2.5-7B-Base | |||||||||||
| Qwen2.5-7B-Base | - | 11.19 | 9.32 | 16.76 | 13.75 | 12.75 | 20.41 | 27.40 | 8.02 | 3.65 | 13.81 |
| Qwen2.5-7B-Instruct | - | 12.95 | 10.45 | 19.15 | 20.05 | 15.65 | 32.61 | 35.29 | 12.08 | 7.81 | 18.80 |
| SimpleRL-7B-Zoo | SimpleZoo-Data. | 14.00 | 10.68 | 26.86 | 21.17 | 18.18 | 28.49 | 28.85 | 15.31 | 8.75 | 19.26 |
| General-Reasoner-7B | WebInstruct-Ver. | 15.96 | 12.39 | 26.86 | 23.15 | 19.59 | 32.95 | 36.55 | 10.63 | 7.71 | 20.78 |
| Verifier Type | Verifier | Active Parameters | Principia VerifyBench | |||
| Agreement with Humans (%) | Precision | Recall | F1 | |||
| Rule-based Verifier | math-verify | - | 5.95% | 5.38 | 6.67 | 5.96 |
| Fine-tuned Verifier | general-verifier | 1.5B | 82.74% | 83.13 | 93.24 | 87.90 |
| Fine-tuned Verifier | CompassVerifier | 3B | 81.55% | 90.74 | 65.33 | 75.97 |
| Fine-tuned Verifier | CompassVerifier | 7B | 88.69% | 93.75 | 80.00 | 86.33 |
| Fine-tuned Verifier | CompassVerifier | 32B | 91.66% | 94.20 | 86.67 | 90.28 |
| Data | Verifier Type | Verifier | Math Obj. (PrincipiaBench) | MCQA | Numerical | Total Avg. | ||||||
| RealMath | Physics | ARB | SuperGPQA ∗ | Avg. | SuperGPQA ∗ | GPQA-Diamond | AIME-24 | AIME-25 | ||||
| - | - | - | 9.43 | 5.80 | 17.81 | 12.18 | 11.31 | 20.46 | 18.09 | 12.5 | 10.21 | 13.31 |
| DeepScaleR | Rule | math-verify | 9.24 | 20.91 | 38.3 | 28.10 | 24.14 | 43.38 | 44.63 | 23.02 | 21.77 | 28.67 |
| DeepScaleR | Model | general-verifier | 16.20 | 18.86 | 38.03 | 30.13 | 25.81 | 42.35 | 45.74 | 25.00 | 23.13 | 29.93 |
| DeepScaleR | Model | GPT-OSS-120B | 15.55 | 19.93 | 39.16 | 31.04 | 27.42 | 45.32 | 46.14 | 24.17 | 22.29 | 30.45 |
| Principia-Collec. | Rule | math-verify | 11.53 | 10.80 | 21.54 | 15.67 | 14.89 | 29.06 | 32.29 | 5.52 | 3.96 | 16.30 |
| Training Data Answer Format | Method | Math Obj. (PrincipiaBench) | MCQA | Numerical | Total Avg. | ||||||||
| Math Obj. | Numer. | MCQA. | RealMath | Physics | ARB | SuperGPQA ∗ | Avg. | SuperGPQA ∗ | GPQA-Diamond | AIME-24 | AIME-25 | ||
| ✗ | ✗ | ✗ | - | 9.43 | 5.80 | 17.81 | 12.18 | 11.31 | 20.46 | 18.09 | 12.5 | 10.21 | 13.31 |
| ✓ | ✗ | ✗ | Single Training | 19.28 | 21.81 | 43.62 | 33.92 | 29.66 | 45.53 | 43.56 | 22.71 | 23.96 | 31.80 |
| ✗ | ✓ | ✗ | Single Training | 17.96 | 20.14 | 40.96 | 30.28 | 27.34 | 46.10 | 43.50 | 26.98 | 23.75 | 31.19 |
| ✗ | ✗ | ✓ | Single Training | 15.84 | 18.75 | 37.23 | 27.87 | 24.96 | 34.74 | 29.29 | 17.71 | 19.79 | 25.15 |
| ✓ | ✓ | ✗ | Joint Training | 17.46 | 19.89 | 40.96 | 32.18 | 27.62 | 43.75 | 42.27 | 22.50 | 20.31 | 29.92 |
| Method | RM | RM Type | RM Size | Math (Easy-to-Verify) | Hard-to-Verify | |||||
| AIME24 | AIME25 | BRUMO25 | HMMT24 | HMMT25 | Average | Physics | ||||
| Qwen3-1.7B | – | – | – | 45.22 | 36.20 | 49.78 | 23.30 | 21.44 | 35.20 | 15.57 |
| Policy Trained on hard-to-verify samples w/o Reference Answer | ||||||||||
| RLHF | Nexusflow/Athene-RM-8B | Scalar | 8B | 40.22 | 32.92 | 38.95 | 17.08 | 19.58 | 29.75 | 13.18 |
| RLHF | Skywork-Reward-V2-Llama-3.1-8B | Scalar | 8B | 48.33 | 36.26 | 47.91 | 22.92 | 24.36 | 35.96 | 16.93 |
| RLHF | nvidia/AceMath-7B-RM | Scalar | 7B | 48.76 | 38.34 | 47.69 | 23.12 | 22.71 | 36.12 | 16.24 |
| Method | Verifier | Verifier | Verifier | Math (Easy-to-Verify) | Hard-to-Verify | |||||
| Type | Size | AIME24 | AIME25 | BRUMO25 | HMMT24 | HMMT25 | Average | Physics | ||
| Qwen3-1.7B | – | – | – | 45.22 | 36.20 | 49.78 | 23.30 | 21.44 | 35.20 | 15.57 |
| Policy Trained on hard-to-verify samples w/ Reference Answer | ||||||||||
| RLVR | Math-Verify | Rule-based | – | 48.96 | 34.79 | 47.51 | 21.24 | 25.83 | 35.67 | 15.95 |
| RLVR | TIGER-Lab/general-verifier | Generative | 1.5B | 50.42 | 41.66 | 49.38 | 23.55 | 23.33 | 37.67 | 17.39 |
| RLLM | Qwen3-1.7B | Generative | 1.7B | 52.50 | 41.46 | 49.59 | 33.30 | 26.66 | 40.70 | 16.93 |
| Method | RM/Verifier | RM Type | RM Size | Math | |||||
| AIME24 | AIME25 | BRUMO25 | HMMT24 | HMMT25 | Average | ||||
| Qwen3-1.7B | – | – | – | 45.22 | 36.20 | 49.78 | 23.30 | 21.44 | 35.20 |
| Policy Trained on easy-to-verify samples w/o Reference Answer | |||||||||
| RLHF | Skywork-Reward-V2-Llama-3.1-8B | Scalar | 8B | 48.96 | 35.21 | 43.96 | 33.30 | 25.20 | 37.33 |
| RLLM | J1-Qwen-32B-RM | Generative | 32B | 53.34 | 45.43 | 52.08 | 26.70 | 27.33 | 41.18 |
| Policy Trained on easy-to-verify samples w/ Reference Answer | |||||||||
| Method | RM | RM Type | RM Size | Datasets | ||
| AlpacaEval 2.0 | ArenaHard 2.0 | ArenaHard 2.0 | ||||
| (LCWR / WR) | (Hard Prompts) | (Creative Writing) | ||||
| Policy Trained on non-verifiable WildChat samples in thinking mode | ||||||
| Qwen3-1.7B | – | – | – | 37.5 / 39.2 | 6.4 (-0.6 / +0.7) | 6.9 (-1.0 / +1.1) |
| RLHF | Nexusflow/Athene-RM-8B | Scalar | 8B | 38.0 / 42.6 | 4.3 (-0.6 / +0.7) | 15.8 (-2.0 / +1.9) |
| RLHF | Skywork-Reward-V2-Llama-3.1-8B | Scalar | 8B | 43.0 / 47.2 | 6.0 (-0.6 / +0.6) | 8.4 (-0.9 / +1.0) |
| Method | RM/Verifier | RM Type | RM Size | Math (Easy-to-Verify) | Hard-to-Verify | |||||
| AIME24 | AIME25 | BRUMO25 | HMMT24 | HMMT25 | Average | Physics | ||||
| Qwen3-1.7B | – | – | – | 45.22 | 36.20 | 49.78 | 23.30 | 21.44 | 35.20 | 15.57 |
| Policy Trained on hard-to-verify samples w/o Reference Answer | ||||||||||
| RLLM | Qwen3-1.7B | Generative | 1.7B | 51.25 | 37.71 | 51.67 | 26.70 | 26.88 | 38.84 | 17.84 |
| RLLM | J1-Qwen3-1.7B | Generative | 1.7B | 52.29 | 39.59 | 45.63 | 30.00 | 21.46 | 37.79 | 15.11 |
| Policy Trained on hard-to-verify samples w/o Reference Answer | ||||||||||
| Method | RM/Verifier | RM Training Data | Math | |||||
| AIME24 | AIME25 | BRUMO25 | HMMT24 | HMMT25 | Average | |||
| Qwen3-1.7B | – | – | 45.22 | 36.20 | 49.78 | 23.30 | 21.44 | 35.20 |
| Policy Trained on hard-to-verify samples w/o Reference Answer | ||||||||
| RLLM | Qwen3-32B (prompted) | – | 54.38 | 43.53 | 53.76 | 26.70 | 31.02 | 41.88 |
| RLLM | J1-Qwen3-32B-RM (off-policy-trained) | Llama-3.1-8B-Instruct | 57.09 | 44.37 | 48.96 | 26.87 | 28.12 | 41.08 |
| RLLM | J1-Qwen3-32B-RM (off-policy-trained) | Qwen3-8B | 54.99 | 45.82 | 50.00 | 23.55 | 27.71 | 40.41 |
| Method | RM | RM Type | #RM Judgments | Datasets | ||
| AlpacaEval 2.0 | ArenaHard 2.0 | ArenaHard 2.0 | ||||
| (LCWR / WR) | (Hard Prompts) | (Creative Writing) | ||||
| Qwen3-1.7B | – | – | – | 37.5 / 39.2 | 6.4 (-0.6 / +0.7) | 6.9 (-1.0 / +1.1) |
| Policy Trained on WildChat non-verifiable samples in thinking mode | ||||||
| RLLM | Qwen3-1.7B (prompted) | Pointwise | 40.1 / 46.2 | 5.3 (-0.7 / +0.8) | 12.4 (-1.6 / +1.4) | |
| RLLM | Qwen3-1.7B (prompted) | Pairwise with pivot | 41.0 / 42.8 | 6.3 (-0.8 / +0.8) | 13.9 (-1.3 / +1.2) | |
| Avg. of HMMT, Brumo, and AIME | |||
| Initial Temperature | Initial Pass@1 | Initial Pass@k | Aggregation Pass@1 |
| T=0.6 | 65.8 | 81.01 | 74.11 |
| T=0.8 | 65.8 | 81.3 | 74.9 |
| T=1.0 | 65.7 | 81.7 | 75.2 |
| Method | Round | HMMT | Brumo | AIME | Olympiad | Avg. |
| Qwen3-4B-Base | Initial | 3.20/5.15 | 16.67/29.04 | 8.20/21.33 | 33.72/47.41 | 15.45/25.73 |
| Agg 1 | 2.50 | 14.25 | 10.30 | 30.28 | 14.33 | |
| Agg 2 | 1.80 | 14.39 | 9.20 | 28.15 | 13.38 | |
| Agg 3 | 1.60 | 11.38 | 8.03 | 25.16 | 11.54 | |
| Dr.GRPO ( Liu et al., 2025d ) | Initial | 9.79/15.94 | 25.05/35.27 | 19.27/28.11 | 45.65 /54.92 | 24.94/33.56 |
| Agg 1 | 11.30 | 26.35 | 21.15 | 47.25 | 26.51 |
| Scientific Reasoning (PrincipiaBench) | Competition Math | ||||||||
| Model | RealMath | Physics | ARB | SuperGPQA | Avg. | AIME | Brumo | HMMT | Avg. |
| Experiments on Qwen3-4B-Base | |||||||||
| Qwen3-4B-Base | 9.11 | 5.69 | 17.44 | 11.01 | 10.81 | 8.20 | 16.67 | 3.20 | 9.36 |
| +aggregation | 8.35 | 4.79 | 18.24 | 10.68 | 10.52 | 10.30 | 14.25 | 2.50 | 9.02 |
| Dr.GRPO | 17.40 | 15.65 | 35.26 | 29.3 | 24.40 | 20.93 | 23.19 | 8.14 | 17.42 |
| +aggregation | 19.32 | 21.06 | 44.17 | 32.28 | 29.21 | 25.95 | 31.07 | 13.10 | 23.37 |
| Method | RM/Verifier | RM Type | RM Size | Math (Easy-to-Verify) | Hard-to-Verify | |||||
| AIME24 | AIME25 | BRUMO25 | HMMT24 | HMMT25 | Average | Physics | ||||
| Policy Trained on hard-to-verify samples w/o Reference Answer | ||||||||||
| Qwen3-8B | – | – | – | 76.00 | 67.30 | 68.75 | 35.41 | 41.25 | 57.74 | 51.70 |
| RLHF | Nexusflow/Athene-RM-8B | Scalar | 8B | 73.34 | 60.42 | 65.00 | 37.91 | 38.33 | 55.00 | 50.35 |
| RLVR | nvidia/AceMath-7B-RM | Scalar | 7B | 72.91 | 65.00 | 65.41 | 33.75 | 42.49 | 55.91 | 50.76 |
| RLLM | Qwen3-8B | Generative | 8B | 75.84 | 70.00 | 69.58 | 50.00 | 43.33 | 61.75 | 54.32 |
| Method | RM/Verifier | RM Type | RM Size | Math (easy-to-verify) | Hard-to-Verify | |||||
| AIME24 | AIME25 | BRUMO25 | HMMT24 | HMMT25 | Average | Physics | ||||
| Qwen3-1.7B | – | – | – | 13.33 | 10.83 | 16.04 | 6.25 | 5.21 | 10.33 | 8.75 |
| Policy Trained on hard-to-verify samples w/o Reference Answer | ||||||||||
| RLHF | Nexusflow/Athene-RM-8B | Scalar | 8B | 0 5.84 | 0 6.25 | 12.08 | 0 2.28 | 0 1.65 | 0 5.62 | 7.95 |
| RLHF | Skywork-Reward-V2-Llama-3.1-8B | Scalar | 8B | 21.87 | 20.21 | 21.87 | 12.92 | 10.00 | 17.37 | 10.00 |
| RLHF | nvidia/AceMath-7B-RM | Scalar | 7B | 20.21 | 16.87 | 18.96 | 0 7.72 | 0 5.84 | 13.92 | 10.45 |
| Method | RM/Verifier | RM Type | RM Size | AIME24 | AIME25 | HMMT24 | HMMT25 | BRUMO25 | MATH500 | Average |
| OctoThinker-8B-Hybrid-Base | – | – | – | 2.49 | 2.70 | 4.30 | 0.00 | 0.00 | 52.36 | 10.32 |
| Policy Trained on hard-to-verify samples w/o Reference Answer | ||||||||||
| RLHF | Skywork-Reward-V2-Llama-3.1-8B | Scalar | 8B | 7.30 | 3.74 | 6.88 | 3.30 | 0.00 | 60.04 | 13.54 |
| RLLM | Qwen3-1.7B (prompted) | Generative | 1.7B | 7.93 | 4.37 | 9.79 | 6.70 | 0.83 | 60.86 | 15.08 |
| w/ OctoThinker-8B-Hybrid-Base | – | – | – | +5.44 | +1.67 | +5.49 | +6.70 | +0.83 | +8.50 | +4.76 |
| Policy Trained on hard-to-verify samples w/ Reference Answer | ||||||||||
| Method | RM | RM Type | RM Size | Datasets | ||
| AlpacaEval 2.0 | ArenaHard 2.0 | ArenaHard 2.0 | ||||
| (LCWR / WR) | (Hard Prompts) | (Creative Writing) | ||||
| Policy Trained on non-verifiable WildChat samples in thinking mode | ||||||
| Qwen3-8B | – | – | – | 63.1 / 65.1 | 22.4 (-1.5 / +1.7) | 33.3 (-2.5 / +2.7) |
| RLHF | Nexusflow/Athene-RM-8B | Scalar | 8B | 70.9 / 71.2 | 25.4 (-1.6 / +1.6) | 57.9 (-2.8 / +2.0) |
| RLHF | Skywork-Reward-V2-Llama-3.1-8B | Scalar | 8B | 68.6 / 72.3 | 26.4 (-1.6 / +1.7) | 50.9 (-2.9 / +2.9) |