| Literature DB >> 29137159 |
Faxian Cao1, Zhijing Yang2, Jinchang Ren3, Mengying Jiang4, Wing-Kuen Ling5.
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.Entities:
Keywords: discriminative random field (DRF); extreme learning machine (ELM); hyperspectral image (HSI); loopy belief propagation (LBP); spectral-spatial classification
Year: 2017 PMID: 29137159 PMCID: PMC5713108 DOI: 10.3390/s17112603
Source DB: PubMed Journal: Sensors (Basel) ISSN: 1424-8220 Impact factor: 3.576
Figure 1Graph example of MRF.
Figure 2Message passing of LBP at t-th iteration.
The training sample and test samples of Indian Pines and Pavia University.
| Indian Pines | Pavia University | |||||||
|---|---|---|---|---|---|---|---|---|
| Class | Train | Test | Class | Train | Test | Class | Train | Test |
| Alfalfa | 6 | 54 | Oats | 2 | 20 | Asphalt | 548 | 6631 |
| Corn-no till | 144 | 1434 | Soybeans-no till | 97 | 968 | Meadows | 548 | 18,649 |
| Corn-min till | 84 | 834 | Soybeans-min till | 247 | 2468 | Gravel | 392 | 2099 |
| Corn | 24 | 234 | Soybeans-clean till | 62 | 614 | Trees | 524 | 3064 |
| Grass/pasture | 50 | 497 | Wheat | 22 | 212 | Metal sheets | 265 | 1345 |
| Grass/tree | 75 | 747 | Woods | 130 | 1294 | Bare soil | 532 | 5029 |
| Grass/pasture-mowed | 3 | 26 | Bldg-grass-tree-drives | 38 | 380 | Bitumen | 375 | 1330 |
| Hay-windrowed | 49 | 489 | Stone-steel towers | 10 | 95 | Bricks | 514 | 3682 |
| Total | 1043 | 10366 | Shadows | 231 | 947 | |||
| Total | 3921 | 42,776 | ||||||
Figure 3The impact of hidden neurons of ELM in the datasets: (a) Indian Pines; (b) Pavia University.
Figure 4The impact of sparseness parameters in the datasets: (a) Indian Pines; (b) Pavia University.
Figure 5AVIRIS Indian Pines training maps.
Figure 6The overall accuracy of Indian Pines image: (a) SMLR (OA = 75.76%); (b) KSMLR (OA = 84.34%); (c) NLELM (OA = 86.93%); (d) LELM (OA = 79.43%); (e) SMLR-LBP (OA = 98.26%); (f) KSMLR-LBP (OA = 99.05%); (g) NLELM-LBP (OA = 87.95%); (h) Proposed method (OA = 99.75%).
Indian Pines: overall, average, and individual class accuracy (in percent) and k statistic of different classification methods with 10% training samples. The best accuracy in each row is shown in bold.
| Class | SMLR | KSMLR | LELM | NLELM | SMLR-LBP | KSMLR-LBP | NLELM-LBP | PROPOSED METHOD |
|---|---|---|---|---|---|---|---|---|
| Alfalfa | 30.52 | 74.26 | 35.37 | 71.11 | 97.78 | 100 | 90.37 | |
| Corn-no till | 75.87 | 82.49 | 79.27 | 85.82 | 99.02 | 99.40 | 85.68 | |
| Corn-min till | 51.35 | 70.86 | 58.26 | 72.58 | 92.55 | 97.35 | 68.79 | |
| Corn | 37.35 | 68.68 | 43.29 | 69.10 | 99.27 | 95.00 | 77.44 | |
| Grass/pasture | 86.82 | 89.46 | 89.76 | 93.64 | 97.36 | 98.23 | 93.64 | |
| Grass/tree | 94.28 | 96.37 | 96.32 | 97.39 | 95.70 | |||
| Grass/pasture-mowed | 6.92 | 45.00 | 11.54 | 70.38 | 71.92 | 91.54 | 45.00 | |
| Hay-windrowed | 99.37 | 98.51 | 99.57 | 99.04 | 98.73 | |||
| Oats | 5 | 38.50 | 11.50 | 63.50 | 16.50 | 48.00 | ||
| Soybeans-no till | 61.03 | 74.91 | 66.69 | 80.79 | 96.27 | 96.34 | 80.74 | |
| Soybeans-min till | 74.46 | 84.51 | 80.23 | 87.66 | 99.91 | 90.41 | 99.93 | |
| Soybeans-clean till | 68.96 | 82.20 | 72.98 | 84.98 | 98.50 | 82.85 | ||
| Wheat | 96.75 | 99.15 | 99.39 | 98.96 | 98.77 | |||
| Woods | 95.04 | 95.20 | 95.65 | 96.51 | 99.69 | 97.26 | ||
| Bldg-grass-tree-drives | 67.13 | 73.05 | 64.08 | 70.45 | 95.47 | 99.50 | 83.53 | |
| Stone-steel towers | 69.26 | 70.32 | 70.42 | 77.05 | 99.58 | 98.63 | 98.63 | |
| OA | 75.76 | 84.34 | 79.43 | 86.93 | 98.26 | 99.05 | 87.95 | |
| AA | 63.66 | 77.72 | 67.15 | 82.44 | 91.51 | 98.47 | 83.47 | |
| k | 72.22 | 82.09 | 76.38 | 85.06 | 98.02 | 98.92 | 86.36 | |
| Execution Time (seconds) | 0.02 | 0.41 | 0.19 | 0.31 | 38.74 | 40.70 | 39.59 | 38.95 |
Figure 7Pavia University training maps.
Figure 8The overall accuracy of Pavia University image: (a) SMLR (OA = 78.78%); (b) KSMLR (OA = 93.00%); (c) NLELM (OA = 93.94%); (d) LELM (OA = 91.23%); (e) SMLR-LBP (OA = 95.68%); (f) KSMLR-LBP (OA = 99.42%); (g) NLELM-LBP (OA = 99.61%); (h) Proposed method (OA = 99.82%).
Pavia University: overall, average, and individual class accuracy (in percent) and k statistic of different classification methods with 10% training samples. The best accuracy in each row is shown in bold.
| Class | SMLR | KSMLR | LELM | NLELM | SMLR-LBP | KSMLR-LBP | NLELM-LBP | PROPOSED METHOD |
|---|---|---|---|---|---|---|---|---|
| Asphalt | 72.27 | 89.43 | 85.27 | 88.82 | 98.62 | 99.49 | ||
| Meadows | 79.08 | 94.16 | 92.17 | 94.61 | 93.70 | 99.34 | 99.88 | |
| Gravel | 71.99 | 85.08 | 78.06 | 87.41 | 99.14 | 99.64 | 99.92 | |
| Trees | 94.90 | 97.92 | 97.38 | 98.16 | 99.27 | 98.54 | 99.64 | |
| Metal sheets | 99.58 | 99.34 | 98.85 | 99.39 | 100.00 | |||
| Bare soil | 74.26 | 94.77 | 93.90 | 95.43 | 99.93 | 100.00 | ||
| Bitumen | 78.66 | 93.82 | 93.69 | 95.34 | 100.00 | |||
| Bricks | 73.37 | 87.52 | 90.05 | 90.94 | 99.93 | 99.63 | 99.85 | |
| Shadows | 96.88 | 99.61 | 99.70 | 99.97 | 99.87 | 94.14 | ||
| OA | 78.78 | 93.00 | 91.23 | 93.94 | 96.93 | 99.59 | 99.62 | |
| AA | 82.33 | 93.49 | 92.12 | 94.56 | 98.94 | 99.77 | 99.09 | |
| k | 72.73 | 90.82 | 88.54 | 92.04 | 95.98 | 99.46 | 99.49 | |
| Execution Time (seconds) | 0.19 | 4.40 | 0.48 | 3.83 | 1237.1 | 5288.6 | 1201.2 |
The classification results of the proposed method and other methods. The best accuracy in each row is shown in bold.
| Datasets | Index | EMP-ELM | S-ELM | G-ELM | PROPOSED METHOD |
|---|---|---|---|---|---|
| Indian Pines data set with 10% training samples | OA | - | 97.78 | 99.08 | |
| AA | - | 97.10 | 98.68 | ||
| k | - | 97 | 98.95 | ||
| Pavia University data set with 9% training samples | OA | 99.65 | - | - | |
| AA | 99.60 | - | - | ||
| k | 99.52 | - | - |