arXiv Machine Learning

Training Language Models to Fail Gracefully

The paper proposes a method called GracefulRL to train language models to fail gracefully. By reshaping the reinforcement learning optimization landscape, the models learn to preserve successful behavior while executing a safe fallback policy when actions are unreliable, all from outcome-level reward signals without extra annotations or inference-time checks. Experiments across various tasks and scales show that models maintain task-solving capability while selectively falling back when success is unlikely.

arXiv Computation and Language
3d ago

Reward Misspecification Degrades Chain-of-Thought Monitorability

The paper investigates how reward misspecification in reinforcement learning harms the ability to monitor chain-of-thought (CoT) reasoning in language models. By creating a testbed with true and misspecified rewards, the authors show that models can hack the reward while still producing CoTs that appear aligned, leading to low monitorability. They propose a mitigation strategy that collects diverse reward rollouts, infers a reward description with a language model, and injects this description into the input, improving monitorability by up to 42% absolute across models and tasks.

By Tianyi Lorena Yan, John Hewitt
arXiv Computation and Language
2d ago

Mitigating Reward Hacking in Preference Reward Models with Evolving Penalties

The paper introduces Dynalize, an online penalty system that mitigates reward hacking in preference reward models during policy training. By having a language model periodically analyze rollouts and generate penalties for behaviors the reward model inadvertently rewards, Dynalize prevents the policy from inflating scores without improving quality. Experiments in clinical counseling and creative writing show that Dynalize sustains longer training and yields better downstream performance, suppressing undesirable patterns even in multi‑turn settings and after retraining the reward model.

By Afonso Baldo, Ant\'onio Farinhas, Marcos Treviso, Ricardo Rei, Nuno M. Guerreiro
arXiv AI
Oct 12

MENTOR: Learning from Failure with On-Policy Pedagogical Intervention

The paper introduces MENTOR, a multi‑agent framework that provides on‑policy pedagogical intervention to address zero‑reward collapse in reinforcement learning from verifiable rewards. MENTOR diagnoses failures in real time and generates targeted prompts to guide the learning agent, integrating seamlessly with existing policy optimization pipelines such as GRPO. Experiments on math and code generation tasks show that MENTOR significantly reduces collapse, achieving up to 78.3% relative gains on Llama‑3.2‑3B‑Instruct.

By Guanlan Yue, Hang Yu, Xiaoyong Huang, Ruipeng Li, Wei Zhang, Bingchang Liu, Rui Wang
arXiv Machine Learning
Jun 2

Advantage Collapse in Group Relative Policy Optimization: Diagnosis and Mitigation

arXiv:2605. 21125v2 Announce Type: replace Abstract: Group Relative Policy Optimization (GRPO), a prominent algorithm within the Reinforcement Learning from Verifiable Rewards (RLVR) framework, has achieved strong results in improving the reasoning capabilities of large language models (LLMs).

By Xixiang He, Qiyao Sun, Ao Cheng, Xingming Li, Xuanyu Ji, Hailun Lu, Runke Huang, Qingyong Hu
Hugging Face Trending Papers
Jul 12

To Answer or to Abstain: Mitigating Search-Agent Hallucinations via Abstention-Aware Reinforcement Learning

Recent advances in equipping Large Language Models (LLMs) with search tools and outcome-reward reinforcement learning (RL) have achieved new state-of-the-art results on open-domain QA tasks. However, we argue that current training paradigms harbor a critical vulnerability: they predominantly reward correct answers but fail to penalize fabricated ones when retrieval fails, thereby implicitly exacerbating hallucinations.

arXiv AI
Jun 19

Beyond Reasoning Gains: Mitigating General-Capability Forgetting in Large Reasoning Models

arXiv:2510. 21978v2 Announce Type: replace-cross Abstract: Reinforcement learning with verifiable rewards (RLVR) has delivered impressive gains in mathematical and multimodal reasoning and has become a standard post-training paradigm for contemporary language and vision-language models.

By Hoang Phan, Xianjun Yang, Yuanshun Yao, Jingyu Zhang, Shengjie Bi, Xiaocheng Tang, Madian Khabsa, Lijuan Liu, Deren Lei