A Convolutional Neural Network with Fletcher–Reeves Algorithm for Hyperspectral Image Classification

• Main Authors: Chen Chen, Yi Ma, Guangbo Ren
• Format: Article
• Language: English
• Published: MDPI AG 2019-06-01
• Series: Remote Sensing
• Subjects: convolutional neural network (CNN); Fletcher–Reeves algorithm (F–R); conjugate gradient; coastal wetland classification; hyperspectral imagery
• Online Access: https://www.mdpi.com/2072-4292/11/11/1325

Deep learning models, especially the convolutional neural networks (CNNs), are very active in hyperspectral remote sensing image classification. In order to better apply the CNN model to hyperspectral classification, we propose a CNN model based on Fletcher−Reeves algorithm (F−R CNN), which uses the Fletcher−Reeves (F−R) algorithm for gradient updating to optimize the convergence performance of the model in classification. In view of the fact that there are fewer optional training samples in practical applications, we further propose a method of increasing the number of samples by adding a certain degree of perturbed samples, which can also test the anti-interference ability of classification methods. Furthermore, we analyze the anti-interference and convergence performance of the proposed model in terms of different training sample data sets, different batch training sample numbers and iteration time. In this paper, we describe the experimental process in detail and comprehensively evaluate the proposed model based on the classification of CHRIS hyperspectral imagery covering coastal wetlands, and further evaluate it on a commonly used hyperspectral image benchmark dataset. The experimental results show that the accuracy of the two models after increasing training samples and adjusting the number of batch training samples is improved. When the number of batch training samples is continuously increased to 350, the classification accuracy of the proposed method can still be maintained above 80.7%, which is 2.9% higher than the traditional one. And its time consumption is less than that of the traditional one while ensuring classification accuracy. It can be concluded that the proposed method has anti-interference ability and outperforms the traditional CNN in terms of batch computing adaptability and convergence speed.