首页 | 官方网站   微博 | 高级检索  
     

基于特征重要性的高光谱图像分类
引用本文:张因国,陶于祥,罗小波,刘明皓.基于特征重要性的高光谱图像分类[J].红外技术,2020,42(12):1185-1191.
作者姓名:张因国  陶于祥  罗小波  刘明皓
作者单位:1.重庆邮电大学 计算机科学与技术学院,重庆 400065
基金项目:重庆市应用开发计划重点项目;国家自然科学基金
摘    要:为了减少高光谱图像中的冗余以及进一步挖掘潜在的分类信息,本文提出了一种基于特征重要性的卷积神经网络(convolutional neural networks,CNN)分类模型。首先,利用贝叶斯优化训练得到的随机森林模型(random forest,RF)对高光谱遥感图像进行特征重要性评估;其次,依据评估结果选择合适数目的高光谱图像波段,以作为新的训练样本;最后,利用三维卷积神经网络对所得样本进行特征提取并分类。基于两个实测的高光谱遥感图像数据,实验结果均表明:相比原始光谱信息直接采用支持向量机(support vector machine,SVM)和卷积神经网络的分类效果,本文所提基于特征重要性的高光谱分类模型能够在降维的同时有效提高高光谱图像的分类精度。

关 键 词:高光谱图像    特征重要性    波段选择    卷积神经网络    支持向量机
收稿时间:2020-07-21

Hyperspectral Image Classification Based on Feature Importance
Affiliation:1.College of Computer Science and Technology, Chongqing University of Posts and Telecommunications, Chongqing 400065, China2.Chongqing Institute of Meteorological Science, Chongqing 401147, China
Abstract:To reduce the redundancy in hyperspectral images and further explore their potential classification information, a convolutional neural network(CNN) classification model based on feature importance is proposed. First, the random forest(RF) model obtained by Bayesian optimization training is used to evaluate the importance of hyperspectral images. Second, an appropriate number of hyperspectral image bands are selected as new training samples according to the evaluation results. Finally, the 3D-CNN is used to extract and classify the obtained samples. Based on two sets of measured hyperspectral remote sensing image data, the experimental results demonstrate the following: compared with the original spectral information obtained directly using a support vector machine(SVM) and the CNN classification effect, the proposed hyperspectral classification model based on feature importance can effectively improve the classification accuracy of hyperspectral images while reducing dimensionality.
Keywords:
本文献已被 万方数据 等数据库收录!
点击此处可从《红外技术》浏览原始摘要信息
点击此处可从《红外技术》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司    京ICP备09084417号-23

京公网安备 11010802026262号