Command Palette
Search for a command to run...
Shorten Spatial-spectral RNN with Parallel-GRU for Hyperspectral Image Classification
Haowen Luo

Abstract
Convolutional neural networks (CNNs) attained a good performance in hyperspectral sensing image (HSI) classification, but CNNs consider spectra as orderless vectors. Therefore, considering the spectra as sequences, recurrent neural networks (RNNs) have been applied in HSI classification, for RNNs is skilled at dealing with sequential data. However, for a long-sequence task, RNNs is difficult for training and not as effective as we expected. Besides, spatial contextual features are not considered in RNNs. In this study, we propose a Shorten Spatial-spectral RNN with Parallel-GRU (St-SS-pGRU) for HSI classification. A shorten RNN is more efficient and easier for training than band-by-band RNN. By combining converlusion layer, the St-SSpGRU model considers not only spectral but also spatial feature, which results in a better performance. An architecture named parallel-GRU is also proposed and applied in St-SS-pGRU. With this architecture, the model gets a better performance and is more robust.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| hyperspectral-image-classification-on-indian | St-SS-pGRU | Overall Accuracy: 90.35% |
| hyperspectral-image-classification-on-pavia | St-SS-pGRU | Overall Accuracy: 98.44% |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.