arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.04006cs.LGcs.AIcs.CLcs.CVcs.IR
收藏

长距离竞技场:高效Transformers的基准

Long Range Arena: A Benchmark for Efficient Transformers

Google Research(谷歌研究院);Google DeepMind(谷歌DeepMind)

Yi Tay, Mostafa Dehghani, Samira Abnar, Yikang Shen, Dara Bahri, Philip Pham, Jinfeng Rao, Liu Yang, Sebastian Ruder, Donald Metzler

摘要:

Transformers不能很好地扩展到长序列长度,这主要是因为二次自注意复杂度。在最近几个月里,已经提出了一系列高效、快速的Transformers来解决这个问题,而不是声称比普通Transformer模型更好或可比较的模型质量。到目前为止,对于如何评估这类模型还没有达成共识。此外,在广泛的任务和数据集上不一致的基准测试使得难以评估许多模型之间的相对模型质量。本文提出了一个系统的和统一的基准,LRA,专门专注于评估模型质量下的长上下文场景。我们的基准测试是一套任务,包括从$1K$到$16K$令牌的序列,涵盖了广泛的数据类型和形式,如文本,自然,合成图像和数学表达式,需要相似性,结构和视觉空间推理。我们系统地评估了10个成熟的长距离Transformer模型(Reformers,Linformers,Linear Transformers,Sinkhorn Transformers,Performers,Synthesizers,Sparse Transformers和Longformers)在我们新提出的基准套件。LRA为更好地理解这类高效的Transformer模型铺平了道路,促进了这一方向的更多研究,并提出了新的挑战性任务。我们的基准代码将在https://github.com/google-research/long-range-arena上发布。

英文摘要:

Transformers do not scale very well to long sequence lengths largely because of quadratic self-attention complexity. In the recent months, a wide spectrum of efficient, fast Transformers have been proposed to tackle this problem, more often than not claiming superior or comparable model quality to vanilla Transformer models. To this date, there is no well-established consensus on how to evaluate this class of models. Moreover, inconsistent benchmarking on a wide spectrum of tasks and datasets makes it difficult to assess relative model quality amongst many models. This paper proposes a systematic and unified benchmark, LRA, specifically focused on evaluating model quality under long-context scenarios. Our benchmark is a suite of tasks consisting of sequences ranging from $1K$ to $16K$ tokens, encompassing a wide range of data types and modalities such as text, natural, synthetic images, and mathematical expressions requiring similarity, structural, and visual-spatial reasoning. We systematically evaluate ten well-established long-range Transformer models (Reformers, Linformers, Linear Transformers, Sinkhorn Transformers, Performers, Synthesizers, Sparse Transformers, and Longformers) on our newly proposed benchmark suite. LRA paves the way towards better understanding this class of efficient Transformer models, facilitates more research in this direction, and presents new challenging tasks to tackle. Our benchmark code will be released at https://github.com/google-research/long-range-arena.

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

↑