Learn when successful attempts become scarce
MENTOR addresses a failure mode in reinforcement learning with verifiable rewards: training can stall when a model receives zero reward on too many attempts. As task difficulty increases, successful examples become scarce and the learning signal can disappear.
The authors argue that failed reasoning trajectories still contain useful diagnostic information. An unsuccessful attempt may reveal the point where reasoning went wrong, a missing concept or an incorrect strategy. Discarding that information leaves the learner without a targeted account of what to change.
Add instruction tied to the current failure
MENTOR introduces a live, on-policy mentor that examines failed trajectories, diagnoses their underlying errors and generates instructional prompts for the learner's next attempt. It integrates with GRPO without requiring changes to the training infrastructure, according to the paper.
The distinction is timing and specificity. Feedback follows the learner's current attempts instead of relying solely on a fixed set of demonstrations prepared before training. The authors evaluate the approach on mathematical and coding tasks and report reduced zero-reward collapse and improved performance.
Their headline improvement includes an average relative gain of up to 78.3% for Llama-3.2-3B-Instruct over its base model. A relative gain should not be interpreted as the same number of percentage points or as a result that applies to every model.
Evaluate the diagnosis as well as the answer
Franklin analysis: a useful replication would retain the failed attempt, mentor diagnosis, instructional prompt and subsequent answer together. That makes it possible to examine whether improvement followed a correct diagnosis or merely additional sampling.
The added mentor also belongs in the accounting of training resources. A comparison should disclose how much extra instruction and computation it uses. The study supports testing failure-driven teaching under its reported experimental conditions; it does not establish that every difficult task can be solved by supplying another hint.
Comments