A Hybrid Capsule Network for Hyperspectral Image Classification
Limited training data, high dimensionality, image complexity, and similarity between classes are challenges confronting hyperspectral image (HSI) classification often resulting in suboptimal classification performance. The capsule network (CapsNet) preserves the hierarchy between different parts of...
Enregistré dans:
| Auteurs principaux: | , , , |
|---|---|
| Format: | article |
| Langue: | EN |
| Publié: |
IEEE
2021
|
| Sujets: | |
| Accès en ligne: | https://doaj.org/article/56cf8cfa274e417e809e64db5b0f38b9 |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| Résumé: | Limited training data, high dimensionality, image complexity, and similarity between classes are challenges confronting hyperspectral image (HSI) classification often resulting in suboptimal classification performance. The capsule network (CapsNet) preserves the hierarchy between different parts of the entity in an image by replacing scalar representations with vectors and can address these aforementioned issues. Motivated by CapsNet, this article presents a novel end-to-end deep learning (DL) architecture, the hybrid capsule network (HCapsNet), for HSI classification. HCapsNet employs 2-D and 3-D convolutional neural networks (CNNs) to extract higher level spatial and spectral features. In order to establish a route between capsules in the lower layers to the most-related capsule in the higher layer, dynamic routing (DR) is used to identify several overlapped objects during training sessions. Hyperparameter optimization is performed using nested cross-validation (nested-CV) to ensure thorough generalization evaluation. The proposed HCapsNet significantly outperformed the state-of-the-art methods in terms of overall classification accuracy on three widely used hyperspectral datasets, Indian Pines dataset achieving (<inline-formula><tex-math notation="LaTeX">$>3\%$</tex-math></inline-formula>, <inline-formula><tex-math notation="LaTeX">$p< {1}\times 1\times 10^{-11}$</tex-math></inline-formula>), the University of Pavia dataset (<inline-formula><tex-math notation="LaTeX">$>4\%$</tex-math></inline-formula>, <inline-formula><tex-math notation="LaTeX">$p< {1}\times 1\times 10^{-9}$</tex-math></inline-formula>), the Salinas Valley dataset (<inline-formula><tex-math notation="LaTeX">$>3\%$</tex-math></inline-formula>, <inline-formula><tex-math notation="LaTeX">$p< {1}\times 1\times 10^{-10}$</tex-math></inline-formula>) when using only 1% of the data for training. The performance of all CNN-based approaches degraded significantly with smaller training sample sizes. HCapsNet, therefore, is demonstrated to offer significant advantages in HSI classification problems with low sample sizes. |
|---|