IAPO: Influence-Aware Policy Optimization for Credit Assignment in Multi-Turn Service Agents
cs.LG
Submitted: 2026-08-25
Updated: 2026-08-26
Comments: 12 pages, 3 figures, 5 tables. Preprint
License: http://creativecommons.org/licenses/by-nc-nd/4.0/
The gist: Large Language Model (LLM) agents increasingly solve long-horizon tasks through multi-turn interactions with users and external tools.
Terminology
Abstract
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: τ squared-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.
Sources
- $\tau^2$-Bench: Evaluating Conversational Agents in a Dual-Control Environment
- Process Reinforcement through Implicit Rewards
- DeepSeek-V3 Technical Report
- Tree Search for LLM Agent Reinforcement Learning
- Step-DPO: Step-wise Preference Optimization for Long-chain Reasoning of LLMs
- Outcome-Grounded Advantage Reshaping for Fine-Grained Credit Assignment in Mathematical Reasoning
- Understanding R1-Zero-Like Training: A Critical Perspective
- GPT-4 Technical Report
- GRPO-$\lambda$: Credit Assignment improves LLM Reasoning
- UserBench: An Interactive Gym Environment for User-Centric Agents
- UserRL: Training Interactive User-Centric Agent via Reinforcement Learning
- AgentChangeBench: A Multi-Dimensional Evaluation Framework for Goal-Shift Robustness in Conversational AI
- Credit Assignment with Resets in Language Model Reasoning
- Proximal Policy Optimization Algorithms
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Hindsight Credit Assignment for Long-Horizon LLM Agents
- Gemini: A Family of Highly Capable Multimodal Models
- MINT: Evaluating LLMs in Multi-turn Interaction with Tools and Language Feedback
- RAGEN: Understanding Self-Evolution in LLM Agents via Multi-Turn Reinforcement Learning
- Reinforcing Multi-Turn Reasoning in LLM Agents via Turn-Level Reward Design
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks