MathSmith: Towards Extremely Hard Mathematical Reasoning by Forging Synthetic Problems with a Reinforced Policy

Authors

  • Shaoxiong Zhan Tsinghua University
  • Yanlin Lai Tsinghua University
  • Ziyu Lu Tsinghua University
  • Dahua Lin The Chinese University of Hong Kong
  • Ziqing Yang SenseTime Research
  • Fei Tan East China Normal University

DOI:

https://doi.org/10.1609/aaai.v40i41.40760

Abstract

Large language models have achieved substantial progress in mathematical reasoning, yet their advancement is limited by the scarcity of high-quality, high-difficulty training data. Existing synthesis methods largely rely on transforming human-written templates, limiting both diversity and scalability. We propose MathSmith, a novel framework for synthesizing challenging mathematical problems to enhance LLM reasoning. Rather than modifying existing problems, MathSmith constructs new ones from scratch by randomly sampling concept–explanation pairs from PlanetMath, ensuring data independence and avoiding contamination. To increase difficulty, we design nine predefined strategies as soft constraints during rationales. We further adopts reinforcement learning to jointly optimize structural validity, reasoning complexity, and answer consistency. The length of the reasoning trace generated under autoregressive prompting is used to reflect cognitive complexity, encouraging the creation of more demanding problems aligned with long-chain-of-thought reasoning. Experiments across five benchmarks, categorized as easy & medium (GSM8K, MATH-500) and hard (AIME2024, AIME2025, OlympiadBench), show that MathSmith consistently outperforms existing baselines under both short and long CoT settings. Additionally, a weakness-focused variant generation module enables targeted improvement on specific concepts. Overall, MathSmith exhibits strong scalability, generalization, and transferability, highlighting the promise of high-difficulty synthetic data in advancing LLM reasoning capabilities.

Downloads

Published

2026-03-14

How to Cite

Zhan, S., Lai, Y., Lu, Z., Lin, D., Yang, Z., & Tan, F. (2026). MathSmith: Towards Extremely Hard Mathematical Reasoning by Forging Synthetic Problems with a Reinforced Policy. Proceedings of the AAAI Conference on Artificial Intelligence, 40(41), 34602-34610. https://doi.org/10.1609/aaai.v40i41.40760

Issue

Section

AAAI Technical Track on Natural Language Processing VI