中文

KPConvX:使用核注意力现代化核点卷积

计算机视觉与模式识别 2024-05-24 v1

摘要

在深度点云理解领域,KPConv是一种独特的架构,它使用核点在空间中定位卷积权重,而不是依赖多层感知器(MLP)编码。虽然它最初取得了成功,但后来被采用更新设计和训练策略的现代MLP网络所超越。基于核点原理,我们提出了两种新颖的设计:KPConvD(深度可分离KPConv),一种更轻量的设计,允许使用更深的架构;以及KPConvX,一种创新设计,使用核注意力值来缩放KPConvD的深度可分离卷积权重。通过使用KPConvX结合现代架构和训练策略,我们能够在ScanObjectNN、Scannetv2和S3DIS数据集上超越当前最先进的方法。我们通过消融研究验证了我们的设计选择,并发布了我们的代码和模型。

关键词

引用

@article{arxiv.2405.13194,
  title  = {KPConvX: Modernizing Kernel Point Convolution with Kernel Attention},
  author = {Hugues Thomas and Yao-Hung Hubert Tsai and Timothy D. Barfoot and Jian Zhang},
  journal= {arXiv preprint arXiv:2405.13194},
  year   = {2024}
}

备注

CVPR 2024