Shuffle Mamba: State Space Models with Random Shuffle for Multi-Modal Image Fusion

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Cao, Ke, He, Xuanhua, Hu, Tao, Xie, Chengjun, Zhou, Man, Zhang, Jie
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911472435593216
author Cao, Ke
He, Xuanhua
Hu, Tao
Xie, Chengjun
Zhou, Man
Zhang, Jie
author_facet Cao, Ke
He, Xuanhua
Hu, Tao
Xie, Chengjun
Zhou, Man
Zhang, Jie
contents Multi-modal image fusion integrates complementary information from different modalities to produce enhanced and informative images. Although State-Space Models, such as Mamba, are proficient in long-range modeling with linear complexity, most Mamba-based approaches use fixed scanning strategies, which can introduce biased prior information. To mitigate this issue, we propose a novel Bayesian-inspired scanning strategy called Random Shuffle, supplemented by a theoretically feasible inverse shuffle to maintain information coordination invariance, aiming to eliminate biases associated with fixed sequence scanning. Based on this transformation pair, we customized the Shuffle Mamba Framework, penetrating modality-aware information representation and cross-modality information interaction across spatial and channel axes to ensure robust interaction and an unbiased global receptive field for multi-modal image fusion. Furthermore, we develop a testing methodology based on Monte-Carlo averaging to ensure the model's output aligns more closely with expected results. Extensive experiments across multiple multi-modal image fusion tasks demonstrate the effectiveness of our proposed method, yielding excellent fusion quality compared to state-of-the-art alternatives. The code is available at https://github.com/caoke-963/Shuffle-Mamba.
format Preprint
id arxiv_https___arxiv_org_abs_2409_01728
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Shuffle Mamba: State Space Models with Random Shuffle for Multi-Modal Image Fusion
Cao, Ke
He, Xuanhua
Hu, Tao
Xie, Chengjun
Zhou, Man
Zhang, Jie
Computer Vision and Pattern Recognition
Multi-modal image fusion integrates complementary information from different modalities to produce enhanced and informative images. Although State-Space Models, such as Mamba, are proficient in long-range modeling with linear complexity, most Mamba-based approaches use fixed scanning strategies, which can introduce biased prior information. To mitigate this issue, we propose a novel Bayesian-inspired scanning strategy called Random Shuffle, supplemented by a theoretically feasible inverse shuffle to maintain information coordination invariance, aiming to eliminate biases associated with fixed sequence scanning. Based on this transformation pair, we customized the Shuffle Mamba Framework, penetrating modality-aware information representation and cross-modality information interaction across spatial and channel axes to ensure robust interaction and an unbiased global receptive field for multi-modal image fusion. Furthermore, we develop a testing methodology based on Monte-Carlo averaging to ensure the model's output aligns more closely with expected results. Extensive experiments across multiple multi-modal image fusion tasks demonstrate the effectiveness of our proposed method, yielding excellent fusion quality compared to state-of-the-art alternatives. The code is available at https://github.com/caoke-963/Shuffle-Mamba.
title Shuffle Mamba: State Space Models with Random Shuffle for Multi-Modal Image Fusion
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.01728