arXiv:2501.19201cs.CLcs.AI2025-01被引 57

将复杂推理压缩成精简思维令牌,提升效率且不损失准确率

Efficient Reasoning with Hidden Thinking

  • 用抽象思维令牌压缩长文本推理过程
  • 压缩后零样本准确率保持甚至提升
  • 可还原推理链,验证信息损失极小

链式思维(CoT)推理已成多模态大模型解决复杂问题的强大框架,但其冗长的文本形式带来显著效率瓶颈。本文提出Heima(隐藏小羊),一种高效的CoT压缩框架,将长篇推理压缩为少量抽象思维令牌,在保留核心推理逻辑的同时消除冗余。我们从信息论角度进行理论分析,量化压缩带来的信息损失,证明只要保留非平凡的互信息,推理能力即可维持。为深入探索该信息差距,设计自适应解释器,可将压缩后的思维令牌映射回变长文本序列,重建原始推理过程。在多个推理基准测试中,Heima显著提升推理效率,同时保持或超越原有零样本准确率。解释器成功重构连贯推理路径,表明信息损失微乎其微,验证了该框架的有效性。本工作为可扩展的隐式推理模型铺平道路,深化了对大模型高效推理机制的理解。

原文摘要 · Abstract (English)

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 Heima (as hidden llama), an effective CoT compression framework that condenses lengthy CoTs into a small set of abstract thinking tokens, preserving essential reasoning while removing redundancy. We then conduct a theoretical analysis from an information-theoretic perspective, quantifying the information gap induced by compression, showing that reasoning capability is preserved when non-trivial mutual information is retained. To further explore and quantify this information gap, we design the adaptive interpreter that maps thinking tokens back to variable-length textual sequences, thereby reconstructing the reasoning process. Experiments across diverse reasoning benchmarks demonstrate that Heima improves reasoning efficiency, while maintaining or even achieving better zero-shot accuracy. Moreover, the interpreter reconstructs coherent reasoning progresses from compressed thinking tokens, revealing that the information gap is minimal and validating the effectiveness of the proposed framework. This work paves the way for scalable latent reasoning models and advances our understanding of efficient reasoning processes in large models. Code: https://github.com/shawnricecake/Heima

推理压缩链式思维信息论大模型

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