arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2006.03555cs.LGcs.CLstat.ML
收藏

通过线性可扩展长上下文Transformers进行蛋白质掩蔽语言建模

Masked Language Modeling for Proteins via Linearly Scalable Long-Context Transformers

Google(谷歌);University of Cambridge(剑桥大学);Alan Turing Institute(艾伦·图灵研究所);DeepMind(深度思维)

Krzysztof Choromanski, Valerii Likhosherstov, David Dohan, Xingyou Song, Andreea Gane, Tamas Sarlos, Peter Hawkins, Jared Davis, David Belanger, Lucy Colwell, Adrian Weller

摘要:

Transformer模型在各种领域都取得了最先进的成果。然而,对训练注意力机制以学习远距离输入之间的复杂依赖关系的成本的关注继续增长。作为回应,利用学习到的注意力矩阵的结构和稀疏性的解决方案已经蓬勃发展。然而,涉及长序列的实际应用,例如生物序列分析,可能无法满足这些假设,从而排除了对这些模型的探索。为了解决这个问题,我们提出了一个新的Transformer架构,表演者,通过正交随机特征(FAVOR)的基础上快速注意。我们的机制在序列中的令牌数量上线性而不是二次地缩放,其特征在于次二次空间复杂度,并且不包含任何稀疏模式先验。此外,它提供了强有力的理论保证:无偏估计的注意矩阵和一致收敛。它还与预先训练的常规Transformers向后兼容。我们证明了它的有效性的蛋白质序列建模的挑战性任务,并提供详细的理论分析。

英文摘要:

Transformer models have achieved state-of-the-art results across a diverse range of domains. However, concern over the cost of training the attention mechanism to learn complex dependencies between distant inputs continues to grow. In response, solutions that exploit the structure and sparsity of the learned attention matrix have blossomed. However, real-world applications that involve long sequences, such as biological sequence analysis, may fall short of meeting these assumptions, precluding exploration of these models. To address this challenge, we present a new Transformer architecture, Performer, based on Fast Attention Via Orthogonal Random features (FAVOR). Our mechanism scales linearly rather than quadratically in the number of tokens in the sequence, is characterized by sub-quadratic space complexity and does not incorporate any sparsity pattern priors. Furthermore, it provides strong theoretical guarantees: unbiased estimation of the attention matrix and uniform convergence. It is also backwards-compatible with pre-trained regular Transformers. We demonstrate its effectiveness on the challenging task of protein sequence modeling and provide detailed theoretical analysis.

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

↑