rStar2-Agent: Agentic Reasoning Technical Report logo

rStar2-Agent: Agentic Reasoning Technical Report

Free

14B math reasoning model with agentic RL, surpassing 671B models

FreeFree tier
Type
Open Source

About rStar2-Agent: Agentic Reasoning Technical Report

rStar2-Agent is a 14B math reasoning model trained with agentic reinforcement learning (RL) to achieve frontier-level performance. It demonstrates advanced cognitive behaviors such as thinking carefully before using Python coding tools and reflecting on code execution feedback to autonomously explore, verify, and refine intermediate steps. The model's capabilities are enabled through three key innovations: (i) an efficient RL infrastructure with a reliable Python code environment supporting high-throughput execution on limited GPU resources (64 MI300X GPUs); (ii) GRPO-RoC, an agentic RL algorithm with a Resample-on-Correct rollout strategy that mitigates environment noises from coding tools; (iii) an efficient agent training recipe starting with non-reasoning SFT and progressing through multi-RL stages. rStar2-Agent achieves average pass@1 scores of 80.6% on AIME24 and 69.8% on AIME25, surpassing DeepSeek-R1 (671B) with significantly shorter responses. The model also generalizes to alignment, scientific reasoning, and agentic tool-use tasks.

Key Features

Agentic reinforcement learning training
GRPO-RoC algorithm with Resample-on-Correct rollout strategy
Efficient RL infrastructure supporting high-throughput Python code execution on 64 MI300X GPUs
Multi-stage training recipe from SFT to RL
Autonomous exploration, verification, and refinement of reasoning steps using coding tools
State-of-the-art pass@1 scores on AIME24 (80.6%) and AIME25 (69.8%)
Shorter response length compared to larger models like DeepSeek-R1
Generalization to alignment, scientific reasoning, and agentic tool-use tasks
Open-source code and training recipes

Pros & Cons

Pros
  • Achieves frontier-level performance with only 14B parameters
  • Training completed in just 510 RL steps within one week on limited GPU resources
  • Open-source code and recipes enable reproducibility and further research
  • Demonstrates advanced cognitive behaviors like self-reflection and iterative refinement
Cons
  • Primarily evaluated on math benchmarks; generalization to other domains is demonstrated but less extensively
  • Dependence on a reliable Python execution environment may limit applicability in some settings
  • Research-stage project; not packaged as a user-facing product

Best For

Mathematical reasoning and problem-solvingAgentic tool-use tasks involving coding and execution feedbackScientific reasoningAlignment research