arXiv:2412.06106cs.CLcs.LG2024-12被引 4

提出高效上下文传播的Perceiver架构,提升长文本建模速度与效果

Efficient Context Propagating Perceiver Architectures for Auto-Regressive Language Modeling

  • 融合上下文与潜在序列进行自回归训练,增强信息保留
  • 保持与LongLoRA相当的计算复杂度,实现半线性效率
  • 通过分段注意力机制提升语言建模性能,适合长序列任务

Transformer架构中注意力机制的二次复杂度限制了长序列的高效处理。尽管已有研究尝试将注意力复杂度从O(n²)降低至准线性,但如何在降低复杂度的同时保持高性能仍是难题。本文以PerceiverAR为基础,探索上下文保留与注意力复杂度之间的权衡,提出四种新架构,其中最优者命名为高效上下文传播Perceiver(ECP)。ECP克服了PerceiverAR仅依赖潜在序列的缺陷,同时利用上下文与潜在序列进行自回归训练;其注意力复杂度与LongLoRA相当,计算高效。更重要的是,通过成对分段注意力机制,ECP能更优地提取信息,显著提升语言建模表现。实验表明,ECP在Wikitext-103、PG-19和sCIFAR-10上均优于当前主流Transformer模型。

原文摘要 · Abstract (English)

One of the key challenges in Transformer architectures is the quadratic complexity of the attention mechanism, which limits the efficient processing of long sequences. Many recent research works have attempted to provide a reduction from the $O(n^2)$ time complexity of attention to semi-linear complexity. However, it remains an unsolved problem in the sense of maintaining high performance when complexity is reduced. One of the important works in this respect is the Perceiver class of architectures that have demonstrated excellent performance, while reducing the computation complexity. In this paper, we use the PerceiverAR as a basis and explore the design space of different trade-offs between preserving context and reducing attention complexity. To this end, we develop four new architectural paradigms, the best performing of which we denote as the Efficient Context propagating Perceiver (ECP). ECP has two major advantages over the PerceiverAR. First, the ECP architecture overcomes the main drawback of PercieverAR by utilizing both the context and the latent sequences in autoregressive training. Second, the ECP architecture operates with the same attention complexity as LongLoRA, making it computationally efficient. More importantly, via pairwise segment attention, it extracts better information resulting in improved language modeling. Empirically, we demonstrate that the ECP architecture significantly outperforms other state-of-the-art Transformer models on Wikitext-103, PG-19 and sCIFAR-10.

Perceiver自回归长序列注意力

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。