摘要
arXiv:2501.10053v2 宣传类型:替换
摘要:利用大型语言模型(LLMs)的自主决策能力在推理任务中展现了卓越的性能。然而,尽管迭代或递归检索增强生成(RAG)技术取得了成功,但在面对复杂问题时,这些方法往往受限于单一的解决方案空间。在这篇论文中,我们提出了一种在RAG中集成系统分析与高效推理动作的新型思考模式,通过蒙特卡洛树搜索(MCTS)大幅激活内在的推理能力,并借助MCTS扩展特定任务的解决方案空间,我们称之为AirRAG。具体而言,我们的方法设计了五种基本的推理动作,并通过MCTS扩展到广泛的基于树的推理空间。该方法还整合了自一致性验证以探索潜在的推理路径和推理扩展定律。此外,采用了计算上最优化的策略来将更多的推理资源分配给关键动作,从而提高整体性能。实验结果表明了AirRAG的有效性,展示出在复杂问答数据集上显著的性能提升。此外,AirRAG 具有灵活性和轻量级的特点,使其易于与其他先进技术集成。