arXiv:2601.21598cs.AI2026-01被引 1

用强化学习优化大模型隐空间推理,提升准确率并减少 token 消耗。

Beyond Imitation: Reinforcement Learning for Active Latent Planning

  • 将隐空间推理建模为条件变分自编码器,生成更平滑的潜在表示。
  • 引入一致性奖励驱动强化学习,使推理过程更合理,准确率提升4.1%。
  • 适合追求高效、高精度逻辑推理的开发者和研究者使用。

为实现高效且密集的思维链(CoT)推理,隐空间推理方法通过微调大语言模型(LLM),用连续隐变量替代离散语言标记,相比传统语言式思维链可减少标记消耗,并具备在密集隐空间中规划的潜力。然而,当前隐变量通常基于模仿语言标签进行监督,而同一问题可能存在多种等价但不同的思维链标签,被动模仿任意一种可能导致隐变量表示和推理策略质量下降,造成训练与测试间的明显差距。本文强调主动规划在隐变量表示空间中的重要性,提出主动隐空间规划方法(ATP-Latent)。该方法将隐变量的监督过程建模为条件变分自编码器(VAE),获得更平滑的隐空间;同时,通过基于解码内容一致性的辅助连贯性奖励,开展强化学习(RL),引导生成最优推理策略。在 LLaMA-1B 上的实验表明,ATP-Latent 在四个基准测试中相较先进基线实现 +4.1% 准确率提升和 -3.3% 标记消耗。代码已开源:https://github.com/zz1358m/ATP-Latent-master。

原文摘要 · Abstract (English)

Aiming at efficient and dense chain-of-thought (CoT) reasoning, latent reasoning methods fine-tune Large Language Models (LLMs) to substitute discrete language tokens with continuous latent tokens. These methods consume fewer tokens compared to the conventional language CoT reasoning and have the potential to plan in a dense latent space. However, current latent tokens are generally supervised based on imitating language labels. Considering that there can be multiple equivalent but diverse CoT labels for a question, passively imitating an arbitrary one may lead to inferior latent token representations and latent reasoning policies, undermining the potential planning ability and resulting in clear gaps between training and testing. In this work, we emphasize the importance of active planning over the representation space of latent tokens in achieving the optimal latent reasoning policy. So, we propose the \underline{A}c\underline{t}ive Latent \underline{P}lanning method (ATP-Latent), which models the supervision process of latent tokens as a conditional variational auto-encoder (VAE) to obtain a smoother latent space. Moreover, to facilitate the most reasonable latent reasoning policy, ATP-Latent conducts reinforcement learning (RL) with an auxiliary coherence reward, which is calculated based on the consistency between VAE-decoded contents of latent tokens, enabling a guided RL process. In experiments on LLaMA-1B, ATP-Latent demonstrates +4.1\% accuracy and -3.3\% tokens on four benchmarks compared to advanced baselines. Codes are available on https://github.com/zz1358m/ATP-Latent-master.

隐空间推理强化学习大模型思维链

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。