论文阅读【PCT: Point Cloud Transformer】

PCT: Point Cloud Transforme

论文地址:PCT: Point Cloud Transformer

发表于:2021年清华大学发表于CVMJ的论文

本文的主要贡献总结如下:

1、提出了一种新的基于转换器的点云学习框架PCT,该框架非常适合于非结构化、无序的不规则域点云数据。

2、提出了使用隐式拉普拉斯算子和归一化细化的偏移注意(offset-attention)?,与Transformer中的原始自注意(self-attention)模块相比,它具有固有的置换不变性,更适合点云学习。

3、大量的实验表明,具有显式局部上下文增强的PCT在形状分类、零件分割和正常估计任务上达到了最先进的性能。

难点

  1. 点的无序性

特点

  1. transformer的使用
  2. 局部特征的注意

introduction

相比自然语言处理的几点改进处:

  1. 基于坐标的输入嵌入模块

    融合了原始的位置编码和输入的位置编码到基于坐标的输入嵌入模块,因为每个点的位置唯一,所以基于这个模块生成独一无二的编码特征来代表空间位置。

  2. 优化了的偏移注意力模块

    基于原始注意力机制的优化

    改进点: 从注意特征更改为注意input of self-attention module 和 attention feature之间的偏移

    优势:

    (1) 通过刚性变换,同一物体的绝对坐标可以完全不同。因此,相对坐标通常更具有鲁棒性【绝对坐标可以不同,相对坐标更具参考价值】

    (2) 拉普拉斯矩阵(度矩阵与邻接矩阵之间的偏移量)被证明是非常有效的图卷积学习[3]【Q:通过图卷积来论证这一观点存疑?】

    在处理中将每行的和缩放为1(与原处理同样的操作)

  3. 邻近嵌入模块

    考虑局部点之间的注意力模块来关注局部几何特征

贡献

  1. 提出基于transformer的PCT的架构,针对处理不规则领域中无结构的、无序的点云
  2. 提出offset-attention(带有拉普拉斯算子)和归一化增强,比原始的self-attention更适合点云。
  3. 大量的实验表明,具有显式局部上下文增强的PCT在形状分类、部分分割和正常估计任务上取得了最先进的性能。

主要框架与细节

主要的框架与原始的transformer有相同的设计理念,只有位置编码被丢弃,输入特征(即点的坐标)已经包含了位置信息。

encoder中过程公式如下,Fe为 Input Embedding Module模块输出维度为128。encoder输出的特征Fo经过MA(Max-pooling和Average-pooling)提取全局有效特征Fg

对于分割的任务的操作:

整合了Fo和Fg, 经过结果LBR(Linear, BatchNorm and ReLU layers)和dropout操作之后输出各点类别

Input Embedding Module

两个级联的LBR(Linear, BatchNorm and ReLU layers),模块输出维度de=128

模型输入

点云特征维度N*d, N为点数,d为点的特征数,论文中为3, 可以自行调整

Offset-Attention(encoder中的一层)

Neighbor Embedding Module 组成:2个LBR+2个SG, SG作用逐步增大感受野,类似于CNN的作用, SG通过使用欧氏距离的KNN找到采样点的局部邻居团簇,整合特征.

采样方法使用farthest point sampling (FPS) algorithm,做到降采样的效果

SG该部分整体计算过程

评估

分割功能的模型测试在S3DIS数据集上, 为室内数据集,共分为13类

code

模型训练代码-其他人补充
官方作者发布

相关推荐
豆芽8192 小时前
感受野(Receptive Field)
人工智能·python·深度学习·yolo·计算机视觉
机器学习之心3 小时前
飞蛾扑火算法优化+Transformer四模型回归打包(内含MFO-Transformer-LSTM及单独模型)
算法·回归·lstm·transformer·飞蛾扑火算法优化
ccLianLian3 小时前
深度学习·经典模型·Transformer
人工智能·深度学习·transformer
Want5954 小时前
DeepSeek: 探索未来的深度学习搜索引擎
人工智能·深度学习·搜索引擎·aigc
kyle~5 小时前
深度学习---pytorch搭建深度学习模型(附带图片五分类实例)
pytorch·深度学习·分类
不学习怎么给老板打工?7 小时前
yolov8使用
深度学习
AI产品备案7 小时前
算法备案类型解析:如何判断你的算法属于哪种类型?
深度学习·安全
sagima_sdu7 小时前
论文笔记-基于多层感知器(MLP)的多变量桥式起重机自适应安全制动与距离预测
论文阅读
Jackilina_Stone7 小时前
【论文阅读/复现】RT-DETR的网络结构/训练/推理/验证/导出模型
论文阅读·python·目标检测·rt-detr
一点.点8 小时前
Sce2DriveX: 用于场景-到-驾驶学习的通用 MLLM 框架——论文阅读
论文阅读·大模型·自动驾驶·端到端