Threshold Attention Network for Semantic Segmentation of Remote Sensing Images

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Long, Wei, Zhang, Yongjun, Cui, Zhongwei, Xu, Yujie, Zhang, Xuexue
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913648341942272
author Long, Wei
Zhang, Yongjun
Cui, Zhongwei
Xu, Yujie
Zhang, Xuexue
author_facet Long, Wei
Zhang, Yongjun
Cui, Zhongwei
Xu, Yujie
Zhang, Xuexue
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2501_07984
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Threshold Attention Network for Semantic Segmentation of Remote Sensing Images
Long, Wei
Zhang, Yongjun
Cui, Zhongwei
Xu, Yujie
Zhang, Xuexue
Computer Vision and Pattern Recognition
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.
title Threshold Attention Network for Semantic Segmentation of Remote Sensing Images
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2501.07984