基于特征重要性的高光谱图像分类

• 作者: 张因国 陶于祥 罗小波 刘明皓
• 来源期刊: 红外技术
• 学科: 工学
• 关键词: 高光谱图像 特征重要性 波段选择 卷积神经网络 支持向量机
• 年，卷（期）: 2020,（12）
• 所属期刊栏目: 图像处理与仿真
• 页码范围: 1185-1191
• 页数: 7页
• 分类号: TP751
• 语种: 中文

摘要

为了减少高光谱图像中的冗余以及进一步挖掘潜在的分类信息,本文提出了一种基于特征重要性的卷积神经网络(convolutional neural networks,CNN)分类模型.首先,利用贝叶斯优化训练得到的随机森林模型(random forest,RF)对高光谱遥感图像进行特征重要性评估;其次,依据评估结果选择合适数目的高光谱图像波段,以作为新的训练样本;最后,利用三维卷积神经网络对所得样本进行特征提取并分类.基于两个实测的高光谱遥感图像数据,实验结果均表明:相比原始光谱信息直接采用支持向量机(support vector machine,SVM)和卷积神经网络的分类效果,本文所提基于特征重要性的高光谱分类模型能够在降维的同时有效提高高光谱图像的分类精度.