arXiv:cs.CL· Jiwon Jeon, Myungsik Cho, Jeonghye Kim, Youngchul Sung·· 13 小时前
UPTURN:学习在何处重定向 LLM 推理以实现自纠正
Inject a Turn, Not a Token: Learning Where to Redirect LLM Reasoning for Self-Correction
AI 导读
研究者提出 UPTURN 框架,用 130K 参数 MLP 头在冻结编码器上定位失败推理的干预位置,再以新的用户轮次注入指令并重新生成后续内容,无需微调 LLM。
正文
Abstract:Large language models struggle to reliably identify and correct errors in their own reasoning. Unlike prior training-based approaches that fine-tune an LLM to generate revisions or feedback, we learn where to intervene in failed rollouts and determine how to redirect them. To choose where to intervene, we train a 130K-parameter MLP head on a frozen encoder using only final-answer correctness. For redirection, we find that under a fixed generation budget, the tested inline injections fail to reliably redirect reasoning, whereas presenting the same instruction in a new user turn changes the model's course. The resulting correction framework, UPTURN (User-Prompt TURN injection at a learned position), requires no LLM fine-tuning: it retains a selected prefix, injects an instruction as a new user turn, and regenerates the suffix. Across four reasoners and six math benchmarks, UPTURN improves average accuracy in every setting over a 32-rollout baseline. It outperforms self-correction and reflection-token injection baselines under matched correction budgets, transfers across reasoners and to GPQA-Diamond without retraining, and complements GRPO and self-correction methods.
| Subjects: | Computation and Language (cs.CL); Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2610.13672 [cs.CL] |
| (or arXiv:2610.13672v1 [cs.CL] for this version) | |
| https://doi.org/10.48550/arXiv.2610.13672 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Jiwon Jeon [view email]
[v1]
Fri, 9 Oct 2026 14:37:51 UTC (1,694 KB)
来源:arXiv:cs.CL · arxiv.org