Lune

ICML2026顶会

G2^2RPO: Geometric GRPO; Escaping LLM's Reasoning Rut to Break Accuracy--Entropy Trade-off

Ali Rad, Khashayar Filom, Darioush Keivan, Peyman Mohajerin Esfahani, Ehsan Kamalinejad

出版方
2026年份

摘要

Reinforcement learning with verifiable rewards (RLVR) is a cornerstone of post-training for large reasoning models, yet widely used algorithms such as Group Relative Policy Optimization (GRPO) often exhibit diversity collapse. We provide a geometric diagnosis by formalizing GRPO as a dynamical flow on the probability simplex. Under a mode-based coarse-graining of rollouts, we show that GRPO induces a collision field over correct modes, monotonically pushing towards simplex vertices and thus yielding a winner-take-all regime. To address this systematically, we introduce G2^2RPO (Geometric GRPO), which reshapes RLVR via principled vector-field editing. Concretely, we intervene at the advantage level by adding granularity bonuses inversely proportional to mode probabilities, encouraging underrepresented correct modes. The bonus has a natural geometric interpretation, and its potential performance side effects can be mitigated, thereby avoiding the usual accuracy--diversity trade-off. In experiments with 7B and 14B models trained on a math reasoning task and evaluated on AIME 2024/2025, GRPO loses up to 57% of active correct modes. In contrast, G2^{2}RPO increases active correct-mode coverage by 172%--205%, reduces concentration on any single correct mode, prevents the late-stage entropy crash, and improves pass@1 by +1.4 to +7.9 points relative to GRPO. Overall, diversity is not merely a regularizer but a geometric property to be controlled to improve the model without trapping it in a single dominant strategy.

问问这篇 Paper

智能体会读完全文。

Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。

可以从这些问题问起

智能体调用

Luneget_paper_fulltext

在 Lune 里问

免费开始,无需绑卡

它引用的顶会 Paper9

相关 Paper

黄昏的海面,两侧是细线勾勒的悬崖