Hyperspectral image classification based on manifold spectral dimensionality reduction and deep learning method

被引:0
|
作者
Shi Y. [1 ]
Ma D. [1 ]
Lyu J. [1 ]
Li J. [1 ]
Shi J. [1 ]
机构
[1] College of Geomatics, Xi'an University of Science and Technology, Xi'an
来源
Lyu, Jie (rsxust@163.com) | 1600年 / Chinese Society of Agricultural Engineering卷 / 36期
关键词
Convolutional neural network; Dimension reduction; Hyperspectral; Image classification; Machine vision; Manifold learning;
D O I
10.11975/j.issn.1002-6819.2020.06.018
中图分类号
学科分类号
摘要
Hyperspectral remote sensing image is rich in spectral information which has great application potential in forestry, agriculture, geosciences, and other fields. In order to solve the problem of the small sample, multi-dimension, correlation and nonlinearity, and to improve the accuracy of hyperspectral remote sensing image classification, this study proposed a method of hyperspectral image dimension reduction based on t-distribution stochastic neighbor embedding (t-SNE). Convolutional neural network (CNN) was used to extract features and to classify hyperspectral remote sensing images. The t-SNE used the t distribution instead of the Gaussian distribution and defined a symmetric joint probability distribution based on the original SNE, thus it could simplify the gradient calculation. T-distribution was more sensitive to local features because of its long tail character. Using t-distribution instead of Gaussian distribution ensured that the points mapped from high-dimensional space to low-dimensional space were almost unaffected by spatial changes. It was feasible to make intra-class points aggregated closely, and inter-class points dispersed. Meanwhile, it could use the local features of high-dimensional data and maintain the non-linear features of the original data set. To improve the accuracy of hyperspectral remote sensing classification, a novel method based on manifold learning and CNN was proposed. First, the data points in the original high-dimensional space were mapped into the low-dimensional space. The dimensional reduction scale was important for classification results. In order to find the best dimensional reduction scale, an experiment with dimensions ranging from 5 to 30 was conducted. The scale of the Indian Pines dataset was set at 20, the Pavia Center dataset was set at 16 and the Pavia University dataset was set at 18. Perplexity was another important parameter and it had been set at 30 according to the test. Their topological relations were preserved after dimensional reduction. Second, a CNN with a seven layers network structure was designed. It consisted of two convolution layers, two pooling layers, two full connection layers, and one full connection layer. Two convolution layers and two pooling layers existed alternately, and the end of the network related to a full connection layer. A Softmax function was used as a classifier and the AdaGrad algorithm was used for network optimization. With the progress of the optimization process, the learning rate would be reduced for the variables that had decreased a lot. Rectified linear unit (ReLU) has been used as an activation function. The advantages of the ReLU function are more efficient in gradient descent and backpropagation because it avoids the problem of gradient explosion and gradient disappearance and it simplifies the calculation process and reduces the overall calculation cost of CNN. The hyperspectral remote sensing data after dimension reduction was used as the input layer to extract the deep features on CNN. Finally, the spatial-spectral features of hyperspectral images were classified. The robustness of the proposed algorithm was verified in three open datasets; (i) Indian Pines, (ii) Pavia Center and (iii) Pavia University. The overall accuracy of classification in three data sets had reached 99.05%, 99.43%, and 98.90%. The proposed algorithm showed a better effect on dimension reduction compared with the original CNN. Since t-SNE was more sensitive to local features and considered inter-class differences, remarkable results had been achieved for small ground object samples. Compared with the original CNN, the problem of "salt and pepper noise" in the hyperspectral image was solved effectively and the overall classification accuracy was significantly improved. The method of manifold learning and convolutional neural networks could also provide a new approach for the hyperspectral remote sensing image classification.It was usually difficult to obtain the labeled sample data of the hyperspectral image, while the performance of the deep learning model depended on many mark samples. In future work, we would consider how to construct the classification model under the condition of limited labeled samples to obtain better classification results. © 2020, Editorial Department of the Transactions of the Chinese Society of Agricultural Engineering. All right reserved.
引用
收藏
页码:151 / 160
页数:9
相关论文
共 28 条
  • [1] Sun Y., Zhou Y., Yuan M., Et al., UAV real-time monitoring for forest pest based on deep learning, Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 34, 21, pp. 74-81, (2018)
  • [2] Yang Y., Zhao Y., Qin K., Et al., Prediction of black soil nutrient content based on airborne hyperspectral remote sensing, Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 35, 20, pp. 94-101, (2019)
  • [3] Lin H., Zhang X., Yang Y., Et al., Retrieval of the mineral abundance and particle size distribution at the landing site of Yutu rover with hyperspectral remote sensing data, Journal of Remote Sensing, 23, 5, pp. 831-840, (2017)
  • [4] Lu Y., Study on Manifold Learning Algorithms of Hyperspectral Image, (2016)
  • [5] Cortes C., Vapnik V., Support-vector networks, Machine Learning, 20, 3, pp. 273-297, (1995)
  • [6] Leo B., Jerome H.F., Richard A.O., Et al., Classification and regression trees, European Journal of Operational Research, 19, 1, pp. 144-160, (1984)
  • [7] Macqueen J.B., Some methods for classification and analysis of multi-variate observations, Proceedings of Berkeley Symposium on Mathematical Statistics & Probability, pp. 281-297, (1965)
  • [8] Dartnall H.J.A., Lythgoe J.N., The spectral clustering of visual pigments, Vision Research, 5, 3, pp. 81-100, (1965)
  • [9] De Morsier F., Borgeaud M., Gass V., Et al., Kernel low-rank and sparse graph for unsupervised and semi-supervised classification of hyperspectral images, IEEE Transactions on Geoscience and Remote Sensing, 54, 6, pp. 1-11, (2016)
  • [10] Zhang P., Hu S., Fine crop classification by remote sensing in complex planting areas based on field parcel, Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 35, 20, pp. 125-134, (2019)