MMSFormer: Multimodal Transformer for Material and Semantic Segmentation

7 Sep 2023  ·  Md Kaykobad Reza, Ashley Prater-Bennette, M. Salman Asif ·

Leveraging information across diverse modalities is known to enhance performance on multimodal segmentation tasks. However, effectively fusing information from different modalities remains challenging due to the unique characteristics of each modality. In this paper, we propose a novel fusion strategy that can effectively fuse information from different modality combinations. We also propose a new model named Multi-Modal Segmentation TransFormer (MMSFormer) that incorporates the proposed fusion strategy to perform multimodal material and semantic segmentation tasks. MMSFormer outperforms current state-of-the-art models on three different datasets. As we begin with only one input modality, performance improves progressively as additional modalities are incorporated, showcasing the effectiveness of the fusion block in combining useful information from diverse input modalities. Ablation studies show that different modules in the fusion block are crucial for overall model performance. Furthermore, our ablation studies also highlight the capacity of different input modalities to improve performance in the identification of different types of materials. The code and pretrained models will be made available at https://github.com/csiplab/MMSFormer.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Semantic Segmentation FMB Dataset MMSFormer (RGB-Infrared) mIoU 61.70 # 1
Semantic Segmentation FMB Dataset MMSFormer (RGB) mIoU 57.20 # 2
Semantic Segmentation MCubeS MMSFormer (RGB-A-D-N) mIoU 53.11% # 1
Semantic Segmentation MCubeS MMSFormer (RGB) mIoU 50.44% # 7
Semantic Segmentation MCubeS MMSFormer (RGB-A) mIoU 51.30% # 4
Semantic Segmentation MCubeS MMSFormer (RGB-A-D) mIoU 52.05% # 2
Semantic Segmentation MCubeS (P) MMSFormer (RGB-A-D) mIoU 52.03 # 1
Semantic Segmentation MCubeS (P) MMSFormer (RGB) mIoU 50.44 # 5
Semantic Segmentation MCubeS (P) MMSFormer (RGB-A) mIoU 51.30 # 2
Thermal Image Segmentation PST900 MMSFormer mIoU 87.45 # 3

Methods