mlsanityData to 7 Oct 2026

EBPO: Empirical Bayes Shrinkage for Stabilizing Group-Relative Policy Optimization

arXiv 2602.05165 ↗First posted 5 Feb 2026cs.LG cs.AI cs.CLcited by 4

On the map: LLM reasoning → Reinforcement learning for reasoning

Abstract (arXiv)

Reinforcement Learning with Verifiable Rewards (RLVR) has proven effective for enhancing the reasoning capabilities of Large Language Models (LLMs). However, dominant approaches like Group Relative Policy Optimization (GRPO) face critical stability challenges: they suffer from high estimator variance under computational constraints (small group sizes) and vanishing gradient signals in saturated failure regimes where all responses yield identical zero rewards. To address this, we propose Empirical Bayes Policy Optimization (EBPO), a novel framework that regularizes local group-based baselines by borrowing strength from the policy's accumulated global statistics. Instead of estimating baselines in isolation, EBPO employs a shrinkage estimator that dynamically balances local group statistics with a global prior updated via Welford's online algorithm. Theoretically, we demonstrate that EBPO guarantees strictly lower Mean Squared Error (MSE), bounded entropy decay, and non-vanishing penalty signals in failure scenarios compared to GRPO. Empirically, EBPO consistently outperforms GRPO and other established baselines across diverse benchmarks, including AIME and OlympiadBench. Notably, EBPO exhibits superior training stability, achieving high-performance gains even with small group sizes, and benefits significantly from difficulty-stratified curriculum learning.

Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 32 of its 38 references that are in the map. First posted on arXiv 5 Feb 2026.

Builds on

32 of 38 references.

Closest ancestors · 12

Same direction, newest first

  1. 2025Shrinking the Variance: Shrinkage Baselines for Reinforcement Learning with Verifiable Rewards5 Nov
  2. 2025Let it Calm: Exploratory Annealed Decoding for Verifiable Reinforcement Learning6 Oct
  3. 2025No Prompt Left Behind: Exploiting Zero-Variance Prompts in LLM Reinforcement Learning via Entropy-Guided Advantage Shaping26 Sep
  4. 2025EDGE-GRPO: Entropy-Driven GRPO with Guided Error Correction for Advantage Diversity29 Jul
  5. 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoning2 Jun
  6. 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Models28 May
  7. 2025DISCO Balances the Scales: Adaptive Domain- and Difficulty-Aware Reinforcement Learning on Imbalanced Data21 May
  8. 2025VAPO: Efficient and Reliable Reinforcement Learning for Advanced Reasoning Tasks7 Apr
  9. 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
  10. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
  11. 2023Let's Verify Step by Step31 May
  12. 2022Solving math word problems with process- and outcome-based feedback25 Nov

Other references · 20

  1. 2023GPT-4 Technical Reportcited by 18,132
  2. 2024The Llama 3 Herd of Modelscited by 14,192
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  4. 2025Qwen3 Technical Reportcited by 8,195
  5. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  6. 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
  7. 2022STaR: Bootstrapping Reasoning With Reasoningcited by 895
  8. 2025Group Sequence Policy Optimizationcited by 788
  9. 2018Recasting Gradient-Based Meta-Learning as Hierarchical Bayescited by 180
  10. 2024Large Language Models to Enhance Bayesian Optimizationcited by 127
  11. 2020Empirical Bayes Transductive Meta-Learning with Synthetic Gradientscited by 48
  12. 2019Specifying Weight Priors in Bayesian Deep Neural Networks with Empirical Bayescited by 32
8 more
  1. 2025Don't Pass@k: A Bayesian Framework for Large Language Model Evaluationcited by 15
  2. 2025Semantically-Aware Rewards for Open-Ended R1 Training in Free-Form Generationcited by 11
  3. 2025Heterogeneous Adaptive Policy Optimization: Tailoring Optimization to Every Token's Naturecited by 11
  4. 2025AutoRubric: Rubric-Based Generative Rewards for Faithful Multimodal Reasoningcited by 11
  5. 2026Token-Level LLM Collaboration via FusionRoutecited by 9
  6. 2025Confidence in Large Language Model Evaluation: A Bayesian Approach to Limited-Sample Challengescited by 7
  7. 2025Mixture-of-Minds: Multi-Agent Reinforcement Learning for Table Understandingcited by 6
  8. 2023An Empirical Bayes Framework for Open-Domain Dialogue Generationcited by 5

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Advantage Collapse in Group Relative Policy Optimization: Diagnosis and Mitigation
  2. 2026Stabilizing Reinforcement Learning for Diffusion Language Models
  3. 2026On the Hidden Objective Biases of Group-based Reinforcement Learning
  4. 2025Group-Relative REINFORCE Is Secretly an Off-Policy Algorithm: Demystifying Some Myths About GRPO and Its Friends
  5. 2026EP-GRPO: Entropy-Progress Aligned Group Relative Policy Optimization with Implicit Process Guidance
  6. 2025ESPO: Entropy Importance Sampling Policy Optimization
  7. 2026iGRPO: Self-Feedback-Driven LLM Reasoning
  8. 2026CoDistill-GRPO: A Co-Distillation Recipe for Efficient Group Relative Policy Optimization