轻量大模型也能精准规划复杂任务,靠的是动态参数与分层推理。
LightPlanner: Unleashing the Reasoning Capabilities of Lightweight Large Language Models in Task Planning
- 用可变参数函数调用替代固定技能模板,实现精细动作控制。
- 在空间语义推理任务中成功率达91.6%,比ReAct高14.9%。
- 适合资源受限的边缘设备部署,适合做机器人智能规划。
近年来,轻量级大语言模型因其低计算开销和适合边缘部署,在机器人领域备受关注。然而,在涉及动态语义逻辑推理的复杂任务规划中,其表现仍不理想。为此,我们提出LightPlanner,通过充分挖掘轻量级LLM的推理能力,提升其在复杂任务规划中的性能。不同于传统使用固定技能模板的规划器,LightPlanner通过参数化函数调用控制机器人动作,并动态生成参数值,实现细粒度技能控制,显著提高复杂场景下的任务成功率。此外,引入分层深度推理机制,在每一步动作决策前,分别验证动作执行(反馈一致性)、语义解析(目标一致性)和参数生成(参数有效性),确保后续控制正确性。同时,设计记忆模块存储历史动作,降低上下文长度,提升长期任务规划效率。我们在包含40,000条动作指令的LightPlan-40k数据集上训练LightPlanner-1.5B模型,实验表明,尽管参数量最小,其任务成功率最高。在涉及空间语义推理的任务中,成功率超过ReAct 14.9%。此外,实证证明LightPlanner可在边缘设备上运行。
原文摘要 · Abstract (English)
In recent years, lightweight large language models (LLMs) have garnered significant attention in the robotics field due to their low computational resource requirements and suitability for edge deployment. However, in task planning -- particularly for complex tasks that involve dynamic semantic logic reasoning -- lightweight LLMs have underperformed. To address this limitation, we propose a novel task planner, LightPlanner, which enhances the performance of lightweight LLMs in complex task planning by fully leveraging their reasoning capabilities. Unlike conventional planners that use fixed skill templates, LightPlanner controls robot actions via parameterized function calls, dynamically generating parameter values. This approach allows for fine-grained skill control and improves task planning success rates in complex scenarios. Furthermore, we introduce hierarchical deep reasoning. Before generating each action decision step, LightPlanner thoroughly considers three levels: action execution (feedback verification), semantic parsing (goal consistency verification), and parameter generation (parameter validity verification). This ensures the correctness of subsequent action controls. Additionally, we incorporate a memory module to store historical actions, thereby reducing context length and enhancing planning efficiency for long-term tasks. We train the LightPlanner-1.5B model on our LightPlan-40k dataset, which comprises 40,000 action controls across tasks with 2 to 13 action steps. Experiments demonstrate that our model achieves the highest task success rate despite having the smallest number of parameters. In tasks involving spatial semantic reasoning, the success rate exceeds that of ReAct by 14.9 percent. Moreover, we demonstrate LightPlanner's potential to operate on edge devices.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。