Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Parameter-Efficient Fine-Tuning in Spectral Domain for Point Cloud Learning

About

Recently, leveraging pre-training techniques to enhance point cloud models has become a prominent research topic. However, existing approaches typically require full fine-tuning of pre-trained models to achieve satisfactory performance on downstream tasks, which is storage-intensive and computationally demanding. To address this issue, we propose a novel Parameter-Efficient Fine-Tuning (PEFT) method for point cloud, called \textbf{PointGST} (\textbf{Point} cloud \textbf{G}raph \textbf{S}pectral \textbf{T}uning). PointGST freezes the pre-trained model and introduces a lightweight, trainable Point Cloud Spectral Adapter (PCSA) for fine-tuning parameters in the spectral domain. The core idea is built on two observations: 1) The inner tokens from frozen models might present confusion in the spatial domain; 2) Task-specific intrinsic information is important for transferring the general knowledge to the downstream task. Specifically, PointGST transfers the point tokens from the spatial domain to the spectral domain, effectively de-correlating confusion among tokens by using orthogonal components for separation. Moreover, the generated spectral basis involves intrinsic information about the downstream point clouds, enabling more targeted tuning. As a result, PointGST facilitates the efficient transfer of general knowledge to downstream tasks while significantly reducing training costs. Extensive experiments on challenging point cloud datasets across various tasks demonstrate that PointGST not only outperforms its fully fine-tuning counterpart but also significantly reduces trainable parameters, making it a promising solution for efficient point cloud learning. The code will be made available at https://github.com/jerryfeng2003/PointGST

Dingkang Liang, Tianrui Feng, Xin Zhou, Yumeng Zhang, Zhikang Zou, Xiang Bai• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationS3DIS (Area 5)
mIOU57.9
799
Part SegmentationShapeNetPart (test)
mIoU (Inst.)85.8
312
Point Cloud ClassificationModelNet40 (test)
Accuracy93.6
224
Object ClassificationScanObjectNN OBJ_BG
Accuracy98.97
215
Part SegmentationShapeNetPart
mIoU (Instance)85.8
198
Object ClassificationScanObjectNN PB_T50_RS
Accuracy94.83
195
Object ClassificationModelNet40 (test)--
180
Object ClassificationScanObjectNN OBJ_ONLY
Overall Accuracy97.59
166
Point Cloud ClassificationScanObjectNN PB_T50_RS (test)
Overall Accuracy89.49
91
Shape classificationScanObjectNN PB_T50_RS
OA89.97
72
Showing 10 of 24 rows

Other info

Code

Follow for update