Hyperspectral Remote Sensing Image Classification Using Three-Dimensional-squeeze-and-excitation-densenet (3D-Se-densenet)

Guandong Li,Chunju Zhang,Runmin Lei,Xueying Zhang,Zhourun Ye,Xiaoli Li
DOI: https://doi.org/10.1080/2150704x.2019.1697001
IF: 2.369
2020-01-01
Remote Sensing Letters
Abstract:This study introduces the attention mechanism in hyperspectral remote sensing image (HSI) classification which can strengthen the information provided by important features, and weaken the non-essential information. We introduced the Squeeze-and-Excitation (SE) block embedded in three-dimensional densely connected convolutional network (3D-DenseNet) to form 3D-SE-DenseNet for HSI classifications. This model can learn a powerful network with low training costs and fast convergence speed, and avoids overfitting on small sample datasets. Two different 3D-SE-DenseNet models of 3D-SE-DenseNet and 3D-SE-DenseNet-BC were set up. Results from experiments show that the 3D-SE-DenseNet performs well on the Indian Pines, Pavia University, Botswana, and Kennedy Space Centre datasets.
What problem does this paper attempt to address?