Unbalanced Geologic Body Classification of Hyperspectral Data Based on Squeeze and Excitation Networks at Tianshan Area

Yuchen Liang,Zhengang Zhao,Hao Wang,Ying Cao,Tao Huang,Yasmine Medjadba,Yuntao Wang,Run Cheng Jiao,Siying Chen,Xianchuan Yu
DOI: https://doi.org/10.1109/igarss39084.2020.9323795
2020-01-01
Abstract:Hyperspectral data contains abundant spectral do main information, which is of great significance to classification of objects. However, due to the lack of labeled data, it is difficult to get an acceptable result by just using the small number of labeled data. We propose a semi-supervised classification model based on convolutional neural network and introduce the attention mechanism to balance the sample weight. After the convolution of the multi-layer network, more information is concentrated on the channels, so we use the Squeeze-and-Excitation block, which can adaptively recalibrates the channel-wise characteristic response by explicitly modelling the inter-channel dependencies. At the same time, we used focal loss to reduce the problem of poor training caused by uneven samples. We test our model on hyperspectral data at Tianshan area. From the result, we can find that our method can get a great result on the mineral classification task, which can be used for making geological map.
What problem does this paper attempt to address?