Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

AudioRWKV: Efficient and Stable Bidirectional RWKV for Audio Pattern Recognition

About

Recently, Transformers (e.g., Audio Spectrogram Transformers, AST) and state-space models (e.g., Audio Mamba, AuM) have achieved remarkable progress in audio modeling. However, the O(L^2) computational complexity of the Transformer architecture hinders efficient long-sequence processing, while the Mamba architecture tends to become unstable when scaling parameters and data. To address these challenges, this paper proposes AudioRWKV (A-RWKV), a highly efficient and stable architecture for audio modeling. Specifically, we inherit the stable and efficient recurrent formulation of RWKV7 and replace its 1D token-shift operation with a 2D depthwise separable convolution to better capture local spectro-temporal patterns. Furthermore, we adapt the original causal WKV kernel into a bidirectional WKV kernel (Bi-WKV), enabling global context modeling over the entire audio sequence while maintaining linear computational complexity. Benefiting from the inherent stability of the RWKV7 foundation, A-RWKV scales seamlessly to larger model sizes. Experimental results demonstrate that, under the same linear-model regime, A-RWKV-S (22M) achieves performance parity with AuM-B (92M) while exhibiting more stable throughput than AST; for long-form audio (~5 minutes 28 seconds), WKV7 achieves up to a 13.3X speedup in processing.

Jing Wang, Maoxiang Wu, Jiayu Xiong, Jianlong Kwan, Jun Xue• 2025

Related benchmarks

TaskDatasetResultRank
Audio ClassificationESC-50
Accuracy80.4
461
Audio ClassificationAudioSet 20K
mAP17.25
151
Audio ClassificationAudioSet 2M
mAP40.91
102
Audio ClassificationVGG-Sound--
83
Audio ClassificationNSynth Pitch
Accuracy91.35
8
Audio ClassificationSpeech Commands V2
Accuracy93.01
4
Audio ClassificationVGGSound (fine-tuning)
Accuracy48.91
3
Keyword SpottingSpeech Commands v2 (fine-tuning)
Accuracy96.83
3
Audio ClassificationNP (fine-tuning)
Accuracy93.44
2
Environmental Sound ClassificationESC-50 (fine-tuning)
Accuracy86.8
2
Showing 10 of 10 rows

Other info

Follow for update