电子学报2024,Vol.52Issue(10) :3541-3551.DOI:10.12263/DZXB.20240477

基于轻量级全连接张量映射网络的高光谱图像分类方法

Lightweight Fully-Connected Tensorial Mapping Network for Hyper-spectral Image Classification

林知心 郑玉棒 马天宇 王蕊 李恒超
电子学报2024,Vol.52Issue(10) :3541-3551.DOI:10.12263/DZXB.20240477

基于轻量级全连接张量映射网络的高光谱图像分类方法

Lightweight Fully-Connected Tensorial Mapping Network for Hyper-spectral Image Classification

林知心 1郑玉棒 1马天宇 1王蕊 1李恒超1
扫码查看

作者信息

  • 1. 西南交通大学信息科学与技术学院,四川 成都 611756
  • 折叠

摘要

近年来,基于卷积神经网络的深度学习模型已经在高光谱图像分类领域取得优异表现.然而,模型性能的提升通常依赖于更深、更宽的网络结构,导致参数量和计算量增长,从而限制了模型在机载或星载载荷中的实际部署.为此,本文提出基于轻量级全连接张量映射网络的高光谱图像分类方法.根据全连接张量网络分解的映射思想以及高光谱图像"图谱合一"的结构特点,本文设计两种张量映射卷积单元,通过使用多个具有全连接结构的小尺寸卷积核代替原始卷积核,降低了卷积层的时间和空间复杂度.此外,基于新单元构建残差双分支张量模块.双分支结构共享同一组权重参数,并采用通道分割操作减少特征通道数,提升特征提取过程的实时性.本文所提模型通过使用新单元和新模块充分挖掘高光谱图像的局部空谱信息和全局光谱信息,有效提高了分类性能并减少硬件资源消耗.在三个常用高光谱图像数据集上的实验结果表明,所提模型相较于其他现有工作具有更高的分类性能以及更低的参数量和计算量.

Abstract

In recent years,convolutional neural networks have demonstrated outstanding performance in HSIC(Hy-perspectral Image Classification).However,the improvement of model performance involves adopting deeper and broader network architectures,leading to an increased number of parameters and operations,thus hindering deployment in airborne or on-board devices.To this end,this paper introduces a HSIC method based on the LiteFCTMN(Lightweight Fully-Con-nected Tensorial Mapping Network).We design two convolutional units based on the mapping way of FCTN(Fully-Con-nected Tensor Network)decomposition and the structural characteristics of HSIs.By mapping the original convolution ker-nel to multiple small-sized convolution kernels with fully-connected structures,the complexity of the novel units is reduced while their expressiveness is improved.In addition,the RDT(Residual Double-Branch Tensorial)module is constructed us-ing the designed units.In this module,two branches share the same weights,and a channel split operation is employed to re-duce the number of feature channels,thereby reducing complexity.The proposed model strategically leverages both local spatial-spectral information from RDT and global spectral information from the new units,resulting in enhanced classifica-tion performance and reduced hardware consumption.Experimental results on three widely used HSI datasets demonstrate that the proposed model achieves superior classification performance and lower complexity compared to the state-of-the-art works.

关键词

高光谱图像分类/模型压缩/全连接张量网络分解/卷积神经网络/张量神经网络/轻量卷积模块

Key words

hyperspectral image classification/model compression/fully-connected tensor network decomposition/convolutional neural network/tensorial neural network/lightweight convolutional module

引用本文复制引用

基金项目

国家自然科学基金(62271418)

出版年

2024
电子学报
中国电子学会

电子学报

CSTPCDCSCD北大核心
影响因子:1.237
ISSN:0372-2112
参考文献量27
段落导航相关论文