Hyperspectral Image Classification Based on 3-D Octave Convolution with Spatial-Spectral Attention Network

Xu Tang*, Fanbo Meng, Xiangrong Zhang, Yiu Ming Cheung*, Jingjing Ma, Fang Liu, Licheng Jiao

*Corresponding author for this work

Research output: Contribution to journalJournal articlepeer-review

115 Citations (Scopus)

Abstract

In recent years, with the development of deep learning (DL), the hyperspectral image (HSI) classification methods based on DL have shown superior performance. Although these DL-based methods have great successes, there is still room to improve their ability to explore spatial-spectral information. In this article, we propose a 3-D octave convolution with the spatial-spectral attention network (3DOC-SSAN) to capture discriminative spatial-spectral features for the classification of HSIs. Especially, we first extend the octave convolution model using 3-D convolution, namely, a 3-D octave convolution model (3D-OCM), in which four 3-D octave convolution blocks are combined to capture spatial-spectral features from HSIs. Not only the spatial information can be mined deeply from the high- and low-frequency aspects but also the spectral information can be taken into account by our 3D-OCM. Second, we introduce two attention models from spatial and spectral dimensions to highlight the important spatial areas and specific spectral bands that consist of significant information for the classification tasks. Finally, in order to integrate spatial and spectral information, we design an information complement model to transmit important information between spatial and spectral attention features. Through the information complement model, the beneficial parts of spatial and spectral attention features for the classification tasks can be fully utilized. Comparing with several existing popular classifiers, our proposed method can achieve competitive performance on four benchmark data sets.

Original languageEnglish
Pages (from-to)2430-2447
Number of pages18
JournalIEEE Transactions on Geoscience and Remote Sensing
Volume59
Issue number3
Early online date14 Jul 2020
DOIs
Publication statusPublished - Mar 2021

Scopus Subject Areas

  • Electrical and Electronic Engineering
  • General Earth and Planetary Sciences

User-Defined Keywords

  • Attention mechanism
  • deep learning (DL)
  • hyperspectral image (HSI) classification
  • information complement
  • spatial-spectral features

Fingerprint

Dive into the research topics of 'Hyperspectral Image Classification Based on 3-D Octave Convolution with Spatial-Spectral Attention Network'. Together they form a unique fingerprint.

Cite this