HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Instance-aware Dynamic Prompt Tuning for Pre-trained Point Cloud Models

Zha Yaohua ; Wang Jinpeng ; Dai Tao ; Chen Bin ; Wang Zhi ; Xia Shu-Tao

Instance-aware Dynamic Prompt Tuning for Pre-trained Point Cloud Models

Abstract

Pre-trained point cloud models have found extensive applications in 3Dunderstanding tasks like object classification and part segmentation. However,the prevailing strategy of full fine-tuning in downstream tasks leads to largeper-task storage overhead for model parameters, which limits the efficiencywhen applying large-scale pre-trained models. Inspired by the recent success ofvisual prompt tuning (VPT), this paper attempts to explore prompt tuning onpre-trained point cloud models, to pursue an elegant balance betweenperformance and parameter efficiency. We find while instance-agnostic staticprompting, e.g. VPT, shows some efficacy in downstream transfer, it isvulnerable to the distribution diversity caused by various types of noises inreal-world point cloud data. To conquer this limitation, we propose a novelInstance-aware Dynamic Prompt Tuning (IDPT) strategy for pre-trained pointcloud models. The essence of IDPT is to develop a dynamic prompt generationmodule to perceive semantic prior features of each point cloud instance andgenerate adaptive prompt tokens to enhance the model's robustness. Notably,extensive experiments demonstrate that IDPT outperforms full fine-tuning inmost tasks with a mere 7% of the trainable parameters, providing a promisingsolution to parameter-efficient learning for pre-trained point cloud models.Code is available at \url{https://github.com/zyh16143998882/ICCV23-IDPT}.

Code Repositories

zyh16143998882/iccv23-idpt
Official
pytorch
Mentioned in GitHub
zyh16143998882/IDPT
Official
pytorch
Mentioned in GitHub
zyh16143998882/aaai24-pointfemae
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
3d-point-cloud-classification-on-modelnet40IDPT
Overall Accuracy: 94.4
3d-point-cloud-classification-on-scanobjectnnIDPT
OBJ-BG (OA): 93.12
Overall Accuracy: 88.51
few-shot-3d-point-cloud-classification-on-1IDPT
Overall Accuracy: 97.3
few-shot-3d-point-cloud-classification-on-2IDPT
Overall Accuracy: 97.9
few-shot-3d-point-cloud-classification-on-3IDPT
Overall Accuracy: 92.8
few-shot-3d-point-cloud-classification-on-4IDPT
Overall Accuracy: 95.4

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Instance-aware Dynamic Prompt Tuning for Pre-trained Point Cloud Models | Papers | HyperAI