arXiv:2510.03632cs.AI2025-10KDD被引 6

用信息论方法提升大模型推理效率,不靠试错也能判断每步好坏。

MITS: Enhanced Tree Search Reasoning for LLMs via Pointwise Mutual Information

  • 用点互信息(PMI)量化每步推理质量,指导搜索方向。
  • 在多个推理基准上超越基线,准确率更高且更省计算资源。
  • 适合需要高效可靠推理的AI系统开发者或研究者使用。

树搜索已成为大语言模型测试时推理的代表性框架,如思维树和蒙特卡洛树搜索。然而,对中间推理步骤的质量进行即时可靠的定量评估仍具挑战性,且大量路径探索带来高昂计算成本。为此,我们提出互信息树搜索(MITS),一种基于信息论原理引导推理的新框架。MITS引入基于点互信息(PMI)的有效评分函数,实现推理路径的逐步评估,并通过束搜索实现树扩展,无需昂贵的前向预测模拟,从而在保持计算效率的同时取得更优推理性能。该框架还配备基于熵的动态采样策略,自适应地将计算资源分配给不确定性高的推理步骤,以最大化探索收益。最终预测采用加权投票机制,结合PMI得分与预测一致性。在多个推理基准上的全面实验表明,MITS持续优于基线方法,建立了一个原理清晰、高效的大型语言模型推理框架。代码已开源:https://github.com/plusnli/MITS。

原文摘要 · Abstract (English)

Tree search has become as a representative framework for test-time reasoning with large language models (LLMs), exemplified by methods such as Tree-of-Thought and Monte Carlo Tree Search. However, it remains difficult to provide instant and reliable quantitative assessments of intermediate reasoning step quality, and extensive path exploration is computationally costly. To address this, we propose Mutual Information Tree Search (MITS), a novel framework that guides reasoning with information-theoretic principles. MITS introduces an effective scoring function based on pointwise mutual information (PMI), which enables step-wise evaluation of reasoning paths and search tree expansion via beam search without expensive look-ahead simulations, achieving superior reasoning performances while maintaining computational efficiency. The framework is complemented by an entropy-based dynamic sampling strategy that adaptively allocates computational resources to uncertain reasoning steps where exploration is most beneficial. For final prediction, MITS employs a weighted voting scheme that combines PMI scores with prediction consensus. Through comprehensive experiments on diverse reasoning benchmarks, MITS consistently surpasses baseline methods, establishing a principled and efficient framework for LLM reasoning. The code is available at https://github.com/plusnli/MITS.

大模型推理树搜索信息论PMI

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