中文

RWKV 模型综述

计算与语言 2025-01-07 v2 人工智能

摘要

Receptance Weighted Key Value (RWKV) 模型提供了一种变革性的 Transformer 架构替代方案,融合了循环和注意力系统的优势。不同于传统 Transformer 依赖强烈自注意力机制,RWKV 能以最小的计算需求有效捕获长程依赖。通过利用循环框架,RWKV 解决了 Transformer 在长序列任务中存在的一些计算效率问题。RWKV 近年来因其在多个领域的稳健性能而引起广泛关注。尽管如此,关于 RWKV 模型的系统性综述尚不存在。本文旨在填补这一空白,作为 RWKV 架构、其核心原理及其 varied applications(如自然语言生成、自然语言理解和计算机视觉)的首个综合性综述。我们评估了 RWKV 与传统 Transformer 模型的比较,突出其高效处理长序列和降低计算成本的能力。此外,我们探讨了 RWKV 面临的挑战,并提出了未来研究和发展的潜在方向。我们持续维护相关开源材料于:https://github.com/MLGroupJLU/RWKV-Survey。

关键词

引用

@article{arxiv.2412.14847,
  title  = {A Survey of RWKV},
  author = {Zhiyuan Li and Tingyu Xia and Yi Chang and Yuan Wu},
  journal= {arXiv preprint arXiv:2412.14847},
  year   = {2025}
}

备注

19 pages