摘要
为了在有限训练样本情况下充分提取高光谱影像的空间光谱特征,提高分类精度,提出一种结合空洞卷积和密集网络的高光谱影像分类方法。首先,构建多尺度空洞特征提取模块,引入不同数量的空洞卷积层和普通卷积层通过级联的方式增大模型的感受野,并提取多尺度特征。然后,在多尺度空洞特征提取模块之间建立密集连接,实现特征复用的同时缓解梯度消失问题,而模块内部无密集连接,避免构建深度网络而导致网络参数过多的问题。最后,将得到的特征依次通过池化层,全连接层和Softmax层完成分类。另外,本文在全连接层后加入dropout正则化防止出现过拟合。在Indian Pines和WHU-Hi-Longkou数据集上与经典分类方法进行对比,本文方法 OA分别为98.75%和98.82%。实验结果表明,本文设计的网络模型在有限训练样本情况下,分类效果最优。
In order to fully extract the spatial-spectral features of hyperspectral image with limited training samples and improve classification accuracy,a hyperspectral image classification method combining dilated convolution and dense network is proposed.Firstly,a multi-scale dilated feature extraction module is constructed by introducing different numbers of dilated convolutional layers and ordinary convolutional layers to increase the receptive field of model through cascading and extract multi-scale features.Then,the dense connections are established between multi-scale dilated feature extraction modules to achieve feature reuse while alleviating the problem of gradient vanishing.However,there are no dense connections within the modules to avoid the problem of building a deep network with excessive network parameters.Finally,the obtained features are sequentially classified through pooling layers,fully connected layers,and Softmax layers.In addition,this study adds the dropout regularization after the fully connected layer to prevent overfitting.Compared with classical classification methods on the Indian Pines and WHU-Hi-Longkou datasets,our method provides an OA of 98.75%and 98.82%,respectively.The experimental results show that the network model designed in this study provides the best classification performance at the limited sample conditions.
作者
涂潮
刘万军
赵琳琳
曲海成
Tu Chao;Liu Wanjun;Zhao Linlin;Qu Haicheng(School of Geomatics,Liaoning Technical University,Fuxin 123000,China;School of Software,Liaoning Technical University,Huludao 125105,China)
出处
《仪器仪表学报》
EI
CAS
CSCD
北大核心
2024年第4期206-216,共11页
Chinese Journal of Scientific Instrument
基金
国家自然科学基金面上项目(42271409)资助。
关键词
高光谱影像
多尺度特征融合
空洞卷积
密集网络
hyperspectral image
multi-scale feature fusion
dilated convolution
dense network
作者简介
通信作者:涂潮,2018年于无锡太湖学院获得学士学位,2021年于辽宁工程技术大学获得硕士学位,现为辽宁工程技术大学博士生,主要研究方向高光谱影像分类、图像处理和计算机视觉。E-mail:745700558@qq.com;刘万军,1982年于燕山大学获得学士学位,1991年于辽宁工程技术大学获得硕士学位,现为辽宁工程技术大学教授,博士生导师,主要研究方向计算机视觉、模式识别与人工智能。E-mail:2149541390@qq.com。
