Reasoning Cache: Continual Improvement Over Long Horizons via Short-Horizon RL
AI 摘要
RC算法通过迭代解码,利用LLM的生成和总结能力,实现推理链的持续改进,提升模型在长推理任务上的性能。
主要贡献
- 提出了一种新的迭代解码算法RC
- 证明RC可以提升模型在长推理任务上的外推能力
- 验证了RC在HMMT 2025数据集上的有效性
方法论
RC算法利用LLM的response generation和summarization能力,迭代构建推理链,并通过RL训练,使模型能在更长的推理horizon上持续改进。
原文摘要
Large Language Models (LLMs) that can continually improve beyond their training budgets are able to solve increasingly difficult problems by adapting at test time, a property we refer to as extrapolation. However, standard reinforcement learning (RL) operates over fixed problem distributions and training budgets, which limits extrapolation amidst distribution shift at test time. To address this, we introduce RC, an iterative decoding algorithm that replaces standard autoregressive decoding during both training and inference. RC exploits an asymmetry between the response generation and summarization capabilities of LLMs to construct reasoning chains that consistently improve across iterations. Models trained to use RC can extrapolate and continually improve over reasoning horizons more than an order of magnitude longer than those seen during training. Empirically, training a 4B model with RC using a 16k-token training budget improves performance on HMMT 2025 from 40% to nearly 70% with 0.5m tokens at test time, outperforming both comparably sized models and many larger reasoning LLMs. Finally, we also show that models trained with RC can more effectively leverage existing scaffolds to further scale test-time performance, due to the improved summary-conditioned generation abilities learned through training.