<RETURN_TO_BASE

Microsoft Unveils ARTIST: A Reinforcement Learning Framework Empowering LLMs with Dynamic Tool Use and Agentic Reasoning

'Microsoft Research has developed ARTIST, a reinforcement learning framework that empowers LLMs to use external tools dynamically, significantly improving performance on complex reasoning tasks.'

Advancing LLM Reasoning with Reinforcement Learning

Large Language Models (LLMs) have achieved significant progress in complex reasoning by leveraging innovations in architecture, scale, and training techniques such as Reinforcement Learning (RL). RL guides LLMs through reward signals, enabling the development of longer, coherent thought processes that adapt according to task complexity. However, most RL-enhanced LLMs depend heavily on static internal knowledge and text-only reasoning, limiting their effectiveness on tasks that require real-time data, domain-specific expertise, or exact computations. This shortfall becomes especially apparent in knowledge-intensive or open-ended problems, where the lack of access to external tools causes inaccuracies and hallucinations.

The Need for Agentic Reasoning and Tool Integration

To address these limitations, recent research has focused on agentic reasoning, where LLMs dynamically interact with external tools and environments during the reasoning process. These tools include web searches, APIs, and code execution platforms, while environments may range from simulated browsers to operating systems. Agentic reasoning empowers models to plan, adapt, and solve tasks interactively rather than relying on static inference alone. However, existing methods for integrating tools often require manually designed prompts or supervised fine-tuning, which hampers scalability and generalization.

Introducing ARTIST: A New Framework from Microsoft Research

Microsoft Research presents ARTIST (Agentic Reasoning and Tool Integration in Self-improving Transformers), a novel framework that combines agentic reasoning, reinforcement learning, and dynamic tool use to enhance LLM capabilities. ARTIST enables models to autonomously decide when, how, and which tools to utilize during multi-step reasoning tasks without needing step-level supervision. By integrating tool queries and outputs into the reasoning process, ARTIST improves interaction with external environments and overall problem-solving effectiveness.

Training with Group Relative Policy Optimization

ARTIST employs the Group Relative Policy Optimization (GRPO) technique for training, which avoids the use of value functions and relies on outcome-based group rewards. The framework structures rollouts into phases: reasoning, tool queries, tool outputs, and final answers. It uses a composite reward system that incentivizes correctness, adherence to proper formatting, and successful tool usage. This structure facilitates adaptive, multi-step problem-solving strategies.

Performance and Impact

ARTIST has demonstrated impressive performance gains on challenging mathematical and function-calling benchmarks including AMC, AIME, and Olympiad. It surpasses strong baselines such as GPT-4o, achieving up to 22% improvement in Pass@1 accuracy over base models and over 35% compared to other tool-augmented LLM approaches. The advantage stems from ARTIST's agentic reinforcement learning approach, which allows strategic tool use and refinement of multi-step solutions. Compared to prompt-based tool integration, ARTIST exhibits superior tool invocation, response quality, and deeper reasoning capabilities. Even on simpler datasets like MATH-500, ARTIST shows notable improvements through selective tool utilization.

Setting a New Standard for Adaptive AI

ARTIST marks a significant advancement by enabling LLMs to autonomously plan, adapt, and interact with external tools and environments. It learns effective tool-use strategies without detailed supervision, leading to improved accuracy, more interpretable reasoning paths, and robust behaviors. This work underscores the promising potential of agentic reinforcement learning in developing more adaptive and capable AI systems.

🇷🇺

Сменить язык

Читать эту статью на русском

Переключить на Русский