Cliff: Learning Process Rewards from the First Mistake
Cliff trains on the location of a rollout’s first reasoning error, rewarding the valid prefix and penalizing the suffix. It offers finer supervision without a specialized reward model.
**Cliff** asks an off-the-shelf teacher model to find the first mistake in each rollout. It converts the split into token-level advantages: positive for the correct prefix and negative after the error.
For verifiable agent tasks, capture where a trajectory first becomes invalid instead of assigning only a final pass/fail reward. This avoids spending supervision on reasoning already conditioned on a broken prefix.
**Cliff** asks an off-the-shelf teacher model to find the first mistake in each rollout. It converts the split into token-level advantages: positive for the correct prefix and negative after the error. For verifiable agent tasks, capture where a trajectory first becomes invalid instead of assigning only a final pass/fail reward. This avoids spending supervision on reasoning already conditioned on a broken prefix. Across **12 scenarios**, Cliff reportedly beat on-policy distillation by **15%** and standard GRPO by **7%**. The abstract does not specify task composition, compute overhead, or sensitivity to teacher misclassification.
This refines process supervision from scoring whole trajectories to identifying the first invalid step, preserving credit for the valid prefix and avoiding supervision on reasoning already corrupted by an error. It complements outcome-based agent evals, but its advantage depends on a teacher reliably locating that boundary; the supplied results do not quantify this sensitivity or added cost.