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

超越思维链:语言模型中有效的思维图推理

Beyond Chain-of-Thought, Effective Graph-of-Thought Reasoning in Language Models

Shanghai Jiao Tong University(上海交通大学);Wuhan University(武汉大学)

Yao Yao, Zuchao Li, Hai Zhao

AI总结:

针对思维链(CoT)线性推理的局限性,本文提出了思维图(GoT)推理方法,通过图结构和门控融合机制建模非线性的思维过程,在AQUA-RAT和ScienceQA任务上显著提升了语言模型的推理性能。

摘要:

随着语言模型(LMs)在NLP任务中的广泛应用,研究人员发现了思维链(CoT)通过生成中间步骤来辅助LMs完成复杂推理任务的潜力。然而,人类的思维过程通常是非线性的,而非简单的顺序思维链。因此,我们提出了思维图(GoT)推理,它不仅将人类思维过程建模为链,还将其建模为图。通过将思维单元表示为节点、将其间的连接表示为边,我们的方法捕捉了人类思维的非顺序特性,并允许对思维过程进行更真实的建模。GoT采用了一个两阶段框架,包含一个额外的GoT编码器用于思维图表示,并通过门控融合机制将图表示与原始输入表示相融合。我们在纯文本推理任务(AQUA-RAT)和多模态推理任务(ScienceQA)上评估了GoT的性能。我们的模型在AQUA-RAT测试集上相较于强大的CoT基线取得了显著提升,并在ScienceQA测试集上使用T5-base模型将准确率从85.19%提升至87.59%,超越了最先进的Multimodal-CoT。

英文摘要:

With the widespread use of language models (LMs) in NLP tasks, researchers have discovered the potential of Chain-of-thought (CoT) to assist LMs in accomplishing complex reasoning tasks by generating intermediate steps. However, human thought processes are often non-linear, rather than simply sequential chains of thoughts. Therefore, we propose Graph-of-Thought (GoT) reasoning, which models human thought processes not only as a chain but also as a graph. By representing thought units as nodes and connections between them as edges, our approach captures the non-sequential nature of human thinking and allows for a more realistic modeling of thought processes. GoT adopts a two-stage framework with an additional GoT encoder for thought graph representation and fuses the graph representation with the original input representation through a gated fusion mechanism. We evaluate GoT's performance on a text-only reasoning task (AQUA-RAT) and a multimodal reasoning task (ScienceQA). Our model achieves significant improvement over the strong CoT baseline on the AQUA-RAT test set and boosts accuracy from 85.19% to 87.59% using the T5-base model over the state-of-the-art Multimodal-CoT on the ScienceQA test set.

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

↑