arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2210.01293cs.CL
收藏

ThinkSum:使用大型语言模型在集合上进行概率推理

ThinkSum: Probabilistic reasoning over sets using large language models

Stanford University(斯坦福大学);Mila, Université de Montréal(蒙特利尔大学米拉研究所);Ohio State University(俄亥俄州立大学);Microsoft Research(微软研究院)

Batu Ozturkler, Nikolay Malkin, Zhen Wang, Nebojsa Jojic

AI总结:

本文提出两阶段概率推断范式 ThinkSum,通过并行检索关联并在 LLM 外聚合推理,显著提升 GPT 模型在 BIG-bench 十三项困难任务上的表现。

摘要:

大型语言模型(LLM)具备强大的高级类比推理能力:能够复现线性文本中出现在其训练数据内(零样本评估)或所提供上下文中(少样本上下文学习)的模式。然而,近期研究表明,即使是更先进的 LLM,也会在需要对多个对象或事实进行推理并作出一连串逻辑推断的场景中失败。我们提出一种两阶段概率推断范式 ThinkSum,它以结构化方式对对象或事实的集合进行推理。在第一阶段(Think——关联检索),针对从提示词或辅助模型调用中提取的一组短语,并行查询 LLM。在第二阶段(Sum——概率推断或推理),聚合这些查询的结果以作出最终预测。我们在 LLM 评估任务套件 BIG-bench 上展示了 ThinkSum 的可能性与优势:在十三项困难任务上,使用 GPT 系列模型取得了相对于当前最优水平的提升,而且通常采用的是小得多的模型变体。我们还将 ThinkSum 与针对 LLM 直接提示所提出的其他修改方案进行比较和对照,例如思维链提示的各种变体。我们的结果表明,由于 ThinkSum 中的概率推断是在 LLM 调用之外执行的,因此 ThinkSum 对提示设计不那么敏感,可产生更具可解释性的预测,并且能够灵活地与潜变量模型结合,以从 LLM 中提取结构化知识。总体而言,我们提出的范式代表了一种增强 LLM 推理能力的有前景方法。

英文摘要:

Large language models (LLMs) have a substantial capacity for high-level analogical reasoning: reproducing patterns in linear text that occur in their training data (zero-shot evaluation) or in the provided context (few-shot in-context learning). However, recent studies show that even the more advanced LLMs fail in scenarios that require reasoning over multiple objects or facts and making sequences of logical deductions. We propose a two-stage probabilistic inference paradigm, ThinkSum, which reasons over sets of objects or facts in a structured manner. In the first stage (Think - retrieval of associations), a LLM is queried in parallel over a set of phrases extracted from the prompt or an auxiliary model call. In the second stage (Sum - probabilistic inference or reasoning), the results of these queries are aggregated to make the final prediction. We demonstrate the possibilities and advantages of ThinkSum on the BIG-bench suite of LLM evaluation tasks, achieving improvements over the state of the art using GPT-family models on thirteen difficult tasks, often with far smaller model variants. We also compare and contrast ThinkSum with other proposed modifications to direct prompting of LLMs, such as variants of chain-of-thought prompting. Our results suggest that because the probabilistic inference in ThinkSum is performed outside of calls to the LLM, ThinkSum is less sensitive to prompt design, yields more interpretable predictions, and can be flexibly combined with latent variable models to extract structured knowledge from LLMs. Overall, our proposed paradigm represents a promising approach for enhancing the reasoning capabilities of LLMs.

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

↑