让大模型在不确定环境中像人一样规划,提升决策可靠性。
PlanU: Large Language Model Reasoning through Planning under Uncertainty
- 用蒙特卡洛树搜索建模每步收益的分位数分布,捕捉不确定性
- 引入好奇度增强的置信上界分数,平衡探索与利用
- 适合需要多步交互、环境随机性强的智能体决策任务
大语言模型在推理任务中表现日益突出,但在不确定性环境下的规划能力仍逊于人类,例如面对随机状态转移的场景。现有方法多关注模型自身采样带来的不确定性,通过多条推理链或搜索树缓解,却忽略了环境不确定性,导致在随机性环境中性能不佳。部分近期方法尝试预测未知变量概率,但不适用于需与环境持续交互的多步推理任务。为此,本文提出PlanU,一种基于大模型的规划方法,将蒙特卡洛树搜索中的节点回报建模为分位数分布,用一组分位数表示回报分布。为平衡探索与利用,引入具有好奇度的置信上界(UCC)评分,评估节点不确定性。大量实验表明,PlanU在大模型于不确定性环境下的推理任务中显著有效。
原文摘要 · Abstract (English)
Large Language Models (LLMs) are increasingly being explored across a range of reasoning tasks. However, LLMs sometimes struggle with reasoning tasks under uncertainty that are relatively easy for humans, such as planning actions in stochastic environments. The adoption of LLMs for reasoning is impeded by uncertainty challenges, such as LLM uncertainty and environmental uncertainty. LLM uncertainty arises from the stochastic sampling process inherent to LLMs. Most LLM-based Decision-Making (LDM) approaches address LLM uncertainty through multiple reasoning chains or search trees. However, these approaches overlook environmental uncertainty, which leads to poor performance in environments with stochastic state transitions. Some recent LDM approaches deal with uncertainty by forecasting the probability of unknown variables. However, they are not designed for multi-step reasoning tasks that require interaction with the environment. To address uncertainty in LLM decision-making, we introduce PlanU, an LLM-based planning method that captures uncertainty within Monte Carlo Tree Search (MCTS). PlanU models the return of each node in the MCTS as a quantile distribution, which uses a set of quantiles to represent the return distribution. To balance exploration and exploitation during tree search, PlanU introduces an Upper Confidence Bounds with Curiosity (UCC) score which estimates the uncertainty of MCTS nodes. Through extensive experiments, we demonstrate the effectiveness of PlanU in LLM-based reasoning tasks under uncertainty.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。