InfoPO: Information-Driven Policy Optimization for User-Centric Agents

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kong, Fanqi, Zhang, Jiayi, Deng, Mingyi, Wu, Chenglin, Luo, Yuyu, Liu, Bang
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918363341520896
author Kong, Fanqi
Zhang, Jiayi
Deng, Mingyi
Wu, Chenglin
Luo, Yuyu
Liu, Bang
author_facet Kong, Fanqi
Zhang, Jiayi
Deng, Mingyi
Wu, Chenglin
Luo, Yuyu
Liu, Bang
contents Real-world user requests to LLM agents are often underspecified. Agents must interact to acquire missing information and make correct downstream decisions. However, current multi-turn GRPO-based methods often rely on trajectory-level reward computation, which leads to credit assignment problems and insufficient advantage signals within rollout groups. A feasible approach is to identify valuable interaction turns at a fine granularity to drive more targeted learning. To address this, we introduce InfoPO (Information-Driven Policy Optimization), which frames multi-turn interaction as a process of active uncertainty reduction and computes an information-gain reward that credits turns whose feedback measurably changes the agent's subsequent action distribution compared to a masked-feedback counterfactual. It then combines this signal with task outcomes via an adaptive variance-gated fusion to identify information importance while maintaining task-oriented goal direction. Across diverse tasks, including intent clarification, collaborative coding, and tool-augmented decision making, InfoPO consistently outperforms prompting and multi-turn RL baselines. It also demonstrates robustness under user simulator shifts and generalizes effectively to environment-interactive tasks. Overall, InfoPO provides a principled and scalable mechanism for optimizing complex agent-user collaboration. Code is available at https://github.com/kfq20/InfoPO.
format Preprint
id arxiv_https___arxiv_org_abs_2603_00656
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle InfoPO: Information-Driven Policy Optimization for User-Centric Agents
Kong, Fanqi
Zhang, Jiayi
Deng, Mingyi
Wu, Chenglin
Luo, Yuyu
Liu, Bang
Artificial Intelligence
Real-world user requests to LLM agents are often underspecified. Agents must interact to acquire missing information and make correct downstream decisions. However, current multi-turn GRPO-based methods often rely on trajectory-level reward computation, which leads to credit assignment problems and insufficient advantage signals within rollout groups. A feasible approach is to identify valuable interaction turns at a fine granularity to drive more targeted learning. To address this, we introduce InfoPO (Information-Driven Policy Optimization), which frames multi-turn interaction as a process of active uncertainty reduction and computes an information-gain reward that credits turns whose feedback measurably changes the agent's subsequent action distribution compared to a masked-feedback counterfactual. It then combines this signal with task outcomes via an adaptive variance-gated fusion to identify information importance while maintaining task-oriented goal direction. Across diverse tasks, including intent clarification, collaborative coding, and tool-augmented decision making, InfoPO consistently outperforms prompting and multi-turn RL baselines. It also demonstrates robustness under user simulator shifts and generalizes effectively to environment-interactive tasks. Overall, InfoPO provides a principled and scalable mechanism for optimizing complex agent-user collaboration. Code is available at https://github.com/kfq20/InfoPO.
title InfoPO: Information-Driven Policy Optimization for User-Centric Agents
topic Artificial Intelligence
url https://arxiv.org/abs/2603.00656