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

MATTER:使用异构知识源的内存增强Transformer

MATTER: Memory-Augmented Transformer Using Heterogeneous Knowledge Sources

Amazon AGI(亚马逊AGI);Department of Computer Science and Engineering, HKUST(香港科技大学计算机科学与工程学系)

Dongkyu Lee, Chandana Satya Prakash, Jack FitzGerald, Jens Lehmann

AI总结:

本文提出MATTER,一种内存增强Transformer,通过固定长度神经记忆从段落和问答对等异构知识源中高效检索,在问答基准上兼顾准确性与速度,并以100倍推理吞吐量取得竞争性结果。

摘要:

利用外部知识对于在知识密集型任务(如问答)中取得高性能至关重要。检索-读取方法被广泛用于将外部知识整合到语言模型中。然而,这种方法由于上下文长度随检索到的知识数量成比例增长,导致计算成本和延迟增加。此外,现有的检索增强模型通常从单一类型的知识源中检索信息,限制了它们对具有不同结构的多样化知识源的可扩展性。在这项工作中,我们提出了一种高效的内存增强Transformer,称为MATTER,旨在从多个异构知识源中检索相关知识。具体来说,我们的模型以固定长度的神经记忆形式,从非结构化来源(段落)和半结构化来源(问答对)中检索和读取信息。我们证明,我们的模型在流行的问答基准上,在准确性和速度方面均优于现有的高效检索增强模型。此外,MATTER在推理过程中具有100倍的吞吐量,同时取得了与传统的读取-检索模型相比具有竞争力的结果。

英文摘要:

Leveraging external knowledge is crucial for achieving high performance in knowledge-intensive tasks, such as question answering. The retrieve-and-read approach is widely adopted for integrating external knowledge into a language model. However, this approach suffers from increased computational cost and latency due to the long context length, which grows proportionally with the number of retrieved knowledge. Furthermore, existing retrieval-augmented models typically retrieve information from a single type of knowledge source, limiting their scalability to diverse knowledge sources with varying structures. In this work, we introduce an efficient memory-augmented transformer called MATTER, designed to retrieve relevant knowledge from multiple heterogeneous knowledge sources. Specifically, our model retrieves and reads from both unstructured sources (paragraphs) and semi-structured sources (QA pairs) in the form of fixed-length neural memories. We demonstrate that our model outperforms existing efficient retrieval-augmented models on popular QA benchmarks in terms of both accuracy and speed. Furthermore, MATTER achieves competitive results compared to conventional read-and-retrieve models while having 100x throughput during inference.

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

↑