基于轻量级光谱-空间注意力交互网络的高光谱地物分类研究
DOI:
作者:
作者单位:

(1.信阳职业技术学院 数学与信息工程学院,河南 信阳,464000; 2.贵州大学 大数据与信息工程学院,贵州 贵阳 550025)

作者简介:

周 予 (1988-),男,硕士,讲师,主要从事电子通信、信号处理等方面研究.

通讯作者:

中图分类号:

基金项目:

教育部科技发展中心产学研创新基金——新一代信息技术创新项目(2018A02047)和贵州省科学 技术基金资助项目(黔科合基础-ZK[2021]重点 001)资助项目


Lightweight spectral-spatial attention interactive network for hyperspectral image classification
Author:
Affiliation:

(1.School of Mathematics and Information Engineering, Xinyang Vocational and Technical College, Xinyang,Henan 464000, China;2.College of Big Data and Information Engineering, Guizhou University, Guiyang, Guizhou 550025, China)

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    通过引入基于卷积神经网络(convolutional neural network,CNN)的分类算法,高光谱图像(hyperspectral image,HSI)分类任务的精度取得显著的提升,但目前主流CNN算法往往较为复杂且参数量大,从而导致网络难以训练以及容易产生过拟合问题。为在保证网络分类性能的前提下实现轻量化,本文提出一个轻量级架构的基于光谱-空间注意力交互机制的CNN网络用于HSI分类。为实现HSI的光谱-空间特征提取,构建了一个轻量化的双路径骨干网络用于两种特征的提取和融合。其次,为提高特征的表征能力,设计了两个注意力模块分别用于光谱和空间特征的权重再调整。同时,为加强双路径特征之间的关联以实现特征的更好融合,注意力交互机制被引入到网络中以进一步提升网络性能。在3个真实HSI数据集上的分类结果表明,本文所提网络可达到99.5%的分类准确度,并相比于其他网络至少减少50%的参数量。

    Abstract:

    By introducing the convolutional neural network (CNN) based classification algorithms,the accuracy of hyperspectral image (HSI) classification task has been significantly improved. However,the present mainstream CNN algorithms are usually complex and have a large number of parameters,which will make network hard to train and cause the over-fitting problem.To achieve lightweight on the premise of ensuring the performance of network,we proposed a lightweight CNN network based on spectral-spatial attention interaction mechanism for HSI classification task.To realize the spectral and spatial feature extraction,we constructed a lightweight dual-path backbone network for the extraction and fusion of the two kinds of features.And,to improve the representation ability of features,we designed two attention modules to adjust the weight of spectral and spatial features respectively.At the same time,to strengthen the correlation between two path features to achieve better fusion,attention interaction mechanism was introduced into the network to further improve the network performance. The classification results on three real HSI data sets show that the proposed network can achieve 99.5% classification accuracy and reduce at least 50% parameter amount compared with other networks .

    参考文献
    相似文献
    引证文献
引用本文

周予,程二丽,张娅莉,刘宇红.基于轻量级光谱-空间注意力交互网络的高光谱地物分类研究[J].光电子激光,2023,34(4):397~404

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-04-20
  • 最后修改日期:2022-06-15
  • 录用日期:
  • 在线发布日期: 2023-04-13
  • 出版日期: