arXiv:2501.07984cs.CV2025-01被引 22

提出新型阈值注意力机制,高效提升遥感图像语义分割精度

Threshold Attention Network for Semantic Segmentation of Remote Sensing Images

  • 设计阈值注意力机制,动态筛选关键特征,降低计算开销
  • 在Vaihingen和Potsdam数据集上达到新最佳性能,优于现有模型
  • 适合遥感图像分割任务,尤其适用于计算资源受限场景

遥感图像语义分割在植被监测、灾害管理与城市规划中至关重要。以往研究显示,自注意力机制(SA)能有效捕捉像素间的长程依赖关系,提升分割效果。然而,高密度的注意力特征图导致计算复杂度呈指数增长,并引入冗余信息,影响特征表示。受传统阈值分割算法启发,本文提出一种新型阈值注意力机制(TAM),显著减少计算负担,同时更精准建模特征图各区域间的相关性。基于此,构建了阈值注意力网络(TANet),包含浅层特征增强模块(AFEM)与深层多尺度特征提取的阈值注意力金字塔池化模块(TAPP)。在ISPRS Vaihingen与Potsdam数据集上的大量实验表明,TANet在分割性能上优于当前最先进模型。

原文摘要 · Abstract (English)

Semantic segmentation of remote sensing images is essential for various applications, including vegetation monitoring, disaster management, and urban planning. Previous studies have demonstrated that the self-attention mechanism (SA) is an effective approach for designing segmentation networks that can capture long-range pixel dependencies. SA enables the network to model the global dependencies between the input features, resulting in improved segmentation outcomes. However, the high density of attentional feature maps used in this mechanism causes exponential increases in computational complexity. Additionally, it introduces redundant information that negatively impacts the feature representation. Inspired by traditional threshold segmentation algorithms, we propose a novel threshold attention mechanism (TAM). This mechanism significantly reduces computational effort while also better modeling the correlation between different regions of the feature map. Based on TAM, we present a threshold attention network (TANet) for semantic segmentation. TANet consists of an attentional feature enhancement module (AFEM) for global feature enhancement of shallow features and a threshold attention pyramid pooling module (TAPP) for acquiring feature information at different scales for deep features. We have conducted extensive experiments on the ISPRS Vaihingen and Potsdam datasets. The results demonstrate the validity and superiority of our proposed TANet compared to the most state-of-the-art models.

遥感图像语义分割注意力机制高效网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。