arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.13408cs.CV

AMMUNet:用于遥感图像分割的多尺度注意力图合并

AMMUNet: Multi-Scale Attention Map Merging for Remote Sensing Image Segmentation

School of Remote Sensing and Information Engineering, Wuhan University(武汉大学遥感信息工程学院)

Yang Yang, Shunyi Zheng

AI总结:

本文提出基于UNet的AMMUNet框架,通过颗粒多头自注意力模块和注意力图合并机制高效降低计算成本并建模全局注意力,在Vaihingen和Potsdam数据集上分别取得75.48%和77.90%的mIoU,实现精确的遥感语义分割。

摘要:

深度学习的发展推动了遥感语义分割的显著进步。注意力机制虽然能够实现全局建模并利用上下文信息,但面临计算成本高的挑战,且需要基于窗口的操作,这削弱了捕获长距离依赖的能力,阻碍了其在遥感图像处理中的有效性。在本文中,我们提出了AMMUNet,这是一个基于UNet的框架,采用多尺度注意力图合并,包含两项关键创新:颗粒多头自注意力(GMSA)模块和注意力图合并机制(AMMM)。与全局多头自注意力机制相比,GMSA高效获取全局信息,同时大幅降低计算成本。这是通过策略性地利用维度对应来对齐粒度以及减少相对位置偏置参数来实现的,从而优化了计算效率。提出的AMMM使用固定的掩码模板,将多尺度注意力图有效结合为统一表示,实现全局注意力机制的建模。实验评估突出了我们方法的优越性能,在具有挑战性的Vaihingen数据集上实现了75.48%的平均交并比(mIoU),在Potsdam数据集上实现了77.90%的卓越成绩,证明了我们的方法在精确遥感语义分割中的优越性。代码可在https://github.com/interpretty/AMMUNet获取。

英文摘要:

The advancement of deep learning has driven notable progress in remote sensing semantic segmentation. Attention mechanisms, while enabling global modeling and utilizing contextual information, face challenges of high computational costs and require window-based operations that weaken capturing long-range dependencies, hindering their effectiveness for remote sensing image processing. In this letter, we propose AMMUNet, a UNet-based framework that employs multi-scale attention map merging, comprising two key innovations: the granular multi-head self-attention (GMSA) module and the attention map merging mechanism (AMMM). GMSA efficiently acquires global information while substantially mitigating computational costs in contrast to global multi-head self-attention mechanism. This is accomplished through the strategic utilization of dimension correspondence to align granularity and the reduction of relative position bias parameters, thereby optimizing computational efficiency. The proposed AMMM effectively combines multi-scale attention maps into a unified representation using a fixed mask template, enabling the modeling of global attention mechanism. Experimental evaluations highlight the superior performance of our approach, achieving remarkable mean intersection over union (mIoU) scores of 75.48\% on the challenging Vaihingen dataset and an exceptional 77.90\% on the Potsdam dataset, demonstrating the superiority of our method in precise remote sensing semantic segmentation. Codes are available at https://github.com/interpretty/AMMUNet.

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

↑