Saved in:
Bibliographic Details
Main Authors: Wu, Feijie, Zhu, Weiwu, Zhang, Yuxiang, Chatterjee, Soumya, Zhu, Jiarong, Mo, Fan, Luo, Rong, Gao, Jing
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2510.26020
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917451745198080
author Wu, Feijie
Zhu, Weiwu
Zhang, Yuxiang
Chatterjee, Soumya
Zhu, Jiarong
Mo, Fan
Luo, Rong
Gao, Jing
author_facet Wu, Feijie
Zhu, Weiwu
Zhang, Yuxiang
Chatterjee, Soumya
Zhu, Jiarong
Mo, Fan
Luo, Rong
Gao, Jing
contents Multi-tool-integrated reasoning enables LLM-empowered tool-use agents to solve complex tasks by interleaving natural-language reasoning with calls to external tools. However, training such agents from outcome-only rewards suffers from credit-assignment ambiguity, obscuring which intermediate tool-use decisions drive success or failure. In this paper, we propose PORTool, an importance-aware policy-optimization algorithm that reinforces agents' tool-use competence from outcome-level supervision while assigning reward at the step level. Specifically, PORTool generates a rewarded rollout tree in which trajectories share prefixes before branching, enabling direct comparisons among alternative tool-use decisions within the same context. It then estimates each step's importance by a correctness-dominant signal, i.e., whether descendants of that step can ultimately produce a correct final answer, plus an auxiliary term indicating whether the step's tool calls satisfy formatting constraints and execute successfully. Using these step-wise importance estimates, PORTool updates the policy to generate efficient tool-call steps, guided by both local comparisons within each branching decision and the overall quality of entire trajectories. Experiments show that PORTool improves final-answer accuracy while reducing tool-call steps compared with state-of-the-art policy-optimization baselines, and ablation studies confirm the robustness of the proposed step-wise importance estimates.
format Preprint
id arxiv_https___arxiv_org_abs_2510_26020
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle PORTool: Importance-Aware Policy Optimization with Rewarded Tree for Multi-Tool-Integrated Reasoning
Wu, Feijie
Zhu, Weiwu
Zhang, Yuxiang
Chatterjee, Soumya
Zhu, Jiarong
Mo, Fan
Luo, Rong
Gao, Jing
Computation and Language
Artificial Intelligence
Machine Learning
Multi-tool-integrated reasoning enables LLM-empowered tool-use agents to solve complex tasks by interleaving natural-language reasoning with calls to external tools. However, training such agents from outcome-only rewards suffers from credit-assignment ambiguity, obscuring which intermediate tool-use decisions drive success or failure. In this paper, we propose PORTool, an importance-aware policy-optimization algorithm that reinforces agents' tool-use competence from outcome-level supervision while assigning reward at the step level. Specifically, PORTool generates a rewarded rollout tree in which trajectories share prefixes before branching, enabling direct comparisons among alternative tool-use decisions within the same context. It then estimates each step's importance by a correctness-dominant signal, i.e., whether descendants of that step can ultimately produce a correct final answer, plus an auxiliary term indicating whether the step's tool calls satisfy formatting constraints and execute successfully. Using these step-wise importance estimates, PORTool updates the policy to generate efficient tool-call steps, guided by both local comparisons within each branching decision and the overall quality of entire trajectories. Experiments show that PORTool improves final-answer accuracy while reducing tool-call steps compared with state-of-the-art policy-optimization baselines, and ablation studies confirm the robustness of the proposed step-wise importance estimates.
title PORTool: Importance-Aware Policy Optimization with Rewarded Tree for Multi-Tool-Integrated Reasoning
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2510.26020