paper-with-me

홈 › Papers

Enhancing RWKV-based Language Models for Long-Sequence Text Generation

2025-02-21 · Xinghan Pan

This paper introduces an enhanced RWKV architecture with adaptive temporal gating mechanisms for improved long-context language modeling. We propose two principal innovations: (1) a position-aware convolutional shift operator that captures local syntactic patterns while preserving global coherence, and (2) a neurally-gated information routing mechanism that dynamically regulates inter-token information flow. Through comprehensive experiments on text generation tasks, our enhanced model demonstrates superior performance compared to the baseline RWKV, achieving 96.5 relative improvement in ROUGE-L scores with only 2.95 increased inference latency. Ablation studies validate the individual contributions of each component, while linguistic analysis reveals the model's adaptive attention to syntactic boundaries and entity coherence. The proposed modifications maintain RWKV's linear computational complexity while significantly enhancing its contextual modeling capabilities, establishing new state-of-the-art performance for recurrent-style architectures in long-form text generation.

📄 PDF Abstract BibTeX arXiv:2502.15485

Code (1)

PStarH/long-seq-rwkv 공식 구현

Tasks

Language ModelingLanguage ModellingText Generation

Methods 이 논문이 사용한 방법론

Softmax The Softmax output function transforms a previous layer's output into a vector of probabilities. It is commonly used for multiclass classification. Given an input vector $x$…
Attention 설명 없음

Similar Papers 제목 키워드 기반

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

VisualRWKV-HD and UHD: Advancing High-Resolution Processing for Visual Language Models

2024-10-15 · Zihang Li, Haowen Hou

Accurately understanding complex visual information is crucial for visual language models (VLMs). Enhancing image resolution can improve visual perception capabilities, not only reducing hallucinations but also boosting …

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 hea…

Natural Language UnderstandingSurveyText Generation

StyleRWKV: High-Quality and High-Efficiency Style Transfer with RWKV-like Architecture

2024-12-27 · Miaomiao Dai, Qianyu Zhou, Lizhuang Ma

Style transfer aims to generate a new image preserving the content but with the artistic representation of the style source. Most of the existing methods are based on Transformers or diffusion models, however, they suffe…

Style Transfer

Efficient RWKV-based Representation Learning for 3D Point Clouds

2026-06-09 · Yun Liu, Xuefeng Yan, Liangliang Nan, Xianzhi Li 외 arxiv

The recent receptance weighted key value (RWKV) model combines RNN-style recurrence, offering a linear-complexity alternative to Transformers' quadratic self-attention for modeling global dependencies. However, when dire…

Representation LearningPoint Clouds