Optimal Parameter Selection in Hyperspectral Classification Based on Convolutional Neural Network

Abstract

Classification is a key technique in hyperspectral image (HSI) applications. Deep learning algorithms, which exhibit strong modeling and representational capabilities, have been successfully adopted in fields such as image and language processing. And convolutional neural networks (CNNs) have been used for HSI classification and some interesting results have been obtained. Owing to local connection and weight sharing, the number of parameters is reduced to some extent, but there are still many parameters and the deeper the network, the larger is the number of parameters. The network performance is strongly influenced by the parameter settings. To obtain the optimal CNN parameters for HSI classification, this paper proposes a classification method based on a CNN with parameter tuning (CNN-PT). The network parameters are tuned in turn according to the unique variable principle. Simulation results show that the proposed CNN-PT method has considerable potential for HSI classification compared to previous methods.