TBTA-D2Net: a novel hyperspectral image classification method based on triple-branch ternary-attention mechanism and Dense2Net

Abstract

ABSTRACTRecently, there has been a growing interest in the hyperspectral image (HSI) classification methods that employ deep learning techniques in small sample cases. To address issues with network degradation and enhance the extraction of discriminative HSI features, this article proposes a TBTA-D2Net network utilizing a triple-branch ternary-attention mechanism and Dense2Net. Furthermore, a new deep model optimizer named Adan is introduced to improve the training speed of the network model. This article takes spatial information as a two-dimensional vector, extracting spectral features as well as spatial-X and spatial-Y features separately in three branches. Each branch includes a Dense2Net bottleneck module and an attention module. Classification is achieved by fusing the features extracted from the three branches. Experimental results on four public datasets indicate that TBTA-D2Net can achieve competitive results over state-of-the-art methods. The code is available at https://github.com/TeresaTing/TBTA-D2Net.