Selective Off-Policy Reference Tuning with Plan Guidance

arXiv:2605.11505 · cs.AI · Submitted 2026-05-12 · Read on arXiv

cs.AI

Submitted: 2026-05-12

Updated: 2026-09-24

License: http://creativecommons.org/licenses/by/4.0/

The gist: Reinforcement learning with verifiable rewards helps reasoning, but GRPO-style methods stall on hard prompts where all sampled rollouts fail.

Terminology

Abstract

Reinforcement learning with verifiable rewards helps reasoning, but GRPO-style methods stall on hard prompts where all sampled rollouts fail. SORT adds a repair update for those failures without changing rollout generation: it derives a plan from the reference solution, compares token probabilities with and without that plan, and gives higher weight to tokens that become more predictable under plan conditioning. This turns all-wrong prompts into selective, structure-aware learning signals instead of uniform imitation. Across three backbones and eight reasoning benchmarks, SORT improves over GRPO and guidance baselines, with largest gains on weaker models.

Sources

Related papers