期刊
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS
卷 19, 期 -, 页码 -出版社
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/LGRS.2020.3026587
关键词
Convolution; Hyperspectral imaging; Training; Feature extraction; Principal component analysis; Cross-attention mechanism (CAM); deep learning; graph convolution
类别
资金
- Hunan Key Laboratory of Intelligent Logistics Technology [2019TP1015]
In this letter, a novel cross-attention mechanism and graph convolution integration algorithm are proposed for hyperspectral data classification. Experimental results show that the proposed algorithm achieves better performances than other algorithms using different methods of training set division.
An attention mechanism assigns different weights to different features to help a model select the features most valuable for accurate classification. However, the traditional attention mechanism algorithm often allocates weights in a one-way fashion, which can result in a loss of feature information. To obtain better hyperspectral data classification results, a novel cross-attention mechanism and graph convolution integration algorithm are proposed in this letter. First, principal component analysis is used to reduce the dimensionality of hyperspectral images to obtain low-dimensional features that are more expressive. Second, the model uses a cross (horizontal and vertical directions) attention algorithm to allocate weights jointly based on its two strategies; then, it adopts a graph convolution algorithm to generate the directional relationships between the features. Finally, the generated deep features and the relationship between the deep features are used to complete the prediction of hyperspectral data. Experiments on three well-known hyperspectral data setsx2014;Indian Pines, the University of Pavia, and Salinasx2014;show that the proposed algorithm achieves better performances than do other well-known algorithms using different methods of training set division.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据