arrow
返回

Cross-Dataset Model Training for Hyperspectral Image Classification Using Self-Supervised Learning

delete2024-01-01
delete1
PRE
AI
白
白静 (Jing Bai) *
Z
Zichen Zhou
郑
郑晨 (Zheng Chen)
Z
Zhu Xiao *
E
Erlong Wei
Y
Yihong Wen
焦李成 封面图
焦李成 (Licheng Jiao)
DOI:10.1109/TGRS.2024.3493969delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
With the development of deep learning and the increase in the amount of data, general artificial intelligence models have become a popular research area nowadays. When facing a new application scenario, a pretraining general model can often show better performance than models trained with new data on its own. However, because of the specificity of the differences in hyperspectral image data bands, the current hyperspectral image classification (HSIC) field has not proposed a better general model training solution, and it is difficult to utilize the information of the existing hyperspectral datasets for model training in the face of a new scenario. In order to solve this problem, this article proposes a generalized hyperspectral classification model training method, which effectively completes the training of hyperspectral classification models across datasets by adaptive channel module and masked self-supervised pretraining method, and can pretrain and fine-tune hyperspectral classification models using multiple datasets. The adaptive channel module is able to solve the band difference problem of using hyperspectral datasets across datasets, and the masked self-supervised learning method solves the label difference and labeling difficulties of training models across datasets. Experimental results on multiple datasets show that the method proposed in this article can effectively use a large amount of data to complete the pretraining of hyperspectral classification models, and the fine-tuning results on downstream datasets have certain advantages relative to current advanced deep learning methods.
Keyword:
Hyperspectral imaging
Training
Data models
Feature extraction
Image classification
Adaptation models
Transformers
Self-supervised learning
Supervised learning
Deep learning
Classification
cross dataset
general model
hyperspectral image
masked autoencoder
self-supervised learning

期刊

IEEE Transactions on Geoscience and Remote Sensing 封面图
IEEE Transactions on Geoscience and Remote Sensing
IF:
8.6
论文数:
2.2W
被引数:
10.7W

机构

H
hunan university
学者数:
4.5W
论文数: 3.3W
被引数: 70
M
ministry of education - china
学者数:
2.5W
论文数: 1.0W
被引数: 13
X
Xidian University
学者数:
2.4W
论文数: 1.9W
被引数: 9.7K
学者 查看更多机构
引用论文

引用论文

暂无论文信息