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

PointRFT: Explicit Reinforcement Fine-tuning for Point Cloud Few-shot Learning

About

Understanding spatial dynamics and semantics in point cloud is fundamental for comprehensive 3D comprehension. While reinforcement learning algorithms such as Group Relative Policy Optimization (GRPO) have recently achieved remarkable breakthroughs in large language models by incentivizing reasoning capabilities through strategic reward design, their potential remains largely unexplored in the 3D perception domain. This naturally raises a pivotal question: Can RL-based methods effectively empower 3D point cloud fine-tuning? In this paper, we propose PointRFT, the first reinforcement fine-tuning paradigm tailored specifically for point cloud representation learning. We select three prevalent 3D foundation models and devise specialized accuracy reward and dispersion reward functions to stabilize training and mitigate distribution shifts. Through comprehensive few-shot classification experiments comparing distinct training paradigms, we demonstrate that PointRFT consistently outperforms vanilla supervised fine-tuning (SFT) across diverse benchmarks. Furthermore, when organically integrated into a hybrid Pretraining-SFT-RFT paradigm, the representational capacity of point cloud foundation models is substantially unleashed, achieving state-of-the-art performance particularly under data-scarce scenarios.

Yankai Wang, Yiding Sun, Qirui Wang, Pengbo Li, Chaoyi Lu, Dongxu Zhang• 2026

Related benchmarks

TaskDatasetResultRank
Few-shot classificationModelNet40 10-way 10-shot
Accuracy95.2
105
Few-shot classificationModelNet40 10-way 20-shot
Accuracy96.5
105
10-way 1-shot Few-shot ClassificationShapeNetCore
Accuracy75.41
15
10-way 5-shot Few-shot ClassificationShapeNetCore
Accuracy76.33
15
Few-shot classificationScanObjectNN 5-way 1-shot (Split 1)
Accuracy56.9
15
Few-shot classificationScanObjectNN 5-way 1-shot (Split 2)
Accuracy58.76
15
Few-shot classificationScanObjectNN 5-way 1-shot (Split 3)
Accuracy61.54
15
Few-shot classificationScanObjectNN 5-way 5-shot (Split 1)
Accuracy77.34
15
Few-shot classificationScanObjectNN 5-way 5-shot (Split 2)
Accuracy76.88
15
Few-shot classificationScanObjectNN 5-way 5-shot (Split 3)
Accuracy0.7923
15
Showing 10 of 10 rows

Other info

Follow for update