English

Selective Off-Policy Reference Tuning with Plan Guidance

Artificial Intelligence 2026-05-14 v2

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.

Keywords

Cite

@article{arxiv.2605.11505,
  title  = {Selective Off-Policy Reference Tuning with Plan Guidance},
  author = {Duc Anh Le and Tien-Phat Nguyen and Thien Huu Nguyen and Linh Ngo Van and Trung Le},
  journal= {arXiv preprint arXiv:2605.11505},
  year   = {2026}
}