arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.09293cs.CV

一种具有局部增强和状态共享的图像融合状态空间模型

A Novel State Space Model with Local Enhancement and State Sharing for Image Fusion

University of Electronic Science and Technology of China(电子科技大学)

Zihan Cao, Xiao Wu, Liang-Jian Deng, Yu Zhong

AI总结:

针对Mamba在图像融合中状态容量有限和光谱信息捕捉不足的问题,提出LE-Mamba模型,通过局部增强块和状态共享技术,在多光谱泛锐化及多光谱-高光谱融合数据集上取得最优性能。

摘要:

在图像融合任务中,来自不同来源的图像具有不同的特征。这推动了众多方法的发展,以探索在保留各自特征的同时更好地融合它们的方式。Mamba作为一种状态空间模型,已在自然语言处理领域出现。最近,许多研究尝试将Mamba扩展到视觉任务。然而,由于图像与因果语言序列的性质不同,Mamba有限的状态容量削弱了其对图像信息建模的能力。此外,Mamba的序列建模能力仅能处理空间信息,无法有效捕捉图像中丰富的光谱信息。受这些挑战的驱动,我们针对图像融合任务定制并改进了视觉Mamba网络。具体来说,我们提出了局部增强视觉Mamba块,称为LEVM。LEVM块能够提高网络的局部信息感知能力,并同时学习局部和全局空间信息。此外,我们提出了状态共享技术以增强空间细节并整合空间与光谱信息。最后,整体网络是一个基于视觉Mamba的多尺度结构,称为LE-Mamba。大量实验表明,所提出的方法在多光谱泛锐化和多光谱与高光谱图像融合数据集上达到了最先进的结果,并证明了所提出方法的有效性。代码可在 \u0068ttps://github.com/294coder/Efficient-MIF 获取。

英文摘要:

In image fusion tasks, images from different sources possess distinct characteristics. This has driven the development of numerous methods to explore better ways of fusing them while preserving their respective characteristics.Mamba, as a state space model, has emerged in the field of natural language processing. Recently, many studies have attempted to extend Mamba to vision tasks. However, due to the nature of images different from causal language sequences, the limited state capacity of Mamba weakens its ability to model image information. Additionally, the sequence modeling ability of Mamba is only capable of spatial information and cannot effectively capture the rich spectral information in images. Motivated by these challenges, we customize and improve the vision Mamba network designed for the image fusion task. Specifically, we propose the local-enhanced vision Mamba block, dubbed as LEVM. The LEVM block can improve local information perception of the network and simultaneously learn local and global spatial information. Furthermore, we propose the state sharing technique to enhance spatial details and integrate spatial and spectral information. Finally, the overall network is a multi-scale structure based on vision Mamba, called LE-Mamba. Extensive experiments show the proposed methods achieve state-of-the-art results on multispectral pansharpening and multispectral and hyperspectral image fusion datasets, and demonstrate the effectiveness of the proposed approach. Codes can be accessed at \url{https://github.com/294coder/Efficient-MIF}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑