arXiv:2412.14847cs.CLcs.AI2024-12综述被引 8

RWKV模型融合循环与注意力机制,高效处理长序列任务。

A Survey of RWKV

  • 用循环结构替代自注意力,降低长序列计算开销
  • 在多领域任务中表现优于传统Transformer,计算成本更低
  • 适合需要高效处理长文本或时序数据的研究者参考

Receptance Weighted Key Value(RWKV)模型为Transformer架构提供了一种新范式,结合了循环系统与注意力机制的优势。与依赖自注意力的常规Transformer不同,RWKV通过递归框架有效捕捉长距离依赖,计算开销极低。该模型在多个任务中表现出色,尤其适用于长序列处理。尽管近年来广受关注,但尚无系统性综述。本文首次全面回顾了RWKV的架构原理、核心机制及其在自然语言生成、自然语言理解、计算机视觉等领域的应用。对比传统Transformer,RWKV展现出更高效的长序列建模能力与更低的计算成本。同时,论文分析了当前挑战,并提出未来研究方向。相关开源资源持续更新:https://github.com/MLGroupJLU/RWKV-Survey。

原文摘要 · Abstract (English)

The Receptance Weighted Key Value (RWKV) model offers a novel alternative to the Transformer architecture, merging the benefits of recurrent and attention-based systems. Unlike conventional Transformers, which depend heavily on self-attention, RWKV adeptly captures long-range dependencies with minimal computational demands. By utilizing a recurrent framework, RWKV addresses some computational inefficiencies found in Transformers, particularly in tasks with long sequences. RWKV has recently drawn considerable attention for its robust performance across multiple domains. Despite its growing popularity, no systematic review of the RWKV model exists. This paper seeks to fill this gap as the first comprehensive review of the RWKV architecture, its core principles, and its varied applications, such as natural language generation, natural language understanding, and computer vision. We assess how RWKV compares to traditional Transformer models, highlighting its capability to manage long sequences efficiently and lower computational costs. Furthermore, we explore the challenges RWKV encounters and propose potential directions for future research and advancement. We consistently maintain the related open-source materials at: https://github.com/MLGroupJLU/RWKV-Survey.

RWKV序列建模Transformer替代高效计算

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