Dissecting AI

All papers

Training Language Models to Fail Gracefully

Daman Arora, Xuecheng Liu, Himanshu Gaurav Singh, Fahim Tajwar, Abitha Thankaraj, Ruslan Salakhutdinov et al.

arXiv 2610.20401 41 pages Read the paper

The short version

The problem
When an artificial intelligence system is unsure, ordinary training often rewards it for guessing because only a correct answer earns credit. That can leave people to catch mistakes, or let a bad action damage files, data, or other work.
The idea
The paper trains the system to choose among three outcomes: succeed, use a safe fallback, or make an error. Its training method is designed to keep learning even if the system starts by always choosing the fallback or making mistakes, and it learns when to back off from the task’s results rather than from special labels saying when to quit. For example, a math model can still answer problems it can solve while saying it does not know on ones it is likely to get wrong.
What it got
On a challenging math test, the method reduced wrong answers from 79.1% to 8.9%, while success fell by only 3.1 percentage points compared with training that rewards success alone.

Module from this paper

GracefulRL: Learning to Fall Back Without Getting Stuck

Expected-reward training can get trapped in fallback or error; GracefulRL uses weighted log-probabilities of success and non-error to preserve a recovery signal.

reinforcement-learningpolicy-gradientslanguage-modelsoptimization