P3T通过原型提示提升3D视觉语言模型泛化能力
P3T: Prototypical Point-level Prompt Tuning with Enhanced Generalization for 3D Vision-Language Models

- 点级提示生成+可学习文本提示,实现高效微调
- 在分类和少样本任务上媲美全量微调
- 适合需要强泛化的3D视觉任务研究者
随着预训练模型在3D点云领域的广泛应用,将其适配到下游任务变得愈发重要。然而,传统全量微调方法计算成本高且存储开销大。尽管提示调优成为高效替代方案,但常因过拟合而损害泛化能力。为此,我们提出原型点级提示调优(P$^3$T),一种针对预训练3D视觉语言模型(VLMs)的参数高效调优方法。P$^3$T包含两个组件:1)点提示器,为输入点云生成实例感知的点级提示;2)文本提示器,在输入文本中使用可学习提示而非手工设计提示。由于两个提示器直接作用于输入数据,P$^3$T在不牺牲泛化能力的前提下实现3D VLM的任务特异性适配。此外,为增强嵌入空间对齐——这是微调3D VLM的关键——我们引入原型损失以降低类内方差。大量实验表明,该方法在分类和少样本学习任务上达到或超越全量微调性能,并在跨数据集设置下的数据分布偏移中表现出稳健的泛化能力。代码已公开于 https://github.com/gyjung975/P3T。
原文摘要 · Abstract (English)
With the rise of pre-trained models in the 3D point cloud domain for a wide range of real-world applications, adapting them to downstream tasks has become increasingly important. However, conventional full fine-tuning methods are computationally expensive and storage-intensive. Although prompt tuning has emerged as an efficient alternative, it often suffers from overfitting, thereby compromising generalization capability. To address this issue, we propose Prototypical Point-level Prompt Tuning (P$^3$T), a parameter-efficient prompt tuning method designed for pre-trained 3D vision-language models (VLMs). P$^3$T consists of two components: 1) \textit{Point Prompter}, which generates instance-aware point-level prompts for the input point cloud, and 2) \textit{Text Prompter}, which employs learnable prompts into the input text instead of hand-crafted ones. Since both prompters operate directly on input data, P$^3$T enables task-specific adaptation of 3D VLMs without sacrificing generalizability. Furthermore, to enhance embedding space alignment, which is key to fine-tuning 3D VLMs, we introduce a prototypical loss that reduces intra-category variance. Extensive experiments demonstrate that our method matches or outperforms full fine-tuning in classification and few-shot learning, and further exhibits robust generalization under data shift in the cross-dataset setting. The code is available at \textcolor{violet}{https://github.com/gyjung975/P3T}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。