Stochastic Depth Residual Network for Hyperspectral Image Classification
File version
Author(s)
Tong, L
Zhou, J
Qian, B
Yu, J
Xiao, C
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
Size
File type(s)
Location
License
Abstract
The convolutional neural network (CNN) is a feedforward neural network with deep structure and convolution operation. In the hyperspectral image (HSI) classification, CNN has demonstrated excellent performance in extracting spectral and spatial information. However, the inherent complexity and high dimension of HSIs still limit the performance of most neural network models. The powerful feature extraction ability of CNN is normally achieved by dozens or more layers, which brings a series of problems such as gradient vanishing, overfitting, and slow training speed. In order to address these problems, this article presents a CNN architecture-based stochastic depth residual network (SDRN), which is specially designed for HSI data. This model takes the original 3-D cube as the input and 3-D convolution is used to extract abundant spectral and spatial features through corresponding residual blocks. In order to reduce the training time, we adopt a stochastic depth strategy. For each small batch, a sublayer is randomly discarded by an identity function. During the testing stage, the residual network with complete depth is used. Experiments on three datasets and a comparison of the state-of-art methods show that SDRN has great advantages in accuracy and training time compared with state-of-the-art HSI classification methods.
Journal Title
IEEE Transactions on Geoscience and Remote Sensing
Conference Title
Book Title
Edition
Volume
Issue
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
Item Access Status
Note
This publication has been entered as an advanced online version in Griffith Research Online.
Access the data
Related item(s)
Subject
Geophysics
Artificial intelligence
Electrical engineering
Geomatic engineering
Persistent link to this record
Citation
Gao, Z; Tong, L; Zhou, J; Qian, B; Yu, J; Xiao, C, Stochastic Depth Residual Network for Hyperspectral Image Classification, IEEE Transactions on Geoscience and Remote Sensing, 2021