发现递归推理本质是策略改进,可显著减少无效计算。
Latent Reasoning in TRMs is Secretly a Policy Improvement Operator
- 将潜空间递归视为策略改进算法,重构训练逻辑
- 在相同性能下减少18倍前向传播次数,避免冗余计算
- 适合研究递归模型优化与强化学习融合的学者
近期小型带潜空间递归的模型在复杂推理任务中表现优异。传统解释认为递归增加了网络深度,从而压缩大模型能力。然而,循环结构下的每步递归并未等效贡献深度,导致性能低于同等前馈深度的单次模型。这引发疑问:潜空间推理何时提升性能,何时造成无效计算?本文揭示,潜空间递归可形式化为策略改进算法。基于此,我们引入强化学习与扩散方法的训练策略。以小规模递归模型(Tiny Recursive Model)为基准,验证改进后可避免无效计算步骤,总前向传递次数减少18倍,同时保持原有性能。整体而言,策略改进视角有助于解释递归行为,并为模型优化提供新思路。
原文摘要 · Abstract (English)
Recently, small models with latent recursion have obtained promising results on complex reasoning tasks. These results are typically explained by the theory that such recursion increases a networks depth, allowing it to compactly emulate the capacity of larger models. However, the performance of recursively added layers remains behind the capabilities of one pass models with the same feed-forward depth. This means that in the looped version, not every recursive step effectively contributes to depth. This raises the question: when and why does latent reasoning improve performance, and when does it result in dead compute? In our work, we demonstrate that latent recursive reasoning provides answer to this question. We show that latent recursive reasoning can be formalized as a policy improvement algorithm. Building on these insights, we propose to use a training schemes from reinforcement learning and diffusion methods for latent reasoning models. Using the Tiny Recursive Model as our testbed, we show that with our modifications we can avoid dead compute steps and reduce the total number of forward passes by 18x while maintaining performance. Broadly speaking, we show how a policy improvement perspective on recursive steps can explain model behavior and provide insights for further improvements.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。