arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.06221cs.AIcs.CLcs.IR
收藏

TRAD:通过逐步思维检索与对齐决策增强大语言模型智能体

TRAD: Enhancing LLM Agents with Step-Wise Thought Retrieval and Aligned Decision

Shanghai Jiao Tong University(上海交通大学);China Pacific Insurance(中国太平洋保险)

Ruiwen Zhou, Yingxuan Yang, Muning Wen, Ying Wen, Wenhao Wang, Chunling Xi, Guoqiang Xu, Yong Yu, Weinan Zhang

AI总结:

针对LLM智能体上下文示例选择利用不佳、检索噪声大的问题,提出TRAD框架,通过步骤级思维检索和对齐决策优化示例选择,在基准测试和真实保险场景中均取得性能提升。

摘要:

得益于大语言模型(LLM)的广博知识与文本理解能力,目前已构建出大量LLM智能体,用于网页导航、在线购物等各类任务。在这类研究中,多数方法借助上下文示例实现无需微调的泛化,但很少有研究关注如何选择并有效利用这些示例的问题。近期,有研究提出基于任务元数据的轨迹级检索方法,将轨迹作为上下文示例,以提升智能体在部分序列决策任务中的整体性能。然而,这类方法存在缺陷:检索到的示例看似合理,却未结合任务特定的状态转移动态,且输入过长,包含大量无关上下文。本文提出一种新型框架TRAD以解决上述问题。TRAD首先执行思维检索(Thought Retrieval),通过思维匹配实现步骤级的示例选择,从而获得更具参考价值的示例,减少无关输入噪声。随后,TRAD引入对齐决策(Aligned Decision),为检索到的示例步骤补充其前置或后续步骤,既能够容忍不完美的思维,也提供了在更多上下文与更少噪声之间取得平衡的选择。在ALFWorld和Mind2Web基准上的大量实验表明,TRAD不仅性能优于当前最优模型,还能有效降低噪声、提升泛化能力。此外,TRAD已部署于一家全球商业保险公司的真实场景中,提升了机器人流程自动化的成功率。

英文摘要:

Numerous large language model (LLM) agents have been built for different tasks like web navigation and online shopping due to LLM's wide knowledge and text-understanding ability. Among these works, many of them utilize in-context examples to achieve generalization without the need for fine-tuning, while few of them have considered the problem of how to select and effectively utilize these examples. Recently, methods based on trajectory-level retrieval with task meta-data and using trajectories as in-context examples have been proposed to improve the agent's overall performance in some sequential decision making tasks. However, these methods can be problematic due to plausible examples retrieved without task-specific state transition dynamics and long input with plenty of irrelevant context. In this paper, we propose a novel framework (TRAD) to address these issues. TRAD first conducts Thought Retrieval, achieving step-level demonstration selection via thought matching, leading to more helpful demonstrations and less irrelevant input noise. Then, TRAD introduces Aligned Decision, complementing retrieved demonstration steps with their previous or subsequent steps, which enables tolerance for imperfect thought and provides a choice for balance between more context and less noise. Extensive experiments on ALFWorld and Mind2Web benchmarks show that TRAD not only outperforms state-of-the-art models but also effectively helps in reducing noise and promoting generalization. Furthermore, TRAD has been deployed in real-world scenarios of a global business insurance company and improves the success rate of robotic process automation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑