START: Self-taught Reasoner with Tools

TL;DR

START enhances reasoning by integrating tools, achieving 63.6% accuracy on GPQA.

cs.CL 🔴 Advanced 2025-03-07 37 views
Chengpeng Li Mingfeng Xue Zhenru Zhang Jiaxi Yang Beichen Zhang Xiang Wang Bowen Yu Binyuan Hui Junyang Lin Dayiheng Liu
large models reasoning tool integration code execution self-learning

Key Findings

Methodology

START employs a self-learning framework with two techniques: Hint-infer and Hint-RFT. Hint-infer involves inserting hints during inference to stimulate tool usage, while Hint-RFT combines Hint-infer with rejection sampling fine-tuning to optimize reasoning trajectories and fine-tune the model.

Key Results

  • START achieves 63.6% accuracy on GPQA, significantly outperforming the base model QwQ-32B-Preview.
  • On AMC23 and AIME24 math benchmarks, START achieves 95.0% and 66.7% accuracy, surpassing existing open-source models.
  • On LiveCodeBench, START achieves 47.3% accuracy, improving by 5.9% over QwQ-32B-Preview.

Significance

START significantly enhances reasoning performance by integrating external tools and a self-learning framework. This innovation addresses the hallucination issue in long-chain reasoning models when facing complex computations, providing a more reliable solution for academia and industry.

Technical Contribution

START introduces a tool-integrated long-chain reasoning model, offering new theoretical guarantees and engineering possibilities. Unlike existing methods, START can invoke external tools for code execution and self-debugging during reasoning.

Novelty

START is the first to integrate tools into long-chain reasoning, significantly enhancing model reasoning capabilities through Hint-infer and Hint-RFT techniques, offering unique innovations compared to existing methods.

Limitations

  • START is limited to integrating Python interpreters, not exploring other tools' potential.
  • The design of hints may disrupt the model's original reasoning flow, affecting its naturalness.

Future Work

Future research could explore integrating more types of tools, such as search engines and specialized libraries, to further enhance the model's reasoning capabilities and applicability.

AI Executive Summary

START: Self-taught Reasoner with Tools is an innovative tool-integrated long-chain reasoning model that significantly enhances reasoning capabilities through a self-learning framework. Existing large reasoning models perform well in complex tasks but often suffer from hallucinations when faced with complex computations. START addresses this issue by integrating external tools, particularly Python interpreters. Experimental results show that START performs excellently across multiple benchmarks, especially in scientific QA and mathematical reasoning tasks. Nonetheless, START has limitations, such as being restricted to Python tool integration. Future research could explore integrating more tools to further enhance performance.

Deep Analysis

Background

Large language models have excelled in natural language processing but still face challenges in complex reasoning tasks. Long-chain reasoning methods enhance model reasoning capabilities through multi-step intermediate reasoning but often hallucinate during complex computations. Tool-integrated reasoning methods effectively address this issue by invoking external tools.

Core Problem

Existing long-chain reasoning models often hallucinate during complex computation tasks, affecting reasoning accuracy. Effectively integrating external tools to enhance model reasoning capabilities is a pressing issue.

Innovation

START integrates external tools through Hint-infer and Hint-RFT techniques, significantly enhancing model reasoning capabilities. Hint-infer stimulates tool usage by inserting hints during inference, while Hint-RFT combines rejection sampling fine-tuning to optimize reasoning trajectories.

Methodology

  • �� Hint-infer: Insert hints during inference to stimulate tool usage.

  • �� Hint-RFT: Combine Hint-infer and rejection sampling fine-tuning to optimize the model.

  • �� Tool integration: Execute code and self-debug through Python interpreters.

Experiments

Experimental design includes multiple benchmarks such as GPQA, AMC23, AIME24, and LiveCodeBench. Fine-tune QwQ-32B-Preview as the base model and compare with existing open-source models.

Results

On GPQA, START achieves 63.6% accuracy, significantly outperforming QwQ-32B-Preview. On math benchmarks, START achieves 95.0% and 66.7% accuracy, surpassing existing open-source models.

Applications

START can be used for complex scientific reasoning and mathematical problem-solving, especially for tasks requiring external tool support, such as code execution and self-debugging.

Limitations & Outlook

START is limited to Python tool integration, not exploring other tools' potential. The design of hints may disrupt the model's original reasoning flow, affecting its naturalness.

Plain Language Accessible to non-experts

Imagine you're cooking in a kitchen. Existing long-chain reasoning models are like a chef with many ingredients and tools but sometimes forgets to use important tools like timers or thermometers, leading to imperfect dishes. START is like a smart assistant who reminds the chef to use these tools at critical moments, ensuring every dish is perfectly presented. In this way, START helps the chef complete complex cooking tasks more effectively.

ELI14 Explained like you're 14

Hey there! Imagine you're playing a super complex game with lots of puzzles to solve. Existing game assistants sometimes give wrong hints, making you stuck. START is like a super smart game assistant that gives you the right hints when you need them, helping you breeze through the game. This way, you can easily solve the game's challenges and become a game master!

Glossary

Long Chain-of-Thought

A reasoning method that enhances model capabilities through multi-step intermediate reasoning.

Used to decompose complex problems to improve reasoning accuracy.

Tool-integrated Reasoning

Enhances model reasoning capabilities by invoking external tools.

Addresses hallucination issues in long-chain reasoning models during complex computations.

Hint-infer

Stimulates tool usage by inserting hints during inference.

Enhances tool usage capabilities during reasoning.

Hint-RFT

Combines Hint-infer and rejection sampling fine-tuning to optimize the model.

Enhances model performance by optimizing reasoning trajectories.

Python Interpreter

A tool used for executing Python code.

Used for code execution and self-debugging to enhance reasoning capabilities.

Open Questions Unanswered questions from this research

  • 1 How to integrate more types of tools to enhance model performance?
  • 2 How does hint design affect the model's original reasoning flow?

Applications

Immediate Applications

Scientific Reasoning

Used to solve complex scientific problems and enhance reasoning accuracy.

Mathematical Problem Solving

Enhances mathematical reasoning capabilities through tool integration.

Long-term Vision

General Reasoning Model

Integrates multiple tools to create a more powerful general reasoning model.

Abstract

Large reasoning models (LRMs) like OpenAI-o1 and DeepSeek-R1 have demonstrated remarkable capabilities in complex reasoning tasks through the utilization of long Chain-of-thought (CoT). However, these models often suffer from hallucinations and inefficiencies due to their reliance solely on internal reasoning processes. In this paper, we introduce START (Self-Taught Reasoner with Tools), a novel tool-integrated long CoT reasoning LLM that significantly enhances reasoning capabilities by leveraging external tools. Through code execution, START is capable of performing complex computations, self-checking, exploring diverse methods, and self-debugging, thereby addressing the limitations of LRMs. The core innovation of START lies in its self-learning framework, which comprises two key techniques: 1) Hint-infer: We demonstrate that inserting artificially designed hints (e.g., ``Wait, maybe using Python here is a good idea.'') during the inference process of a LRM effectively stimulates its ability to utilize external tools without the need for any demonstration data. Hint-infer can also serve as a simple and effective sequential test-time scaling method; 2) Hint Rejection Sampling Fine-Tuning (Hint-RFT): Hint-RFT combines Hint-infer and RFT by scoring, filtering, and modifying the reasoning trajectories with tool invocation generated by a LRM via Hint-infer, followed by fine-tuning the LRM. Through this framework, we have fine-tuned the QwQ-32B model to achieve START. On PhD-level science QA (GPQA), competition-level math benchmarks (AMC23, AIME24, AIME25), and the competition-level code benchmark (LiveCodeBench), START achieves accuracy rates of 63.6%, 95.0%, 66.7%, 47.1%, and 47.3%, respectively. It significantly outperforms the base QwQ-32B and achieves performance comparable to the state-of-the-art open-weight model R1-Distill-Qwen-32B and the proprietary model o1-Preview.

cs.CL