论文阅读【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

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

相关推荐
bittersuite21 分钟前
LeNet,AlexNet
人工智能·深度学习·机器学习
m4Rk_1 小时前
【论文阅读】Agent 记忆机制(20):RecMem——只在信息反复出现时进行长期记忆巩固
论文阅读·人工智能·学习·开源·github
m沐沐1 小时前
【自然语言处理】NLP分词与Word2Vec词向量——从原理到实战
人工智能·深度学习·opencv·机器学习·计算机视觉·自然语言处理·word2vec
满怀冰雪14 小时前
12-PaddlePaddle, 飞桨, 分类模型, 训练循环, 损失函数, 优化器, cross_entropy, Adam
人工智能·深度学习·分类·paddlepaddle
菜鸟‍16 小时前
【论文学习】MICCAI 2024 || SGSeg:通过自引导机制实现胸部X光片语言引导分割的无文本推理
人工智能·深度学习·学习
吴佳浩18 小时前
AI 核心技术解析|OPD:大模型开始复制的不再是知识,而是判断力
人工智能·深度学习·llm
就是一顿骚操作19 小时前
Dropout:神经网络正则化的经典解读
人工智能·深度学习·神经网络·论文解读
AndrewHZ20 小时前
【LLM技术全景】多模态大模型:当语言模型学会“看“和“听“
人工智能·gpt·深度学习·语言模型·自然语言处理·llm·多模态
湘美书院--湘美谈教育20 小时前
湘美谈教育互联网逻辑:AI时代的社会学猜想
大数据·人工智能·深度学习·机器学习·生活
m4Rk_21 小时前
【论文阅读】Agent 记忆机制(22):FluxMem——根据对话结构动态选择记忆组织方式
论文阅读·人工智能·学习