AIMO-2 Winning Solution: Building State-of-the-Art Mathematical Reasoning Models with OpenMathReasoning dataset
On the map: LLM reasoning → Mathematical reasoning
Abstract (arXiv)
This paper presents our winning submission to the AI Mathematical Olympiad - Progress Prize 2 (AIMO-2) competition. Our recipe for building state-of-the-art mathematical reasoning models relies on three key pillars. First, we create a large-scale dataset comprising 540K unique high-quality math problems, including olympiad-level problems, and their 3.2M long-reasoning solutions. Second, we develop a novel method to integrate code execution with long reasoning models through iterative training, generation, and quality filtering, resulting in 1.7M high-quality Tool-Integrated Reasoning solutions. Third, we create a pipeline to train models to select the most promising solution from many candidates. We show that such generative solution selection (GenSelect) can significantly improve upon majority voting baseline. Combining these ideas, we train a series of models that achieve state-of-the-art results on mathematical reasoning benchmarks. To facilitate further research, we release our code, models, and the complete OpenMathReasoning dataset under a commercially permissive license.
Key facts. Cited by 174 papers in the map (data through 14 Oct 2026). Placed in Mathematical reasoning, one of 17 directions in LLM reasoning. Builds on 26 of its 39 references that are in the map. First posted on arXiv 23 Apr 2025.
Builds on
26 of 39 references.
Closest ancestors · 7
Same direction, newest first
- 2025LIMO: Less is More for Reasoning5 Feb
- 2025Leveraging Online Olympiad-Level Math Problems for LLMs Training and Contamination-Resistant Evaluation24 Jan
- 2024OpenMathInstruct-2: Accelerating AI for Math with Massive Open-Source Instruction Data2 Oct
- 2024Building Math Agents with Multi-Turn Iterative Preference Learning4 Sep
- 2024InfinityMATH: A Scalable Instruction Tuning Dataset in Programmatic Mathematical Reasoning9 Aug
- 2024OpenMathInstruct-1: A 1.8 Million Math Instruction Tuning Dataset15 Feb
- 2024MARIO: MAth Reasoning with code Interpreter Output -- A Reproducible Pipeline16 Jan
Other references · 19
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2025s1: Simple test-time scalingcited by 1,341
- 2022Program of Thoughts Prompting: Disentangling Computation from Reasoning for Numerical Reasoning Taskscited by 1,076
- 2024Generative Verifiers: Reward Modeling as Next-Token Predictioncited by 480
- 2024Meta-Rewarding Language Models: Self-Improving Alignment with LLM-as-a-Meta-Judgecited by 204
- 2023Rethinking Benchmark and Contamination for Language Models with Rephrased Samplescited by 195
- 2025ToRL: Scaling Tool-Integrated RLcited by 169
7 more
- 2024Generative Reward Modelscited by 148
- 2024Critique-out-Loud Reward Modelscited by 106
- 2025Agentic Reasoning: A Streamlined Framework for Enhancing LLM Reasoning with Agentic Toolscited by 106
- 20251.4 Million Open-Source Distilled Reasoning Dataset to Empower Large Language Model Trainingcited by 49
- 2024NeMo-Aligner: Scalable Toolkit for Efficient Model Alignmentcited by 41
- 2024Efficient Tool Use with Chain-of-Abstraction Reasoningcited by 33
- 2024Direct Judgement Preference Optimizationcited by 33
Built on by
Citing papers per year: 2025: 72; 2026: 102.
174 papers.
Sort:
By area:
- 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 587
- 2025Kimi K2: Open Agentic Intelligencecited by 567
- 2025OpenThoughts: Data Recipes for Reasoning Modelscited by 304
- 2025A Survey of Reinforcement Learning for Large Reasoning Modelscited by 180
- 2025Llama-Nemotron: Efficient Reasoning Modelscited by 158
- 2025Goedel-Prover-V2: Scaling Formal Theorem Proving with Scaffolded Data Synthesis and Self-Correctioncited by 154
- 2025REASONING GYM: Reasoning Environments for Reinforcement Learning with Verifiable Rewardscited by 130
- 2025NVIDIA Nemotron Nano 2: An Accurate and Efficient Hybrid Mamba-Transformer Reasoning Modelcited by 108
- 2025Nemotron 3 Nano: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoningcited by 104
- 2025VerlTool: Towards Holistic Agentic Reinforcement Learning with Tool Usecited by 96
- 2025CWM: An Open-Weights LLM for Research on Code Generation with World Modelscited by 93
- 2025AceReason-Nemotron: Advancing Math and Code Reasoning through Reinforcement Learningcited by 87
38 more
- 2025AceReason-Nemotron 1.1: Advancing Math and Code Reasoning through SFT and RL Synergycited by 77
- 2025Expected Attention: KV Cache Compression by Estimating Attention from Future Queries Distributioncited by 74
- 2025Think Only When You Need with Large Hybrid-Reasoning Modelscited by 66
- 2025e3: Learning to Explore Enables Extrapolation of Test-Time Compute for LLMscited by 59
- 2025rStar2-Agent: Agentic Reasoning Technical Reportcited by 54
- 2025Advancing Multimodal Reasoning: From Optimized Cold Start to Staged Reinforcement Learningcited by 53
- 2025Nemotron-Cascade: Scaling Cascaded Reinforcement Learning for General-Purpose Reasoning Modelscited by 48
- 2025QuestA: Expanding Reasoning Capacity in LLMs via Question Augmentationcited by 44
- 2026On Data Engineering for Scaling LLM Terminal Capabilitiescited by 43
- 2026OpenResearcher: A Fully Open Pipeline for Long-Horizon Deep Research Trajectory Synthesiscited by 37
- 2025CoT-Self-Instruct: Building high-quality synthetic prompts for reasoning and non-reasoning taskscited by 32
- 2025InfLLM-V2: Dense-Sparse Switchable Attention for Seamless Short-to-Long Adaptationcited by 32
- 2025The Open Proof Corpus: A Large-Scale Study of LLM-Generated Mathematical Proofscited by 30
- 2025Front-Loading Reasoning: The Synergy between Pretraining and Post-Training Datacited by 27
- 2025Nemotron-Math: Efficient Long-Context Distillation of Mathematical Reasoning from Multi-Mode Supervisioncited by 26
- 2025OpenCodeReasoning-II: A Simple Test Time Scaling Approach via Self-Critiquecited by 25
- 2025Set Block Decoding is a Language Model Inference Acceleratorcited by 23
- 2025Not All Tokens Are What You Need In Thinkingcited by 20
- 2025Hybrid Reinforcement: When Reward Is Sparse, It's Better to Be Densecited by 20
- 2025Towards Concise and Adaptive Thinking in Large Reasoning Models: A Surveycited by 18
- 2025ARES: Multimodal Adaptive Reasoning via Difficulty-Aware Token-Level Entropy Shapingcited by 17
- 2025AgentMath: Empowering Mathematical Reasoning for Large Language Models via Tool-Augmented Agentcited by 17
- 2026Distribution-Aligned Sequence Distillation for Superior Long-CoT Reasoningcited by 17
- 2025QFFT, Question-Free Fine-Tuning for Adaptive Reasoningcited by 15
- 2025Don't Pass@k: A Bayesian Framework for Large Language Model Evaluationcited by 15
- 2025From Code Foundation Models to Agents and Applications: A Comprehensive Survey and Practical Guide to Code Intelligencecited by 15
- 2026On-Policy Delta Distillationcited by 15
- 2025Xolver: Multi-Agent Reasoning with Holistic Experience Learning Just Like an Olympiad Teamcited by 13
- 2025The Markovian Thinker: Architecture-Agnostic Linear Scaling of Reasoningcited by 13
- 2025EntroPIC: Towards Stable Long-Term Training of LLMs via Entropy Stabilization with Proportional-Integral Controlcited by 13
- 2025Beyond Exponential Decay: Rethinking Error Accumulation in Large Language Modelscited by 12
- 2025The Synergy Dilemma of Long-CoT SFT and RL: Investigating Post-Training Techniques for Reasoning VLMscited by 12
- 2025Learning to Refine: Self-Refinement of Parallel Reasoning in LLMscited by 12
- 2025Evaluating Parameter Efficient Methods for RLVRcited by 12
- 2026AfriqueLLM: How Data Mixing and Model Architecture Impact Continued Pre-training for African Languagescited by 12
- 2026Depth-Recurrent Attention Mixtures: Giving Latent Reasoning the Attention it Deservescited by 12
- 2026Agentic Proposing: Enhancing Large Language Model Reasoning via Compositional Skill Synthesiscited by 12
- 2026The Last AI Built by Humans: Toward Genuine Recursive Self-Improvementcited by 11
50 most cited of 174 All 174 (JSON)
- 2026Beyond Distribution Matching: Selective Margin Preservation for Supervised Fine-Tuning14 Oct 2026
- 2026BoundaryEvo: Data-Model Co-Evolution at the Capability Boundary for Long-Chain Reasoning Distillation9 Oct 2026
- 2026On-Policy Distillation with Negative-Policy Rollouts6 Oct 2026
- 2026MESH-Harness: Self-Improving Agent Harnesses via Bandit-Guided Compositional Evolution4 Oct 2026
- 2026ResOPD: Tail Residualization for Sparse On-Policy Distillation4 Oct 2026
- 2026Understanding Enrichment in Reinforcement Learning2 Oct 2026
- 2026Text-Centric Post-Training for Omni-Modal Reasoning2 Oct 2026
- 2026Does Scaling Reinforcement Learning Really Require More Training?1 Oct 2026
- 2026Diagnosing On-Policy Self-Distillation for Reasoning Language Models30 Sep 2026
- 2026CORE: COverage CAlibration and Evicted-Mass REdistribution for KV Cache28 Sep 2026
- 2026Learning to Steer, Steering to See: Unveiling the Geometry of RLVR in Large Language Models via Trainable Vectors28 Sep 2026
- 2026Next Thoughts Are Distributions: Generative Autoregressive Reasoning in the Latent Space27 Sep 2026
38 more
- 20261% of Tokens Can Be Enough: On Gradient Estimation in On-Policy Distillation21 Sep 2026
- 2026The Last AI Built by Humans: Toward Genuine Recursive Self-Improvement10 Sep 2026
- 2026CataOPD: Catalytic On-Policy Distillation for Large Language Model Reasoning25 Aug 2026
- 2026Parason: Revealing Subtask and Trial Parallelism in LLM Reasoning25 Aug 2026
- 2026SANE: State Anomaly Neutralization for Stable Extreme-Context Delta-Rule Models23 Aug 2026
- 2026Every Coin Has Two Sides: On the Dual Nature of Generalization in On-Policy Distillation of Large Language Models17 Aug 2026
- 2026QUASAR: Lowering the Loss Floor of Quantization-Aware Training with Loss-Aware Reconstruction14 Aug 2026
- 2026Capacity-Dependent Effects of Data Selection for Reasoning13 Aug 2026
- 2026Fusion Training for Mathematical Generalization in Large Language Models10 Aug 2026
- 2026Test-Time Scaling in Reasoning LLMs: Inference Regimes, Evaluation, and Reproducibility4 Aug 2026
- 2026Learning Latent Reasoning Traces for Scalar Reward Models End-to-End31 Jul 2026
- 2026Reinforcement Learning for Code Optimization28 Jul 2026
- 2026T^2MLR: Transformer with Temporal Middle-Layer Recurrence16 Jul 2026
- 2026On-Policy Delta Distillation16 Jul 2026
- 2026AdaStop: Cost-Aware Early Stopping for DNN Test Selection6 Jul 2026
- 2026Don't Let Gains FADE: Breaking Down Policy Gradient Weights in RL1 Jul 2026
- 2026Building Multi-Task Agentic LLMs via Two-Phase Distillation29 Jun 2026
- 2026EntroRouter: Learning Efficient Model Routing via Entropy Regulation28 Jun 2026
- 2026Prefix-Guided On-Policy Distillation: Mining Golden Trajectories from Rollouts20 Jun 2026
- 2026Fine-Tuning Large Language Models for Quantum Reasoning20 Jun 2026
- 2026STARE: Surprisal-Guided Token-Level Advantage Reweighting for Policy Entropy Stability17 Jun 2026
- 2026SuperThoughts: Reasoning Tokens in Superposition11 Jun 2026
- 2026Ouroboros-Spatial: Closing the Data-Model Loop for Spatial Reasoning10 Jun 2026
- 2026The Role of Feedback Alignment in Self-Distillation9 Jun 2026
- 2026Expert-validated STEM QA6 Jun 2026
- 2026Value-and-Structure Alignment for Routing-Consistent Quantization of Mixture-of-Experts Models4 Jun 2026
- 2026A Primer in Post-Training Reasoning Data: What We Know About How It Works1 Jun 2026
- 2026Knowing What to Solve Before How: Preplan Empowered LLM Mathematical Reasoning28 May 2026
- 2026Extrapolative Weight Averaging Reveals Correctness-Efficiency Frontiers in Code RL27 May 2026
- 2026ResearchMath-14K: Scaling Research-Level Mathematics via Agents27 May 2026
- 2026ReverseMath: Answer Inversion for Scalable and Verifiable Mathematical Problem Generation26 May 2026
- 2026Not only where, But when: Temporal Scheduling for RLVR25 May 2026
- 2026Unified Data Selection for LLM Reasoning21 May 2026
- 2026Language Generation as Optimal Control: Closed-Loop Diffusion in Latent Control Space14 May 2026
- 2026Learning to Build the Environment: Self-Evolving Reasoning RL via Verifiable Environment Synthesis14 May 2026
- 2026Data Difficulty and the Generalization--Extrapolation Tradeoff in LLM Fine-Tuning13 May 2026
- 2026Evolutionary Task Discovery: Advancing Reasoning Frontiers via Skill Composition and Complexity Scaling12 May 2026
- 2026GRC: Unifying Reasoning-Driven Generation, Retrieval and Compression9 May 2026
50 newest of 174 All 174 (JSON)
- 2025OpenThoughts: Data Recipes for Reasoning Modelscited by 304
- 2025A Survey of Reinforcement Learning for Large Reasoning Modelscited by 180
- 2025Llama-Nemotron: Efficient Reasoning Modelscited by 158
- 2025NVIDIA Nemotron Nano 2: An Accurate and Efficient Hybrid Mamba-Transformer Reasoning Modelcited by 108
- 2025Nemotron 3 Nano: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoningcited by 104
- 2025VerlTool: Towards Holistic Agentic Reinforcement Learning with Tool Usecited by 96
- 2025AceReason-Nemotron: Advancing Math and Code Reasoning through Reinforcement Learningcited by 87
- 2025AceReason-Nemotron 1.1: Advancing Math and Code Reasoning through SFT and RL Synergycited by 77
- 2025Think Only When You Need with Large Hybrid-Reasoning Modelscited by 66
- 2025e3: Learning to Explore Enables Extrapolation of Test-Time Compute for LLMscited by 59
- 2025rStar2-Agent: Agentic Reasoning Technical Reportcited by 54
- 2025Advancing Multimodal Reasoning: From Optimized Cold Start to Staged Reinforcement Learningcited by 53
12 most cited of 111 All 111 (JSON)
- 2026BoundaryEvo: Data-Model Co-Evolution at the Capability Boundary for Long-Chain Reasoning Distillation9 Oct 2026
- 2026ResOPD: Tail Residualization for Sparse On-Policy Distillation4 Oct 2026
- 2026Text-Centric Post-Training for Omni-Modal Reasoning2 Oct 2026
- 2026Diagnosing On-Policy Self-Distillation for Reasoning Language Models30 Sep 2026
- 2026Learning to Steer, Steering to See: Unveiling the Geometry of RLVR in Large Language Models via Trainable Vectors28 Sep 2026
- 2026CataOPD: Catalytic On-Policy Distillation for Large Language Model Reasoning25 Aug 2026
- 2026Parason: Revealing Subtask and Trial Parallelism in LLM Reasoning25 Aug 2026
- 2026Fusion Training for Mathematical Generalization in Large Language Models10 Aug 2026
- 2026Test-Time Scaling in Reasoning LLMs: Inference Regimes, Evaluation, and Reproducibility4 Aug 2026
- 2026Learning Latent Reasoning Traces for Scalar Reward Models End-to-End31 Jul 2026
- 2026Don't Let Gains FADE: Breaking Down Policy Gradient Weights in RL1 Jul 2026
- 2026Prefix-Guided On-Policy Distillation: Mining Golden Trajectories from Rollouts20 Jun 2026
12 newest of 111 All 111 (JSON)
- 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 587
- 2025Kimi K2: Open Agentic Intelligencecited by 567
- 2026OpenResearcher: A Fully Open Pipeline for Long-Horizon Deep Research Trajectory Synthesiscited by 37
- 2025From Code Foundation Models to Agents and Applications: A Comprehensive Survey and Practical Guide to Code Intelligencecited by 15
- 2026ResearchMath-14K: Scaling Research-Level Mathematics via Agentscited by 2
- 2026Building Multi-Task Agentic LLMs via Two-Phase Distillationcited by 1
- 2026Patch the Distribution Mismatch: RL Rewriting Agent for Stable Off-Policy SFTcited by 0
- 2026Awakening the Sleeping Agent: Lean-Specific Agentic Data Reactivates General Tool Use in Goedel Provercited by 0
- 2026MESH-Harness: Self-Improving Agent Harnesses via Bandit-Guided Compositional Evolutioncited by 0
- 2026MESH-Harness: Self-Improving Agent Harnesses via Bandit-Guided Compositional Evolution4 Oct 2026
- 2026Building Multi-Task Agentic LLMs via Two-Phase Distillation29 Jun 2026
- 2026ResearchMath-14K: Scaling Research-Level Mathematics via Agents27 May 2026
- 2026Awakening the Sleeping Agent: Lean-Specific Agentic Data Reactivates General Tool Use in Goedel Prover9 Apr 2026
- 2026OpenResearcher: A Fully Open Pipeline for Long-Horizon Deep Research Trajectory Synthesis17 Mar 2026
- 2026GLM-5: from Vibe Coding to Agentic Engineering17 Feb 2026
- 2026Patch the Distribution Mismatch: RL Rewriting Agent for Stable Off-Policy SFT11 Feb 2026
- 2025From Code Foundation Models to Agents and Applications: A Comprehensive Survey and Practical Guide to Code Intelligence23 Nov 2025
- 2025Kimi K2: Open Agentic Intelligence28 Jul 2025
- 2026The Role of Feedback Alignment in Self-Distillationcited by 4
- 2026Every Coin Has Two Sides: On the Dual Nature of Generalization in On-Policy Distillation of Large Language Modelscited by 4
- 20261% of Tokens Can Be Enough: On Gradient Estimation in On-Policy Distillationcited by 2
- 2025Delta Knowledge Distillation for Large Language Modelscited by 0
- 2026On-Policy Distillation with Negative-Policy Rolloutscited by 0
- 2026On-Policy Distillation with Negative-Policy Rollouts6 Oct 2026
- 20261% of Tokens Can Be Enough: On Gradient Estimation in On-Policy Distillation21 Sep 2026
- 2026Every Coin Has Two Sides: On the Dual Nature of Generalization in On-Policy Distillation of Large Language Models17 Aug 2026
- 2026The Role of Feedback Alignment in Self-Distillation9 Jun 2026
- 2025Delta Knowledge Distillation for Large Language Models18 Sep 2025
- 2026Reinforcement Learning for Code Optimizationcited by 0
- 2026Does Scaling Reinforcement Learning Really Require More Training?cited by 0
- 2026Understanding Enrichment in Reinforcement Learningcited by 0
- 2026Understanding Enrichment in Reinforcement Learning2 Oct 2026
- 2026Does Scaling Reinforcement Learning Really Require More Training?1 Oct 2026
- 2026Reinforcement Learning for Code Optimization28 Jul 2026
- 2026On-Policy Delta Distillationcited by 15
- 2025DuPO: Enabling Reliable LLM Self-Verification via Dual Preference Optimizationcited by 7
- 2026On-Policy Delta Distillation16 Jul 2026
- 2025DuPO: Enabling Reliable LLM Self-Verification via Dual Preference Optimization20 Aug 2025
- 2025REASONING GYM: Reasoning Environments for Reinforcement Learning with Verifiable Rewardscited by 130
- 2026On Data Engineering for Scaling LLM Terminal Capabilitiescited by 43
- 2025InfLLM-V2: Dense-Sparse Switchable Attention for Seamless Short-to-Long Adaptationcited by 32
- 2025The Open Proof Corpus: A Large-Scale Study of LLM-Generated Mathematical Proofscited by 30
- 2025Set Block Decoding is a Language Model Inference Acceleratorcited by 23
- 2025EntroPIC: Towards Stable Long-Term Training of LLMs via Entropy Stabilization with Proportional-Integral Controlcited by 13
- 2025Evaluating Parameter Efficient Methods for RLVRcited by 12
- 2026AfriqueLLM: How Data Mixing and Model Architecture Impact Continued Pre-training for African Languagescited by 12
- 2026Depth-Recurrent Attention Mixtures: Giving Latent Reasoning the Attention it Deservescited by 12
- 2026T^2MLR: Transformer with Temporal Middle-Layer Recurrencecited by 7
- 2025Revisiting the Scaling Properties of Downstream Metrics in Large Language Model Trainingcited by 6
- 2026Composition-RL: Compose Your Verifiable Prompts for Reinforcement Learning of Large Language Modelscited by 6
12 most cited of 33 All 33 (JSON)
- 2026Beyond Distribution Matching: Selective Margin Preservation for Supervised Fine-Tuning14 Oct 2026
- 2026Next Thoughts Are Distributions: Generative Autoregressive Reasoning in the Latent Space27 Sep 2026
- 2026SANE: State Anomaly Neutralization for Stable Extreme-Context Delta-Rule Models23 Aug 2026
- 2026QUASAR: Lowering the Loss Floor of Quantization-Aware Training with Loss-Aware Reconstruction14 Aug 2026
- 2026Capacity-Dependent Effects of Data Selection for Reasoning13 Aug 2026
- 2026T^2MLR: Transformer with Temporal Middle-Layer Recurrence16 Jul 2026
- 2026AdaStop: Cost-Aware Early Stopping for DNN Test Selection6 Jul 2026
- 2026EntroRouter: Learning Efficient Model Routing via Entropy Regulation28 Jun 2026
- 2026Expert-validated STEM QA6 Jun 2026
- 2026Extrapolative Weight Averaging Reveals Correctness-Efficiency Frontiers in Code RL27 May 2026
- 2026Not only where, But when: Temporal Scheduling for RLVR25 May 2026
- 2026Data Difficulty and the Generalization--Extrapolation Tradeoff in LLM Fine-Tuning13 May 2026
12 newest of 33 All 33 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2025AMO-Bench: Large Language Models Still Struggle in High School Math Competitions
- 2025RIMO: An Easy-to-Evaluate, Hard-to-Solve Olympiad Benchmark for Advanced Mathematical Reasoning
- 2024Omni-MATH: A Universal Olympiad Level Mathematic Benchmark For Large Language Models
- 2025PromptCoT: Synthesizing Olympiad-level Problems for Mathematical Reasoning in Large Language Models
- 2025Challenging the Boundaries of Reasoning: An Olympiad-Level Math Benchmark for Large Language Models
- 2026AIMO Interpretability Challenge
- 2026Escaping the Cognitive Well: Efficient Competition Math with Off-the-Shelf Models
- 2025UGMathBench: A Diverse and Dynamic Benchmark for Undergraduate-Level Mathematical Reasoning with Large Language Models