We present a Multimodal Interlaced Transformer (MIT) that jointly considers 2D and 3D data for weakly supervised point cloud segmentation. Research studies have shown that 2D and 3D features are complementary for point cloud segmentation. However, existing methods require extra 2D annotations to achieve 2D-3D information fusion. Considering the high annotation cost of point clouds, effective 2D and 3D feature fusion based on weakly supervised learning is in great demand. To this end, we propose a transformer model with two encoders and one decoder for weakly supervised point cloud segmentation using only scene-level class tags. Specifically, the two encoders compute the self-attended features for 3D point clouds and 2D multi-view images, respectively. The decoder implements interlaced 2D-3D cross-attention and carries out implicit 2D and 3D feature fusion. We alternately switch the roles of queries and key-value pairs in the decoder layers. It turns out that the 2D and 3D features are iteratively enriched by each other. Experiments show that it performs favorably against existing weakly supervised point cloud segmentation methods by a large margin on the S3DIS and ScanNet benchmarks. The project page will be available at https://jimmy15923.github.io/mit_web/.
翻译:我们提出了一种多模态交织Transformer(MIT),该模型联合考虑二维和三维数据以实现弱监督点云分割。研究表明,二维和三维特征在点云分割中具有互补性。然而,现有方法需要额外的二维标注来实现二维-三维信息融合。考虑到点云的高标注成本,基于弱监督学习的有效二维与三维特征融合需求迫切。为此,我们提出了一种包含两个编码器和一个解码器的Transformer模型,仅利用场景级类别标签实现弱监督点云分割。具体而言,两个编码器分别计算三维点云和二维多视角图像的自注意力特征,解码器则实现交织的二维-三维交叉注意力机制,并进行隐式的二维与三维特征融合。我们在解码器层中交替切换查询与键值对的角色,使得二维和三维特征能够相互迭代增强。实验表明,该方法在S3DIS和ScanNet基准测试中显著优于现有弱监督点云分割方法。项目页面将发布在https://jimmy15923.github.io/mit_web/。