Abstention Action Can Collapse Error-Penalized RL Training.

Xujun Che, Yuchen Yuan, Weida Zhao, Chenyang Yu· August 4, 2026 View original

Key takeaways

  • Allowing abstention as a discrete action in error-penalized RL can cause models to refuse all answers.
  • This "collapse law" kills reward gradients and KL anchors, leading to misleading performance metrics.
  • The advantage estimator can compound this failure by altering effective penalties.
  • A structural repair involves training a mandatory confidence report and abstaining only at deployment.

Who benefits

AI/ML DevelopmentCustomer ServiceHealthcare (diagnostics)Legal TechEducation Technology

Summary

This research proves that allowing abstention as a discrete action in error-penalized reinforcement learning can lead to a "collapse law," where the model drifts towards refusing all questions, killing both the reward gradient and KL anchor. It proposes a structural repair.

In reinforcement learning, error-penalized scoring rules (rewarding correct answers, penalizing wrong ones, and giving zero for abstaining) are increasingly used to combat issues like hallucination. However, this paper reveals a critical flaw: when abstention is treated as a discrete action, it can lead to a "collapse law." Under specific conditions, the reward gradient and the KL anchor's restoring force are throttled and die together, causing the model to drift towards refusing all questions. This makes it appear as if the model is improving (mean reward rises to zero) while its coverage collapses. The advantage estimator further exacerbates this failure, effectively replacing the designed penalty with a fixed penalty of one, shifting the learned threshold. The proposed solution is structural: train a mandatory confidence report using a strictly proper score alongside a correctness reward, and only apply abstention at deployment by thresholding this report. This ensures the report's gradient and anchor remain active, leading to improved coverage, accuracy, and calibration.

Why it matters

AI developers building systems with abstention capabilities (e.g., for hallucination control) must understand this collapse mechanism to avoid models that become overly cautious and refuse to answer, ensuring reliable and useful AI deployment.

How to implement this in your domain

  1. 1Review current reinforcement learning setups that incorporate abstention as a discrete action.
  2. 2Assess if your models exhibit signs of "collapse law," such as decreasing coverage despite rising average reward.
  3. 3Implement the proposed structural repair: train a separate, mandatory confidence report.
  4. 4Use a strictly proper score for the confidence report and apply abstention only at deployment via thresholding.
  5. 5Monitor coverage, accuracy, and calibration metrics during training and deployment.

Original post by Xujun Che, Yuchen Yuan, Weida Zhao, Chenyang Yu

"arXiv:2608.00301v1 Announce Type: new Abstract: Error-penalized scoring rules ($+1$ for a correct answer, $-\lambda$ for a wrong one, $0$ for abstaining) are increasingly prescribed against hallucination: a rational agent facing such a rule answers exactly when its correctness pr…"

View on X

Originally posted by Xujun Che, Yuchen Yuan, Weida Zhao, Chenyang Yu on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses