4 个月前

参数高效的频谱域点云学习微调

参数高效的频谱域点云学习微调

摘要

近期,利用预训练技术增强点云模型已成为一个热门研究课题。然而,现有的方法通常需要对预训练模型进行全面微调,才能在下游任务中达到令人满意的性能,这不仅存储密集且计算成本高昂。为了解决这一问题,我们提出了一种新的参数高效微调(PEFT)方法,专门针对点云模型,称为PointGST(点云图谱调优)。PointGST冻结了预训练模型,并引入了一个轻量级、可训练的点云图谱适配器(PCSA),以在频谱域内微调参数。该方法的核心思想基于两个观察结果:1)冻结模型中的内部标记在空间域中可能存在混淆;2)特定任务的内在信息对于将一般知识迁移到下游任务至关重要。具体而言,PointGST将点标记从空间域转换到频谱域,通过使用正交分量进行分离,有效减少了标记之间的混淆。此外,生成的频谱基包含了关于下游点云的内在信息,使得调整更加有针对性。因此,PointGST不仅促进了通用知识向下游任务的有效迁移,还显著降低了训练成本。在多个具有挑战性的点云数据集上进行的广泛实验表明,PointGST不仅超越了其全面微调的对应方法,在各种任务中表现出色,而且大幅减少了可训练参数的数量,成为一种高效的点云学习解决方案。它在ScanObjNN OBJ BG、OBJ OBLY和PB T50 RS数据集上的性能分别提高了2.28%、1.16%和2.78%,达到了99.48%、97.76%和96.18%的准确率。这一进展仅使用了0.67%的可训练参数,确立了新的最先进水平。

代码仓库

jerryfeng2003/pointgst
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
3d-point-cloud-classification-on-modelnet40PointGST
Overall Accuracy: 95.3
3d-point-cloud-classification-on-scanobjectnnPointGST
OBJ-BG (OA): 99.48
OBJ-ONLY (OA): 97.76
Overall Accuracy: 96.18

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
参数高效的频谱域点云学习微调 | 论文 | HyperAI超神经