LMFNet: An Efficient Multimodal Fusion Approach for Semantic Segmentation in High-Resolution Remote Sensing

Tong Wang,Guanzhou Chen,Xiaodong Zhang,Chenxi Liu,Xiaoliang Tan,Jiaqi Wang,Chanjuan He,Wenlin Zhou
2024-04-21
Abstract:Despite the rapid evolution of semantic segmentation for land cover classification in high-resolution remote sensing imagery, integrating multiple data modalities such as Digital Surface Model (DSM), RGB, and Near-infrared (NIR) remains a challenge. Current methods often process only two types of data, missing out on the rich information that additional modalities can provide. Addressing this gap, we propose a novel \textbf{L}ightweight \textbf{M}ultimodal data \textbf{F}usion \textbf{Net}work (LMFNet) to accomplish the tasks of fusion and semantic segmentation of multimodal remote sensing images. LMFNet uniquely accommodates various data types simultaneously, including RGB, NirRG, and DSM, through a weight-sharing, multi-branch vision transformer that minimizes parameter count while ensuring robust feature extraction. Our proposed multimodal fusion module integrates a \textit{Multimodal Feature Fusion Reconstruction Layer} and \textit{Multimodal Feature Self-Attention Fusion Layer}, which can reconstruct and fuse multimodal features. Extensive testing on public datasets such as US3D, ISPRS Potsdam, and ISPRS Vaihingen demonstrates the effectiveness of LMFNet. Specifically, it achieves a mean Intersection over Union ($mIoU$) of 85.09\% on the US3D dataset, marking a significant improvement over existing methods. Compared to unimodal approaches, LMFNet shows a 10\% enhancement in $mIoU$ with only a 0.5M increase in parameter count. Furthermore, against bimodal methods, our approach with trilateral inputs enhances $mIoU$ by 0.46 percentage points.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper attempts to address the problem of how to effectively fuse multiple data modalities (such as Digital Surface Model DSM, RGB, Near-Infrared NIR, etc.) in high-resolution remote sensing images to improve the performance of land cover classification tasks. Current methods typically can only handle two types of data, ignoring the rich information provided by other modalities. To solve this problem, the authors propose a lightweight multimodal data fusion network (LMFNet) aimed at achieving the fusion and semantic segmentation of multimodal remote sensing images. Specifically, the main contributions of the paper include: 1. Designing a novel Multimodal Feature Fusion Module (MFM) that can seamlessly fuse any number of modal data without modifying the structure of the fusion module. 2. Proposing a comprehensive multimodal fusion and semantic segmentation network LMFNet, including MFM and the preprocessing flow of multimodal data. 3. Conducting experiments on three public datasets to deeply analyze the internal fusion principles of the proposed fusion module and the details before and after multimodal feature fusion. The uniqueness of LMFNet lies in its ability to handle multiple data types (such as RGB, NirRG, and DSM) simultaneously and minimize the number of parameters through a multi-branch vision transformer with shared weights, ensuring strong feature extraction capabilities. Additionally, the multimodal fusion module in LMFNet includes a Multimodal Feature Fusion Reconstruction Layer and a Multimodal Feature Self-Attention Fusion Layer, capable of reconstructing and fusing multimodal features. Experimental results show that LMFNet achieved an 85.09% mean Intersection over Union (mIoU) on the US3D dataset, improving by 10% compared to single-modal methods, with only an additional 0.5M parameters. Compared to dual-modal methods, the LMFNet with three-modal input improved by 0.46 percentage points in mIoU. These results indicate that LMFNet not only has scalability and accuracy but also demonstrates efficiency in parameter usage, showing its broad application potential in remote sensing image analysis.