arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.03059cs.CVcs.AIcs.LG

Point-PEFT:面向三维预训练模型的参数高效微调

Point-PEFT: Parameter-Efficient Fine-Tuning for 3D Pre-trained Models

Shanghai AI Laboratory(上海人工智能实验室);Northwestern Polytechnical University(西北工业大学)

Yiwen Tang, Ray Zhang, Zoey Guo, Dong Wang, Zhigang Wang, Bin Zhao, Xuelong Li

AI总结:

本文提出Point-PEFT框架,通过冻结预训练三维模型大部分参数并仅微调点先验提示和几何感知适配器模块,在仅使用5%可训练参数的情况下实现优于全量微调的下游任务性能。

摘要:

预训练大模型的流行已经彻底改变了语言、视觉和多模态等多个领域的下游任务。为了最小化下游任务的适配成本,许多参数高效微调(PEFT)技术被提出用于语言和二维图像预训练模型。然而,针对三维预训练模型的专用PEFT方法仍未得到充分探索。为此,我们提出了Point-PEFT,这是一种以最少的可学习参数适配点云预训练模型的新框架。具体而言,对于一个预训练的三维模型,我们冻结其大部分参数,仅在下游任务上微调新添加的PEFT模块,该模块由一个点先验提示(Point-prior Prompt)和一个几何感知适配器(Geometry-aware Adapter)组成。点先验提示采用一组可学习的提示令牌,我们提出利用领域特定知识构建记忆库,并使用无参数注意力来增强提示令牌。几何感知适配器旨在聚合空间邻域内的点云特征,通过局部交互捕获细粒度的几何信息。大量实验表明,我们的Point-PEFT在多个下游任务上能够取得比全量微调更好的性能,同时仅使用5%的可训练参数,证明了我们方法的高效性和有效性。代码已发布在https://github.com/Ivan-Tang-3D/Point-PEFT。

英文摘要:

The popularity of pre-trained large models has revolutionized downstream tasks across diverse fields, such as language, vision, and multi-modality. To minimize the adaption cost for downstream tasks, many Parameter-Efficient Fine-Tuning (PEFT) techniques are proposed for language and 2D image pre-trained models. However, the specialized PEFT method for 3D pre-trained models is still under-explored. To this end, we introduce Point-PEFT, a novel framework for adapting point cloud pre-trained models with minimal learnable parameters. Specifically, for a pre-trained 3D model, we freeze most of its parameters, and only tune the newly added PEFT modules on downstream tasks, which consist of a Point-prior Prompt and a Geometry-aware Adapter. The Point-prior Prompt adopts a set of learnable prompt tokens, for which we propose to construct a memory bank with domain-specific knowledge, and utilize a parameter-free attention to enhance the prompt tokens. The Geometry-aware Adapter aims to aggregate point cloud features within spatial neighborhoods to capture fine-grained geometric information through local interactions. Extensive experiments indicate that our Point-PEFT can achieve better performance than the full fine-tuning on various downstream tasks, while using only 5% of the trainable parameters, demonstrating the efficiency and effectiveness of our approach. Code is released at https://github.com/Ivan-Tang-3D/Point-PEFT.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑