.png){kind=icon}
.png){kind=icon}
Return
Hyperspectral Image Classification Method Based on Data Expansion and Consistency Regularization With Small Samples
.png){kind=icon}
PRE
AI
DOI：10.1109/LGRS.2024.3494552.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
In the hyperspectral image (HSI) classification, convolutional neural networks (CNNs)-based approaches often struggle with the scarcity of labeled samples. The letter proposes an HSI classification method based on data expansion and consistency regularization with small samples. Specifically, we leverage the pixel-pair feature (PPF) to expand the dataset, which facilitates the adequate tuning of CNN parameters and alleviates the issue of overfitting. In addition, a designed CNN structure is employed to extract discriminative features from the limited number of labeled PPFs and numerous unlabeled PPFs. The CNN is trained via minimizing the weighted sum of supervised and unsupervised losses, where the supervised loss is calculated through the cross-entropy function while the unsupervised loss is evaluated with the consistency regularization item. Moreover, reliable references required in the consistency regularization item are provided after making an exponential moving average (EMA) on the outputs of CNNs at different training epochs. Ultimately, we conduct experiments on three real HSI datasets, and the results show that the proposed approach gains superior classification accuracy compared to several existing CNN-based approaches.
Keywords:
Training
Convolutional neural networks
Hyperspectral imaging
Stochastic processes
Vectors
Testing
Support vector machines
Data models
Tuning
Stability analysis
Consistency regularization
convolutional neural network (CNN)
exponential moving average
hyperspectral image (HSI) classification
sample augmentation
Journal
IF:
16.4
Papers:
1.0W
Citations:
5.1K
Organization
View more organizations