让大模型学会减少错误预测,降低人工干预频率。
CoFineLLM: Conformal Finetuning of LLMs for Language-Instructed Robot Planning
- 用置信度感知的微调方法,让大模型生成更小的预测集合。
- 在多个任务中显著缩小预测集大小,帮助率降低30%以上。
- 适合需要高可靠性、少人工介入的机器人自主规划场景。
大语言模型(LLM)最近被用作语言指令机器人的规划器,能生成动作序列完成自然语言任务。但其可靠性受限,尤其在长时程任务中常产生过度自信却错误的输出。置信性预测(CP)通过将模型输出封装为包含正确动作的预测集,以用户设定的置信度保障计划正确性。当预测集为单元素时,系统执行该动作;否则请求人类帮助。然而,由于LLM训练时未考虑预测集,常生成过大集合,尤其在高置信度下,导致频繁求助,限制自主部署。为此,我们提出CoFineLLM(置信性微调框架),首个面向置信性预测的LLM规划器微调方法,显式缩小预测集规模,从而减少人工干预。我们在多个语言指令机器人规划任务上评估该方法,结果表明其在预测集大小和求助率方面均优于不确定性感知与忽略的基线。最后,硬件实验验证了方法对分布外场景的鲁棒性。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have recently emerged as planners for language-instructed agents, generating sequences of actions to accomplish natural language tasks. However, their reliability remains a challenge, especially in long-horizon tasks, since they often produce overconfident yet wrong outputs. Conformal Prediction (CP) has been leveraged to address this issue by wrapping LLM outputs into prediction sets that contain the correct action with a user-defined confidence. When the prediction set is a singleton, the planner executes that action; otherwise, it requests help from a user. This has led to LLM-based planners that can ensure plan correctness with a user-defined probability. However, as LLMs are trained in an uncertainty-agnostic manner, without awareness of prediction sets, they tend to produce unnecessarily large sets, particularly at higher confidence levels, resulting in frequent human interventions limiting autonomous deployment. To address this, we introduce CoFineLLM (Conformal Finetuning for LLMs), the first CP-aware finetuning framework for LLM-based planners that explicitly reduces prediction-set size and, in turn, the need for user interventions. We evaluate our approach on multiple language-instructed robot planning problems and show consistent improvements over uncertainty-aware and uncertainty-agnostic finetuning baselines in terms of prediction-set size, and help rates. Finally, we demonstrate robustness of our method to out-of-distribution scenarios in hardware experiments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。