Aggregated-Attention Transformation Network for Hyperspectral Image Classification

Chunyu Pu,Hong Huang,Yuan Li
DOI: https://doi.org/10.1109/jstars.2022.3190071
IF: 4.715
2022-01-01
IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing
Abstract:Convolutional neural networks (CNNs) have achieved great success in hyperspectral image (HSI) classification. However, most CNNs require a substantial amount of manually labeled training data to yield good performances, and the expert annotation of HSI data is labor-intensive, but existing CNNs fail to effectively extract the joint spatial-spectral features from a small training set. To tackle this issue, inspired by the attention mechanism, an aggregated-attention transformation network (AATN) was proposed for HSI classification. This network takes advantage of the split-transform-merge strategy to change the number of the cardinality (the size of transformation set) for improving generalization ability with a small number of training samples while alleviating the overfitting phenomenon. Moreover, each aggregation attention module is designed to explore the attention across spatial-spectral features of different receptive fields, and interactively guide the recalibration of discriminative features. By retaining the uniform architecture, AATN can be used for HSI classification tasks without introducing excessive computational costs, and the proposed framework can learn representations with limited training samples due to the effectiveness and practicality of the attention-driven features. The proposed AATN obtains competitive performance compared with some state-of-the-art methods in qualitative and quantitative experiments on four benchmark HSI datasets.
What problem does this paper attempt to address?