Papers
arxiv:2609.29050

SLCA-GRPO: Resolving Cross-Segment Credit Misattribution in Tool-Calling RL

Published on Sep 24
· Submitted by
Yan Zhan
on Sep 28
Authors:
,
,
,
,
,
,
,
,

Abstract

Tool-calling agents produce heterogeneous outputs, interleaving structured tool invocations with user-facing natural language summaries. This output heterogeneity presents a structural failure mode in standard on-policy Reinforcement Learning (RL): algorithms like GRPO indiscriminately broadcast a homogeneous trajectory-level scalar advantage to all tokens. Consequently, gradient noise from summary generation leaks into tool-decision tokens, causing cross-segment credit misattribution and brittle optimization. In this work, we propose SLCA-GRPO, a framework incorporating Segment-Locked Credit Assignment (SLCA). To enable scalable exploration without costly real APIs and stable training, we first construct the Schema-Guided LLM Simulator (SGLS) as foundational training infrastructure. Building on this, SLCA decouples advantage estimation at the structural segment level within a single group of rollouts, without requiring additional rollouts from intermediate states. Supported by Hierarchical Rewards (HierR), SLCA routes execution advantages to tool tokens and preference advantages to summary tokens, eliminating advantage contamination (the dominant cross-segment credit misattribution channel) within each policy update. On a 7B backbone, SLCA-GRPO accelerates convergence and outperforms standard GRPO, ToolPO, and RLTR by +2.53 pp on in-domain evaluation, +1.36 pp on the Berkeley Function-Calling Leaderboard (BFCL), and +9.15 pp on τ^2-Bench under the same training budgets, achieving higher accuracy with reduced tool redundancy and costs.

Community

Paper author Paper submitter

Tool-calling RL often broadcasts one trajectory-level GRPO advantage across both tool-call and summary tokens, allowing summary rewards to distort tool decisions. SLCA-GRPO (Segment-Locked Credit Assignment) normalizes tool and summary rewards independently within each rollout group and routes each advantage only to its own tokens, with no additional rollout cost. Across three Qwen backbones and in- and out-of-distribution tool-calling benchmarks, it improves task success while reducing tool turns.

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2609.29050
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2609.29050 in a model README.md to link it from this page.

Datasets citing this paper 1

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2609.29050 in a Space README.md to link it from this page.

Collections including this paper 1