Linear vs. nonlinear extreme learning machine for spectral-spatial classification of hyperspectral images

Cao, Faxian and Yang, Zhijing and Ren, Jinchang and Jiang, Mengying and Ling, Wing-Kuen (2017) Linear vs. nonlinear extreme learning machine for spectral-spatial classification of hyperspectral images. Sensors, 17 (11). ISSN 1424-8220 (https://doi.org/10.3390/s17112603)

[thumbnail of Cao-etal-Sensors-2017-Linear-vs-nonlinear-extreme-learning-machine]
Preview
Text. Filename: Cao_etal_Sensors_2017_Linear_vs_nonlinear_extreme_learning_machine.pdf
Final Published Version
License: Creative Commons Attribution 4.0 logo

Download (10MB)| Preview

Abstract

As a new machine learning approach, the extreme learning machine (ELM) has received much attention due to its good performance. However, when directly applied to hyperspectral image (HSI) classification, the recognition rate is low. This is because ELM does not use spatial information, which is very important for HSI classification. In view of this, this paper proposes a new framework for the spectral-spatial classification of HSI by combining ELM with loopy belief propagation (LBP). The original ELM is linear, and the nonlinear ELMs (or Kernel ELMs) are an improvement of linear ELM (LELM). However, based on lots of experiments and much analysis, it is found that the LELM is a better choice than nonlinear ELM for the spectral-spatial classification of HSI. Furthermore, we exploit the marginal probability distribution that uses the whole information in the HSI and learns such a distribution using the LBP. The proposed method not only maintains the fast speed of ELM, but also greatly improves the accuracy of classification. The experimental results in the well-known HSI data sets, Indian Pines, and Pavia University, demonstrate the good performance of the proposed method.