Saved in:
Bibliographic Details
Main Author: Pan, Xinghan
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2502.15485
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915169348616192
author Pan, Xinghan
author_facet Pan, Xinghan
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2502_15485
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Enhancing RWKV-based Language Models for Long-Sequence Text Generation
Pan, Xinghan
Computation and Language
Artificial Intelligence
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.
title Enhancing RWKV-based Language Models for Long-Sequence Text Generation
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2502.15485