Composite Kernel Learning Network For Hyperspectral Image Classification

Zhe Wu,Jianjun Liu,Jinlong Yang,Zhiyong Xiao,Liang Xiao
DOI: https://doi.org/10.1080/01431161.2021.1934599
IF: 3.531
2021-01-01
International Journal of Remote Sensing
Abstract:The small sample problem has always been a serious challenge in hyperspectral image (HSI) classification. In order to obtain satisfactory results when the training samples are insufficient, the information around the training samples should be fully utilized. In this paper, we focus on small sample learning and propose a novel composite kernel learning network (CKLNet) for HSI classification. First, principal component analysis and extended morphological analysis are utilized to extract features. Then, we introduce generalized kernel method into deep learning technology. The spatial-spectral composite kernel learning (SSCKL) module is developed to construct discriminative and robust spatial-spectral generalized kernel features. In the process of constructing kernel features, the deep correlation information between samples is extracted simultaneously. The kernel hyperparameters in SSCKL are automatically learnt through backpropagation, thus avoiding the need to spend a lot of time on cross-validation. Finally, inspired by U-Net, a global-local feature extraction (GLFE) module is designed to extract spatial features of different scales. A set of classification probability maps can be obtained by the 1 x 1 convolutional layer in the GLFE module. Experimental results on three widely used datasets demonstrate the effectiveness of the proposed CKLNet.
What problem does this paper attempt to address?