Pixel-Based Classification of Hyperspectral Images Using Convolutional Neural Networks

Syed Aamer Hussain1, Ali Tahir1, Junaid Aziz Khan1, Ahmad Salman2
1Institute of Geographical Information Systems, National University of Sciences and Technology, Islamabad, Pakistan
2School of Electrical Engineering and Computer Science, National University of Sciences and Technology, Islamabad, Pakistan

Tóm tắt

The recent progress in geographical information systems, remote sensing (RS) and data analytics enables us to acquire and process large amount of Earth observation data. Convolutional neural networks (CNN) are being used frequently in classification of multi-dimensional images with high accuracy. In this paper, we test CNNs for the classification of hyperspectral RS data. Our proposed CNN is a multi-layered neural network architecture, which is tailored to classify objects based on pixel-wise spatial information using spectral bands of hyperspectral imagery (HSI). We use benchmark satellite imagery in four different HSI datasets for classification using the proposed architecture. Our results are compared with support vector machine (SVM) and extreme learning machine (ELM) algorithms, which are frequently used techniques of machine learning in RS data classification. Moreover, we also provide a comparison with the state-of-the-art CNN approaches, which have been used for HSI classification. Our results show improvements of up to 6% on average over SVM and ELM while up to 4% improvement is observed in comparison with two recently proposed CNN architectures for HSI classification accuracy. On the other hand, the processing time of our proposed CNN is also significantly lower.

Tài liệu tham khảo

Abraham A (2005) Artificial neural networks. In: Sydenham PH, Thorn R (eds) Handbook of measuring system design. https://doi.org/10.1002/0471497398.mm421

Cao X, Zhou F, Xu L, Meng D, Xu Z, Paisley J (2017) Hyperspectral image segmentation with markov random fields and a convolutional neural network. arXiv:1705.00727

Chetlur S, Woolley C, Vandermersch P, Cohen J, Tran J, Catanzaro B et al (2014) CuDNN: efficient primitives for deep learning. arXiv:1410.0759

LeCun Y, Bengio Y, Hinton G (2015) Deep learning. Nature 521:436–444

Marmanis D, Wegner JD, Galliani S, Schindler K, Datcu M, Stilla U (2016) Semantic segmentation of aerial images with an ensemble of CNSS. ISPRS Ann Photogramm Remote Sens Spat Inf Sci 3:473–480