Hyperspectral Band Selection Based on Deep Convolutional Neural Network and Distance Density

In this letter, a band-selection approach based on the deep convolutional neural network (CNN) and distance density (DD) is proposed. This method effectively mitigates the curse of dimensionality for hyperspectral images (HSIs). First, we use the hyperspectral full-band data to train a custom 1-D CNN to obtain a well-trained model. Second, we select band combinations based on DD. Using the rectified linear unit, which is the activation function of the CNN that is only activated with a nonzero value, we can effectively test the band combinations without retraining the model. Finally, the method selects the band combinations with the highest precision as the final selected bands. This precision measure is a new criterion for band selection. To further improve the performance, a data augmentation method based on DD is also proposed. To justify the effectiveness of the proposed method, experiments are conducted on two HSIs. The results show that the proposed method can acquire more satisfactory results than traditional methods.