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

Efficient RWKV-based Representation Learning for 3D Point Clouds

About

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 directly applied to point clouds, RWKV, originally developed for sequential text, struggles to capture local geometric structures and model spatial dependencies effectively. To address this, we propose the \textbf{P-RWKV} block, which bridges the gap between sequence modeling and irregular 3D geometry while preserving the efficiency advantages of RWKV. It consists of a Local Perception Expansion (LPE) component to expand contextual perception along the spatio-temporal sequence and a Spatial Context Enhancement (SCE) component to strengthen spatial awareness. To validate the effectiveness of P-RWKV for point cloud understanding, we construct PointER, a single-modality self-supervised representation learning framework whose encoder is composed of stacked P-RWKV blocks. Furthermore, we extend P-RWKV to a cross-modality setting and integrate the proposed core sub-modules into multiple architectures, demonstrating strong plug-and-play flexibility and architectural generality. Extensive experiments show that the P-RWKV block and its key sub-modules achieve competitive performance across various tasks with lower computational cost and inference latency. Code will be released upon acceptance.

Yun Liu, Xuefeng Yan, Liangliang Nan, Xianzhi Li, Peng Li, Zhe Zhu, Honghua Chen, Mingqiang Wei• 2026

Related benchmarks

TaskDatasetResultRank
Shape classificationModelNet40 (test)--
255
Shape classificationModelNet40
Accuracy93.73
113
Semantic segmentationS3DIS
mIoU75.11
100
Shape classificationScanObjectNN PB_T50_RS--
72
Object Part SegmentationShapeNet Parts 50 (test)
Instance mIoU86.5
32
Shape classificationScanObjectNN (SONN) PB_T50_RS (test)
OA84.93
25
Shape classificationScanObjectNN OBJ_ONLY (test)
Overall Accuracy88.49
21
Shape classificationScanObjectNN OBJ-ONLY
Accuracy89.98
16
Shape classificationScanObjectNN OBJ-BG v1.0 (test)
Accuracy90.38
11
Shape classificationScanObjectNN OBJ-BG
Accuracy90.61
9
Showing 10 of 12 rows

Other info

Follow for update