Efficient Reasoning with Hidden Thinking

LLM Latent Reasoning CoT Multimodal Intelligence Vision-Language Pre-training Visual CoT
链式思维(CoT)推理已经成为提升多模态大语言模型(MLLMs)复杂问题解决能力的强大框架。然而,文本推理的冗长性引入了显著的效率低下问题。在本研究中,我们提出了一种高效的推理框架——**Heima**(作为隐藏的LLaMA),该框架利用隐藏的潜在空间中的链式思维进行推理。我们设计了Heima编码器,通过单一的思考标记将每个中间的CoT浓缩成紧凑且高层次的隐藏表示,从而有效减少冗长性,并减少推理过程中所需的总标记数量。同时,我们设计了与传统大语言模型(LLMs)对应的Heima解码器,以自适应地将隐藏表示解释为可变长度的文本序列,重建与原始CoT高度相似的推理过程。跨多个不同推理基准的实验结果表明,Heima模型在保持甚至提高零样本任务准确性的同时,实现了更高的生成效率。此外,使用Heima解码器对多模态推理过程的有效重建验证了我们方法的稳健性和可解释性。
Chain-of-Thought (CoT) reasoning has become a powerful framework for improving complex problem-solving capabilities in Multimodal Large Language Models (MLLMs). However, the verbose nature of textual reasoning introduces significant inefficiencies. In this work, we propose $\textbf{Heima}$ (as hidden llama), an efficient reasoning framework that leverages reasoning CoTs at hidden latent space. We design the Heima Encoder to condense each intermediate CoT into a compact, higher-level hidden representation using a single thinking token, effectively minimizing verbosity and reducing the overall number of tokens required during the reasoning process. Meanwhile, we design corresponding Heima Decoder with traditional Large Language Models (LLMs) to adaptively interpret the hidden representations into variable-length textual sequence, reconstructing reasoning processes that closely resemble the original CoTs. Experimental results across diverse reasoning MLLM benchmarks demonstrate that Heima model achieves higher generation efficiency while maintaining or even better zero-shot task accuracy. Moreover, the effective reconstruction of multimodal reasoning processes with Heima Decoder validates both the robustness and interpretability of our approach.
许愿