paper-with-me

홈 › Papers

A Survey of RWKV

2024-12-19 · Zhiyuan Li, Tingyu Xia, Yi Chang, Yuan Wu

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.

📄 PDF Abstract BibTeX arXiv:2412.14847

Code (1)

mlgroupjlu/rwkv-survey 공식 구현 tf

Tasks

Natural Language UnderstandingSurveyText Generation

Methods 이 논문이 사용한 방법론

Attention 설명 없음
Linear Layer A Linear Layer is a projection $\mathbf{XW + b}$.
Dropout Dropout is a regularization technique for neural networks that drops a unit (along with connections) at training time with a specified probability $p$ (a common value is…
Multi-Head Attention 설명 없음
Adam 설명 없음
Layer Normalization Unlike batch normalization, Layer Normalization directly estimates the normalization statistics from the summed inputs…
Position-Wise Feed-Forward Layer 설명 없음
Label Smoothing Label Smoothing is a regularization technique that introduces noise for the labels. This accounts for the fact that datasets may have mistakes in them, so maximizing the…

Similar Papers 제목 키워드 기반

Eagle and Finch: RWKV with Matrix-Valued States and Dynamic Recurrence

2024-04-08 · Bo Peng, Daniel Goldstein, Quentin Anthony, Alon Albalak 외

We present Eagle (RWKV-5) and Finch (RWKV-6), sequence models improving upon the RWKV (RWKV-4) architecture. Our architectural design advancements include multi-headed matrix-valued states and a dynamic recurrence mechan…

Med-URWKV: Pure RWKV With ImageNet Pre-training For Medical Image Segmentation

2025-06-12 · Zhenhuan Zhou

Medical image segmentation is a fundamental and key technology in computer-aided diagnosis and treatment. Previous methods can be broadly classified into three categories: convolutional neural network (CNN) based, Transf…

Image SegmentationLong-range modelingMedical Image SegmentationSegmentation+1

RWKV-X: A Linear Complexity Hybrid Language Model

2025-04-30 · Haowen Hou, Zhiyi Huang, Kaifeng Tan, Rongchang Lu 외

In this paper, we introduce RWKV-X, a novel hybrid architecture that combines the efficiency of RWKV for short-range modeling with a sparse attention mechanism designed to capture long-range context. Unlike previous hybr…

Language ModelingLanguage Modellingmodel

RWKV-7 "Goose" with Expressive Dynamic State Evolution

2025-03-18 · Bo Peng, Ruichong Zhang, Daniel Goldstein, Eric Alcaide 외

We present RWKV-7 "Goose", a new sequence modeling architecture with constant memory usage and constant inference time per token. Despite being trained on dramatically fewer tokens than other top models, our 2.9 billion …

In-Context LearningLanguage ModelingLanguage Modelling

RWKV-PCSSC: Exploring RWKV Model for Point Cloud Semantic Scene Completion

2025-11-13 · Wenzhe He, Xiaojun Chen, Wentang Chen, Hongyu Wang 외 arxiv

Semantic Scene Completion (SSC) aims to generate a complete semantic scene from an incomplete input. Existing approaches often employ dense network architectures with a high parameter count, leading to increased model co…