arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.05313cs.CLcs.AI

RAT:检索增强思维激发长程生成中的上下文感知推理能力

RAT: Retrieval Augmented Thoughts Elicit Context-Aware Reasoning in Long-Horizon Generation

Peking University(北京大学);University of California, Los Angeles(加利福尼亚大学洛杉矶分校);Beijing Institute for General Artificial Intelligence(北京通用人工智能研究院)

Zihao Wang, Anji Liu, Haowei Lin, Jiaqi Li, Xiaojian Ma, Yitao Liang

AI总结:

本研究提出检索增强思维(RAT)方法,通过检索信息迭代修正零样本CoT的各思维步骤,有效提升大语言模型在多类长程生成任务中的表现并缓解幻觉。

摘要:

我们探究了借助信息检索迭代修正思维链,如何显著提升大语言模型在长程生成任务中的推理与生成能力,同时大幅缓解幻觉问题。具体而言,所提出的方法——检索增强思维(retrieval-augmented thoughts,RAT)——在生成初始零样本思维链(CoT)后,利用与任务查询、当前及过往思维步骤相关的检索信息,逐一修正每个思维步骤。将RAT应用于GPT-3.5、GPT-4和CodeLLaMA-7b,可大幅提升它们在各类长程生成任务上的表现;平均而言,代码生成的评分相对提升13.63%,数学推理提升16.96%,创意写作提升19.2%,具身任务规划提升42.78%。演示页面可访问:https://craftjarvis.github.io/RAT

英文摘要:

We explore how iterative revising a chain of thoughts with the help of information retrieval significantly improves large language models' reasoning and generation ability in long-horizon generation tasks, while hugely mitigating hallucination. In particular, the proposed method -- *retrieval-augmented thoughts* (RAT) -- revises each thought step one by one with retrieved information relevant to the task query, the current and the past thought steps, after the initial zero-shot CoT is generated. Applying RAT to GPT-3.5, GPT-4, and CodeLLaMA-7b substantially improves their performances on various long-horizon generation tasks; on average of relatively increasing rating scores by 13.63% on code generation, 16.96% on mathematical reasoning, 19.2% on creative writing, and 42.78% on embodied task planning. The demo page can be found at https://craftjarvis.github.io/RAT

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

↑