Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

StitchFusion: Weaving Any Visual Modalities to Enhance Multimodal Semantic Segmentation

About

Multimodal semantic segmentation shows significant potential for enhancing segmentation accuracy in complex scenes. However, current methods often incorporate specialized feature fusion modules tailored to specific modalities, thereby restricting input flexibility and increasing the number of training parameters. To address these challenges, we propose StitchFusion, a straightforward yet effective modal fusion framework that integrates large-scale pre-trained models directly as encoders and feature fusers. This approach facilitates comprehensive multi-modal and multi-scale feature fusion, accommodating any visual modal inputs. Specifically, Our framework achieves modal integration during encoding by sharing multi-modal visual information. To enhance information exchange across modalities, we introduce a multi-directional adapter module (MultiAdapter) to enable cross-modal information transfer during encoding. By leveraging MultiAdapter to propagate multi-scale information across pre-trained encoders during the encoding process, StitchFusion achieves multi-modal visual information integration during encoding. Extensive comparative experiments demonstrate that our model achieves state-of-the-art performance on four multi-modal segmentation datasets with minimal additional parameters. Furthermore, the experimental integration of MultiAdapter with existing Feature Fusion Modules (FFMs) highlights their complementary nature. Our code is available at StitchFusion_repo.

Bingyu Li, Da Zhang, Zhiyuan Zhao, Junyu Gao, Xuelong Li• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationMFNet (test)
mIoU60.7
134
Semantic segmentationNYU V2
mIoU59.6
74
Semantic segmentationPST900 (test)
mIoU89.4
72
Semantic segmentationFMB (test)
mIoU68.7
59
Semantic segmentationDELIVER (test)
mIoU53.4
10
Multimodal Semantic SegmentationMcubes
mIoU55.9
9
Multimodal Semantic SegmentationSUN
mIoU54.8
8
Multimodal Semantic SegmentationDeLiVER
mIoU70.3
6
Semantic segmentationDeLiVER (val)
mIoU68.2
5
Showing 9 of 9 rows

Other info

Code

Follow for update