数字农科院2.0

基于卷积神经网络与注意力机制的高光谱图像分类

文献类型: 中文期刊

作者: 高玉鹏;闫伟红;潘新

作者机构:

关键词: 高光谱图像分类(HSI);卷积神经网络(CNN);coordinate attention;Transformer

期刊名称:光电子·激光

ISSN: 1005-0086

年卷期: 2024 年 35 卷 05 期

页码:

收录情况: 北大核心(2023版) ; ; CSCD(2023-2024年度)

摘要: 由于浅层卷积神经网络(convolutional neural network, CNN)模型感受野的限制,无法捕获远距离特征,在高光谱图像(hyperspectral image, HSI)分类问题中无法充分利用图像空间-光谱信息,很难获得较高精度的分类结果。针对上述问题,本文提出了一种基于卷积神经网络与注意力机制的模型(model based on convolutional neural network and attention mechanism, CNNAM),该模型利用CA (coordinate attention)对图像通道数据进行位置编码,并利用以自注意力机制为核心架构的Transformer模块对其进行远距离特征提取以解决CNN感受野的限制问题。CNNAM在Indian Pines和Salinas两个数据集上得到的总体分类精度分别为97.63%和99.34%,对比于其他模型,本文提出的模型表现出更好的分类性能。另外,本文以是否结合CA为参考进行了消融实验,并证明了CA在CNNAM中发挥重要作用。实验证明将传统CNN与注意力机制相结合可以在HSI分类问题中获得更高的分类精度。

分类号:

  • 相关文献

[1]基于上下文编码器的图像修复算法. 任鹏博,毛克彪. 2023

[2]几种神经网络经典模型综述. 黄东瑞,毛克彪,郭中华,徐乐园,胡泽民,赵瑞. 2023

[3]Semantic Segmentation of Rice Fields in Sub-Meter Satellite Imagery Using an HRNet-CA-Enhanced DeepLabV3+ Framework. Yifan Shao,Pan Pan,Hongxin Zhao,Jiale Li,Guoping Yu,Guomin Zhou,Jianhua Zhang. 2025

[4]UCIW-YOLO: Multi-category and high-precision obstacle detection model for agricultural machinery in unstructured farmland environments. Gangwei Liu,Chengqian Jin,Youliang Ni,Tengxiang Yang,Zheng Liu. 2025

[5]番石榴实蝇性别决定基因Bcotra和Bcotra-2的克隆、序列特征及表达分析. 武强,和丹阳,张桂芬,李建伟,万方浩. 2015

[6]融合Transformer和LSTM的蓝莓根区土壤含水量预测模型. 王亿,曹姗姗,孙伟,胡博,古丽米拉·克孜尔别克,孔繁涛. 2024

[7]基于RT-WEDT的麦穗检测与计数方法. 李婕,杨子豪,郑权,乔江伟,涂静敏. 2024

[8]FE-TCM: Filter-Enhanced Transformer Click Model for Web Search. Yingfei Wang,Jianping Liu,Jian Wang,Xiaofeng Wang,Meng Wang,Xintao Chu. 2023

[9]A Topicality Relevance-Aware Intent Model for Web Search. Wang, Meng,Liu, Jianping,Wang, Jian,Wang, Yingfei,Chu, Xintao. 2023

[10]Protein-protein interaction and site prediction using transfer learning. Tuoyu Liu,Han Gao,Xiaopu Ren,Guoshun Xu,Bo Liu,Ningfeng Wu,Huiying Luo,Yuan Wang,Tao Tu,Bin Yao,Feifei Guan,Yue Teng,Huoqing Huang,Jian Tian. 2023

[11]TransVAE-DTA: Transformer and variational autoencoder network for drug-target binding affinity prediction. Zhou C.,Li Z.,Song J.,Xiang W.. 2024

[12]Enhanced detection of Aspergillus flavus in peanut kernels using a multi-scale attention transformer (MSAT): Advancements in food safety and contamination analysis. Zhen Guo,Jing Zhang,Haifang Wang,Haowei Dong,Shiling Li,Xijun Shao,Jingcheng Huang,Xiang Yin,Qi Zhang,Yemin Guo,Xia Sun,Ibrahim Darwish. 2024

[13]AECA-FBMamba: A Framework with Adaptive Environment Channel Alignment and Mamba Bridging Semantics and Details. Xin Chai,Wenrong Zhang,Zhaoxin Li,Ning Zhang,Xiujuan Chai. 2025

[14]A novel dual-branch spatial-spectral attention fusion model and method: A case study for the detection of nicotine content in tobacco leaves. Fukang Xing,Rongguang Zhu,Shichang Wang,Lingfeng Meng,Fujia Dong,Songfeng Wang,Jie Ren,Zongxiu Bai,Yapeng Kang. 2025

[15]DeepAT: A Deep Learning Wheat Phenotype Prediction Model Based on Genotype Data. Jiale Li,Zikang He,Guomin Zhou,Shen Yan,Jianhua Zhang. 2024

[16]Neuro-sensory evaluation of citrus flavors: A hierarchical spatial fusion approach for emotion-driven food innovation. Zhao, Qian,Yang, Peilin,Liang, Yushen,Xu, Zhenzhen,Chen, Jianle,Chen, Shiguo,Ye, Xingqian,Cheng, Huan. 2025

[17]Non-destructive viability detection of naturally-aged large-sample rice seeds using hyperspectral imaging and deep learning. Hubo Xu,Fei Li,Ziqiang Wang,Wei Zhang,Hui Li,Xingyu Xia,Chunmei Li,Xia Xin. 2025

[18]A Self-Supervised Pre-Trained Transformer Model for Accurate Genomic Prediction of Swine Phenotypes. Weixi Xiang,Zhaoxin Li,Qixin Sun,Xiujuan Chai,Tan Sun. 2025

[19]Transformer is involved in female sex determination in Bemisia tabaci. Liu, Yating,Wang, Wenlu,Wang, Yina,Zhou, Xuguo,Xie, Wen,Zhang, Youjun. 2025

[20]Spectral Kolmogorov-Arnold Transformer for few-shot rice germplasm viability detection using hyperspectral imaging. Hubo Xu,Ziqiang Wang,Fei Li,Wei Zhang,Han Zhang,Hailiang Zhang,Weiqi Li,Zhaohui Xue,Xia Xin. 2025

作者其他论文 更多>>