paper-with-me

홈 › Papers

IAPO: Influence-Aware Policy Optimization for Credit Assignment in Multi-Turn Service Agents

2026-08-25 · Bo Ren, Yirong Mao, Yi Yang, Wenhui Que arxiv

Large Language Model (LLM) agents increasingly solve long-horizon tasks through multi-turn interactions with users and external tools. In these settings, relevant task information often unfolds over time rather than being fully specified at the initial prompt. Service agents make this challenge especially concrete: users may clarify or revise their goals, while tool responses provide information needed for subsequent decisions. Thus, a final reward alone cannot indicate which actions contributed to resolving the task. Recent methods rely on comparative evidence from other trajectories or resampled continuations, or on separately constructed step-level learning signals, to refine credit. However, a completed rollout already records how information and errors flow between agent actions. We introduce Influence-Aware Policy Optimization (IAPO), which represents each rollout as a typed influence-dependency graph over trainable agent actions, with user and tool observations serving as evidence. IAPO converts support-use and failed-use structure into routing weights that redistribute the same trajectory-level advantage. Experiments with Qwen3-4B and Qwen3-8B demonstrate superior performance over multi-turn reinforcement learning (RL) baselines across three service-agent benchmarks: ${τ^2}$-Bench, UserBench, and AgentChangeBench. BFCL-v4 Multi-Turn further shows that these gains do not compromise multi-turn function-calling performance. This work advances the understanding of credit assignment in multi-turn user interactions and provides a principled approach to training service agents from sparse outcome feedback.

📄 PDF Abstract BibTeX arXiv:2608.24588

Code (0)

등록된 구현이 없습니다.

Tasks

Reinforcement Learning

Similar Papers 제목 키워드 기반

IAPO: Information-Aware Policy Optimization for Token-Efficient Reasoning

2026-02-22 · Yinhan He, Yaochen Zhu, Mingjia Shi, Wendy Zheng 외 arxiv

Large language models increasingly rely on long chains of thought to improve accuracy, yet such gains come with substantial inference-time costs. We revisit token-efficient post-training and argue that existing sequence-…

Inference-Aware Prompt Optimization for Aligning Black-Box Large Language Models

2025-08-08 · Saaduddin Mahmud, Mason Nakamura, Kyle Hollins Wray, Shlomo Zilberstein arxiv

Prompt optimization methods have demonstrated significant effectiveness in aligning black-box large language models (LLMs). In parallel, inference scaling strategies such as Best-of-N Sampling and Majority Voting have li…

Text Generation

IAPO: Input Attribution-Aware Policy Optimization for Tool Use in Small Multimodal Agents

2026-06-10 · Yifan Yang, Zhen Zhang, Jiayi Tian, Liyan Tan 외 arxiv

This paper investigates reinforcement learning (RL) methods for improving tool-calling capabilities in multimodal small language model (SLM) agents. While existing works have explored various reward designs to improve ag…

Visual Question AnsweringReinforcement Learning

UniAPO: Unified Multimodal Automated Prompt Optimization

2025-08-25 · Qipeng Zhu, Yanzhe Chen, Huasong Zhong, Yan Li 외 arxiv

Prompting is fundamental to unlocking the full potential of large language models. To automate and enhance this process, automatic prompt optimization (APO) has been developed, demonstrating effectiveness primarily in te…

Structured Role-Aware Policy Optimization for Multimodal Reasoning

2026-05-08 · Bingqing Jiang, Difan Zou arxiv

Reinforcement learning from verifiable rewards (RLVR), especially with Group Relative Policy Optimization (GRPO), has shown strong potential for improving the reasoning capabilities of large vision-language models (LVLMs…

Reinforcement LearningMultimodal Reasoning