Orthogonal Capsule Network with Meta-Reinforcement Learning for Small Sample Hyperspectral Image Classification
Most current hyperspectral image classification (HSIC) models require a large number of training samples, and when the sample size is small, the classification performance decreases. To address this issue, we propose an innovative model that combines an orthogonal capsule network with meta-reinforce...
Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2025-01-01
|
| Series: | Remote Sensing |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2072-4292/17/2/215 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| Summary: | Most current hyperspectral image classification (HSIC) models require a large number of training samples, and when the sample size is small, the classification performance decreases. To address this issue, we propose an innovative model that combines an orthogonal capsule network with meta-reinforcement learning (OCN-MRL) for small sample HSIC. The OCN-MRL framework employs Meta-RL for feature selection and CapsNet for classification with a small data sample. The Meta-RL module through clustering, augmentation, and multiview techniques enables the model to adapt to new HSIC tasks with limited samples. Learning a meta-policy with a Q-learner generalizes across different tasks to effectively select discriminative features from the hyperspectral data. Integrating orthogonality into CapsNet reduces the network complexity while maintaining the ability to preserve spatial hierarchies and relationships in the data with a 3D convolution layer, suitably capturing complex patterns. Experimental results on four rich Chinese hyperspectral datasets demonstrate the OCN-MRL model’s competitiveness in both higher classification accuracy and less computational cost compared to existing CapsNet-based methods. |
|---|---|
| ISSN: | 2072-4292 |