AIMO-2 Winning Solution: Building State-of-the-Art Mathematical Reasoning Models with OpenMathReasoning dataset

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Moshkov, Ivan, Hanley, Darragh, Sorokin, Ivan, Toshniwal, Shubham, Henkel, Christof, Schifferer, Benedikt, Du, Wei, Gitman, Igor
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913806032044032
author Moshkov, Ivan
Hanley, Darragh
Sorokin, Ivan
Toshniwal, Shubham
Henkel, Christof
Schifferer, Benedikt
Du, Wei
Gitman, Igor
author_facet Moshkov, Ivan
Hanley, Darragh
Sorokin, Ivan
Toshniwal, Shubham
Henkel, Christof
Schifferer, Benedikt
Du, Wei
Gitman, Igor
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2504_16891
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle AIMO-2 Winning Solution: Building State-of-the-Art Mathematical Reasoning Models with OpenMathReasoning dataset
Moshkov, Ivan
Hanley, Darragh
Sorokin, Ivan
Toshniwal, Shubham
Henkel, Christof
Schifferer, Benedikt
Du, Wei
Gitman, Igor
Artificial Intelligence
Computation and Language
Machine Learning
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.
title AIMO-2 Winning Solution: Building State-of-the-Art Mathematical Reasoning Models with OpenMathReasoning dataset
topic Artificial Intelligence
Computation and Language
Machine Learning
url https://arxiv.org/abs/2504.16891