The distinction of similar classes has always been the core issue in image classification. In this paper, a new hierarchical classification process based on three-dimensional attention soft augmentation (HC-3DAA) is proposed to improve the accuracy of classifiers, especially for the accuracy between similar classes. In HC-3DAA processing, the merging matrix is firstly constructed based on the validation confusion matrix to measure the similarity among different classes. Specifically, the 3D attention soft augmentation module combined with CutMix is designed for guiding the network model to focus on the key discriminative features. Then the extracted 3D feature differences between similar classes are inserted into the attention module for the reclassification to get higher classification accuracy. To evaluate the performance of HC-3DAA, CutMix models with different feature dimensions and the hierarchical classification module are separately discussed on two widely used hyperspectral datasets. Two different classifiers 3D-CNN and ResNet are included in the comparative analysis. Besides, experimental results also demonstrate that the proposed HC-3DAA outperforms several state-of-the-art attention-based methods.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.