Why mamba is effective? Exploit Linear Transformer-Mamba Network for Multi-Modality Image Fusion

Chenguang Zhu,Shan Gao,Huafeng Chen,Guangqian Guo,Chaowei Wang,Yaoxing Wang,Chen Shu Lei,Quanjiang Fan
2024-09-05
Abstract:Multi-modality image fusion aims to integrate the merits of images from different sources and render high-quality fusion images. However, existing feature extraction and fusion methods are either constrained by inherent local reduction bias and static parameters during inference (CNN) or limited by quadratic computational complexity (Transformers), and cannot effectively extract and fuse features. To solve this problem, we propose a dual-branch image fusion network called Tmamba. It consists of linear Transformer and Mamba, which has global modeling capabilities while maintaining linear complexity. Due to the difference between the Transformer and Mamba structures, the features extracted by the two branches carry channel and position information respectively. T-M interaction structure is designed between the two branches, using global learnable parameters and convolutional layers to transfer position and channel information respectively. We further propose cross-modal interaction at the attention level to obtain cross-modal attention. Experiments show that our Tmamba achieves promising results in multiple fusion tasks, including infrared-visible image fusion and medical image fusion. Code with checkpoints will be available after the peer-review process.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
This paper attempts to address the issues of feature extraction and fusion in multimodal image fusion. Specifically, existing feature extraction and fusion methods are either limited by the inherent local reduction bias and static parameters of Convolutional Neural Networks (CNNs) during inference or constrained by the quadratic computational complexity of Transformers, making it difficult to effectively extract and fuse features. To solve these problems, the authors propose a dual-branch image fusion network named Tmamba, which combines linear transformers and the Mamba structure. This network is capable of maintaining linear complexity while possessing global modeling capabilities. Additionally, it designs a cross-modal interaction module to obtain cross-modal attention, achieving excellent results in multiple fusion tasks, including infrared-visible image fusion and medical image fusion.