农业机械学报2024，Vol.55Issue(3)：183-192,212,11.DOI:10.6041/j.issn.1000-1298.2024.03.018
基于多注意力机制与编译图神经网络的高光谱图像分类
Hyperspectral Image Classification Based on Multi-attention Mechanism and Compiled Graph Neural Networks
摘要
Abstract
In recent years,although some scholars have achieved satisfactory research results on hyperspectral image(HSI)classification,they often fail to achieve ideal classification results when facing small sample learning.Aiming at this problem,a hyperspectral image classification method was proposed by the organic combination of multi-attention mechanism fusion,compiled graph neural network and convolutional neural network.Firstly,a type of multiple mixed attention convolutional neural network(MCNN)and compiled graph neural network(CGNN)was designed,which can effectively retain the spectral and spatial information of HSI with limited learning samples;secondly,the introduced graph encoder and graph decoder can effectively map irregular HSI feature information;finally,the designed multi-attention mechanism can focus on some important HSI feature categories.In addition,the effect of different training samples on different algorithms for learning example classification was also investigated.Experiments on the public dataset Botswana(BS)showed that the proposed method improved the overall classification accuracy(OA)by 2.72 percentage points and 3.86 percentage points compared with the current state-of-the-art algorithms(CNN-enhanced graph convolutional network,CEGCN;weighted feature fusion of convolutional neural network,WFCG).Similarly,the experimental results on the IndianPines(IP)dataset with only 3％of the training sample data showed that the method also improved the OA of the current state-of-the-art algorithms(CEGCN and WFCG)by 0.44 percentage points and 1.42 percentage points,respectively.This demonstrated that the proposed method not only had good spatial and spectral information perception for HSI,but also showed strong classification accuracy with small learning data.关键词
高光谱图像分类/图神经网络/注意力机制/超像素分割Key words
hyperspectral image classification/graph neural network/attention mechanism/superpixel segmentation分类
信息技术与安全科学引用本文复制引用
孙杰,杨静,丁书杰,李少波,胡建军..基于多注意力机制与编译图神经网络的高光谱图像分类[J].农业机械学报,2024,55(3):183-192,212,11.基金项目
国家自然科学基金项目(62166005)、国家重点研发计划项目(2018AAA0101800)、贵州省科技支撑计划项目(QKH[2022]130、QKH[2022]003、QKH[2021]335)和贵阳市科技人才培养对象及培养项目(ZKHT[2023]48-8) （62166005）
