线性复杂度长序列建模
将Mamba的状态空间建模与CNN局部特征提取结合,在不牺牲长程依赖的前提下大幅降低遥感图像场景分类的计算开销。
Qi, Xinhu · Zhijie, Hu · Gao, Yue · 陈彦斌 · 张艳宁
2024
采用状态空间模型Mamba,将复杂度从二次降至线性,适合处理长序列遥感数据。
通过HWC-Mamba在三维图像空间实现1D选择性扫描,模型能全局捕捉空间依赖。
引入CNN分支并与Mamba特征融合,弥补Mamba在局部细节上的不足。
在AID和UCM两个数据集上的实验证明了方法的可行性和有效性。