Efficient Reasoning with Hidden Thinking

Xuan Shen, Yizhou Wang, Xiangxi Shi, Yanzhi Wang, Pu Zhao +1 more
1/31/2025
cs.CLcs.AIcs.LG

No official code found. Best implementation: shawnricecake/Heima (76 stars, last commit 2025-04-21).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

shawnricecake/HeimaCommunity100%

[ICML 2026] Heima

765Shell, PythonJan 30, 20251 years ago

Similar Papers

Cite this paper

@article{shen2025efficient,
  title  = {Efficient Reasoning with Hidden Thinking},
  author = {Xuan Shen and Yizhou Wang and Xiangxi Shi and Yanzhi Wang and Pu Zhao and Jiuxiang Gu},
  year   = {2025},
  eprint = {2501.19201},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2501.19201v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2501.19201.svg)](https://codeofpaper.com/paper/2501.19201)

Discussion