让大模型分步思考并可控搜索,提升复杂问题求解能力
Thinker: Training LLMs in Hierarchical Thinking for Deep Search via Multi-Turn Interaction
- 将复杂问题拆解为可独立求解的子问题,用自然语言与逻辑函数双重表示
- 通过逻辑函数传递依赖关系,保证推理过程逻辑连贯,减少无效搜索
- 自动判断是否需外部检索,仅在必要时调用知识库或网页搜索
高效检索外部知识库和网页对增强大模型推理能力至关重要。以往训练大模型利用外部检索器解决复杂问题的方法多采用端到端强化学习,但忽视了对推理过程的监督,难以保证逻辑连贯性。为此,我们提出Thinker,一种通过多轮交互实现深度搜索的分层思考模型,使推理过程可监督、可验证。该模型将复杂问题分解为可独立求解的子问题,每个子问题同时以自然语言和等价逻辑函数表示,支持知识库与网页搜索。子问题间的依赖关系通过逻辑函数传递,增强推理连贯性。为避免不必要的外部搜索,引入知识边界判定机制,判断子问题是否在模型内在知识范围内,若在则直接回答。实验表明,仅需数百个训练样本,Thinker性能即达到现有基线水平;当使用完整训练集时,在多个数据集和模型规模下显著优于现有方法。代码已开源。
原文摘要 · Abstract (English)
Efficient retrieval of external knowledge bases and web pages is crucial for enhancing the reasoning abilities of LLMs. Previous works on training LLMs to leverage external retrievers for solving complex problems have predominantly employed end-to-end reinforcement learning. However, these approaches neglect supervision over the reasoning process, making it difficult to guarantee logical coherence and rigor. To address these limitations, we propose Thinker, a hierarchical thinking model for deep search through multi-turn interaction, making the reasoning process supervisable and verifiable. It decomposes complex problems into independently solvable sub-problems, each dually represented in both natural language and an equivalent logical function to support knowledge base and web searches. Concurrently, dependencies between sub-problems are passed as parameters via these logical functions, enhancing the logical coherence of the problem-solving process. To avoid unnecessary external searches, we perform knowledge boundary determination to check if a sub-problem is within the LLM's intrinsic knowledge, allowing it to answer directly. Experimental results indicate that with as few as several hundred training samples, the performance of Thinker is competitive with established baselines. Furthermore, when scaled to the full training set, Thinker significantly outperforms these methods across various datasets and model sizes. The source code is available at https://github.com/OpenSPG/KAG-Thinker.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。