mlsanityData to 7 Oct 2026

TRIAGE: Role-Typed Credit Assignment for Agentic Reinforcement Learning

arXiv 2606.32017 ↗First posted 30 Jun 2026cs.LG cs.AIcited by 5

On the map: LLM agents and tool use → Reinforcement learning for LLM agents

Abstract (arXiv)

Agentic reinforcement learning requires assigning credit to environment-facing actions such as searches, clicks, edits, navigation commands, and object interactions. Standard GRPO uses the final verifier outcome as a uniform advantage over all action tokens. This outcome signal is useful but structurally incomplete: it punishes useful exploration in failed rollouts and reinforces redundant or regressive actions in successful rollouts. We propose TRIAGE, a role-typed credit assignment framework that adds a semantic role axis to outcome credit. A structured judge classifies each segment as decisive progress, useful exploration, no-progress infrastructure, or regression, and a fixed role-conditioned rule maps these labels to bounded segment-level process rewards. This keeps verifier outcomes as the source of optimization direction while correcting the two main blind spots of outcome-only credit. We further show that the Bayes-optimal role-measurable correction is the L2 projection of the per-segment advantage residual onto the role variable, and that TRIAGE's fixed role constants approximate this projection, reducing advantage estimation error whenever the judge is reliable; we connect this to lower-variance policy gradients. Across ALFWorld, Search-QA, and WebShop, TRIAGE improves success rates over GRPO for two policy models and outperforms both a scalar judge-derived process reward and an outcome-supervised shared-backbone value baseline. Ablations show that the gain comes from role typing rather than merely adding dense rewards: reliable detection of regression inside successful trajectories is the dominant contributor, while exploration credit provides a consistent secondary gain; on completed ALFWorld and WebShop rollouts, TRIAGE also reduces environment-facing turns by an additional 10.4% and 14.8% relative to GRPO.

Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Placed in Reinforcement learning for LLM agents, one of 16 directions in LLM agents and tool use. Builds on 27 of its 33 references that are in the map. First posted on arXiv 30 Jun 2026.

Builds on

27 of 33 references.

Closest ancestors · 4

Same direction, newest first

  1. 2026Self-Distilled Agentic Reinforcement Learning14 May
  2. 2025SPA-RL: Reinforcing LLM Agents via Stepwise Progress Attribution27 May
  3. 2025Group-in-Group Policy Optimization for LLM Agent Training16 May
  4. 2023Reflexion: Language Agents with Verbal Reinforcement Learning20 Mar

Other references · 23

  1. 2017Proximal Policy Optimization Algorithmscited by 14,441
  2. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  3. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
  4. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
  5. 2024Qwen2.5 Technical Reportcited by 4,713
  6. 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
  7. 2023Let's Verify Step by Stepcited by 3,598
  8. 2023Self-Refine: Iterative Refinement with Self-Feedbackcited by 3,317
  9. 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,155
  10. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
  11. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
  12. 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 1,916
11 more
  1. 2022WebShop: Towards Scalable Real-World Web Interaction with Grounded Language Agentscited by 1,197
  2. 2020ALFWorld: Aligning Text and Embodied Environments for Interactive Learningcited by 1,143
  3. 2022STaR: Bootstrapping Reasoning With Reasoningcited by 895
  4. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
  5. 2024ReFT: Reasoning with Reinforced Fine-Tuningcited by 236
  6. 2017SearchQA: A New Q&A Dataset Augmented with Context from a Search Enginecited by 221
  7. 2026TIP: Token Importance in On-Policy Distillationcited by 90
  8. 2026CRISP: Compressed Reasoning via Iterative Self-Policy Distillationcited by 49
  9. 2026Rubric-based On-policy Distillationcited by 11
  10. 2026Beyond GRPO and On-Policy Distillation: An Empirical Sparse-to-Dense Reward Principle for Language-Model Post-Trainingcited by 9
  11. 2026Dismantling Pathological Shortcuts: A Causal Framework for Faithful LVLM Decodingcited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 5.

5 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026From Reasoning to Agentic: Credit Assignment in Reinforcement Learning for Large Language Models
  2. 2026AgentOPSD: Recursive Self-Distillation for Agentic Reinforcement Learning
  3. 2026TRACE: Turn-level Reward Assignment via Credit Estimation for Long-Horizon Agents
  4. 2026ASCT: Attentive Search over Counterfactual Trees for Credit Assignment in Agentic Reinforcement Learning
  5. 2026CRAFT: Counterfactual Credit Assignment from Free Sibling Rollouts for Self-Distilled Agentic Reinforcement Learning
  6. 2026Learning from Environmental Feedback: Credit Assignment across Multiple Timescales for Agentic Reinforcement Learning
  7. 2026My FAULT: Self-Diagnosis as Credit Assignment in Self-Evolving Agentic Reinforcement Learning
  8. 2026Keep Policy Gradient in Charge: Sibling-Guided Credit Distillation for Long-Horizon Tool-Use Agents