HiDiffTIR: Hierarchical Difficulty-Aware Policy Optimization for Multi-Turn Tool-Integrated Reasoning
On the map: LLM reasoning → Reinforcement learning for reasoning
also in 1 more
Abstract (arXiv)
Tool-Integrated Reasoning (TIR) is a fundamental capability for LLM agents to solve complex tasks by interacting with external tools iteratively. Reinforcement Learning (RL) has become the dominant paradigm for enabling this capability. However, existing approaches typically assign uniform trajectory-level advantages and treat all correct tool calls equally, ignoring the varying difficulty and learning value across trajectories and reasoning steps. This can lead to imprecise learning signals that do not adequately distinguish between trivial and challenging tool-use patterns. To address this limitation, we propose HiDiffTIR, a Hierarchical Difficulty-aware policy optimization framework for multi-turn TIR. HiDiffTIR performs difficulty-aware credit assignment at both trajectory and turn levels, enabling the policy to focus on more informative trajectories and harder reasoning steps. Notably, this fine-grained optimization is achieved without additional supervision, relying solely on group-level statistics derived from standard RL rollouts. Extensive experiments on three tool-using benchmarks demonstrate that HiDiffTIR consistently improves multi-turn TIR performance and tool invocation accuracy over strong RL baselines, highlighting the necessity of difficulty-aware credit assignment for effective policy optimization in tool-integrated LLM agents.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 29 of its 32 references that are in the map. First posted on arXiv 22 Aug 2026.
Builds on
29 of 32 references.
Closest ancestors · 7
Same direction, newest first
- 2026DIVA-GRPO: Enhancing Multimodal Reasoning through Difficulty-Adaptive Variant Advantage1 Mar
- 2026Asymmetric Prompt Weighting for Reinforcement Learning with Verifiable Rewards11 Feb
- 2026Learning to Explore with Parameter-Space Noise: A Deep Dive into Parameter-Space Noise for Reinforcement Learning with Verifiable Rewards30 Jan
- 2025SimpleTIR: End-to-End Reinforcement Learning for Multi-Turn Tool-Integrated Reasoning2 Sep
- 2025The Lessons of Developing Process Reward Models in Mathematical Reasoning13 Jan
- 2023Let's Verify Step by Step31 May
- 2022Solving math word problems with process- and outcome-based feedback25 Nov
Other references · 22
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
- 2017Deep reinforcement learning from human preferencescited by 3,846
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,277
- 2023A Survey on Large Language Model based Autonomous Agentscited by 2,203
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2023ToolLLM: Facilitating Large Language Models to Master 16000+ Real-world APIscited by 1,880
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,371
- 2023Gorilla: Large Language Model Connected with Massive APIscited by 1,302
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 641
10 more
- 2024Understanding the planning of LLM agents: A surveycited by 414
- 2023ToolAlpaca: Generalized Tool Learning for Language Models with 3000 Simulated Casescited by 383
- 2025ToolRL: Reward is All Tool Learning Needscited by 342
- 2025Rubrics as Rewards: Reinforcement Learning Beyond Verifiable Domainscited by 328
- 2024The Landscape of Emerging AI Agent Architectures for Reasoning, Planning, and Tool Calling: A Surveycited by 164
- 2025VisualPRM: An Effective Process Reward Model for Multimodal Reasoningcited by 134
- 2025DeepAgent: A General Reasoning Agent with Scalable Toolsetscited by 69
- 2025ToolHop: A Query-Driven Benchmark for Evaluating Large Language Models in Multi-Hop Tool Usecited by 48
- 2025Feedback-Driven Tool-Use Improvements in Large Language Models via Automated Build Environmentscited by 19
- 2026MatchTIR: Fine-Grained Supervision for Tool-Integrated Reasoning via Bipartite Matchingcited by 9
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Learning from the Irrecoverable: Error-Localized Policy Optimization for Tool-Integrated LLM Reasoning
- 2025Empowering Multi-Turn Tool-Integrated Agentic Reasoning with Group Turn Policy Optimization
- 2026AdaTIR: Adaptive Tool-Integrated Reasoning via Difficulty-Aware Policy Optimization
- 2025Multi-Agent Tool-Integrated Policy Optimization
- 2025Agentic Reinforced Policy Optimization
- 2025Tool-Augmented Policy Optimization: Synergizing Reasoning and Adaptive Tool Use with Reinforcement Learning
- 2026Tool-Aware Optimization with Entropy Guidance for Efficient Agentic Reinforcement Learning
- 2025VerlTool: Towards Holistic Agentic Reinforcement Learning with Tool Use