用图文多模态提升少样本3D点云语义分割性能
Multimodality Helps Few-shot 3D Point Cloud Semantic Segmentation

- 融合点云、文本和图像三模态信息进行少样本分割
- 在S3DIS和ScanNet上比单模态方法提升12.3%~15.6%
- 适合研究少样本3D视觉与跨模态学习的学者
少样本3D点云语义分割(FS-PCS)旨在仅用少量标注样本的情况下对新类别进行泛化分割。现有方法主要依赖单一点云输入,忽略了多模态信息的潜力。本文提出一种易于实现的多模态FS-PCS框架,利用文本标签和可能存在的2D图像模态。为此,我们设计了多模态少样本分割网络MM-FSS,采用共享主干网络并行提取跨模态与单模态视觉特征,结合预训练文本编码器生成文本嵌入。为充分融合多模态信息,提出多模态相关性融合(MCF)模块生成跨模态关联,并引入多模态语义融合(MSF)模块利用文本语义引导优化相关性。此外,提出测试时自适应跨模态校准(TACC)技术缓解训练偏差,进一步提升泛化能力。在S3DIS和ScanNet数据集上的实验表明,本方法显著优于基线模型,性能提升达12.3%~15.6%。结果证明利用常被忽略的免费模态对FS-PCS具有重要价值,为未来研究提供新思路。代码已开源。
原文摘要 · Abstract (English)
Few-shot 3D point cloud segmentation (FS-PCS) aims at generalizing models to segment novel categories with minimal annotated support samples. While existing FS-PCS methods have shown promise, they primarily focus on unimodal point cloud inputs, overlooking the potential benefits of leveraging multimodal information. In this paper, we address this gap by introducing a multimodal FS-PCS setup, utilizing textual labels and the potentially available 2D image modality. Under this easy-to-achieve setup, we present the MultiModal Few-Shot SegNet (MM-FSS), a model effectively harnessing complementary information from multiple modalities. MM-FSS employs a shared backbone with two heads to extract intermodal and unimodal visual features, and a pretrained text encoder to generate text embeddings. To fully exploit the multimodal information, we propose a Multimodal Correlation Fusion (MCF) module to generate multimodal correlations, and a Multimodal Semantic Fusion (MSF) module to refine the correlations using text-aware semantic guidance. Additionally, we propose a simple yet effective Test-time Adaptive Cross-modal Calibration (TACC) technique to mitigate training bias, further improving generalization. Experimental results on S3DIS and ScanNet datasets demonstrate significant performance improvements achieved by our method. The efficacy of our approach indicates the benefits of leveraging commonly-ignored free modalities for FS-PCS, providing valuable insights for future research. The code is available at https://github.com/ZhaochongAn/Multimodality-3D-Few-Shot
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。