NeFut Logo NeFut
中 Admin Login

[CS.AI] Matched-Context Group Policy Optimization for LLM Agent Training

Published at: 2026-10-14 22:00 Last updated: 2026-10-18 08:05
#AI #Machine Learning #LLM

Training long‑horizon language model agents requires assigning credit to intermediate decisions from sparse terminal feedback. Conventional approaches often rely on a global critic or extra reward models, yet useful comparisons between decisions in existing rollouts depend on their context, the evidence for each alternative, and how that evidence is fed into learning.

We introduce Matched‑Context Group Policy Optimization (MCGPO), a procedure that builds weighted action preferences from already collected trajectories. MCGPO first defines comparison contexts using pre‑action state summaries and observed rubric progress, annotated by task rules. A two‑action support mechanism selects between exact matching and coarse matching rules, establishing ordered comparisons within a finite group of sibling rollouts.

To avoid leakage, MCGPO separates candidate decisions from the terminal outcomes used to orient their preferences into fixed folds. Within each fold, evidence‑dependent weights are computed; these weights scale an auxiliary pairwise objective alongside the shared GraphGPO task learning loss.

The main advantage is that MCGPO reuses rollout data and the full history of each response, preserving a critic‑free training pipeline. Experiments span six environment‑backbone settings, each with three random seeds. Results show:

Three‑seed factorial controls confirm the benefit of opposite‑fold evidence and evidence weighting; across all four representation‑control settings, joint state‑rubric matching attains the highest mean performance.

Review

Original Source: https://arxiv.org/abs/2610.13755

[h] Back to Home