Hyperspectral Image Classification With Spectral and Spatial Graph Using Inductive Representation Learning Network

Convolutional neural networks (CNN) have achieved excellent performance for the hyperspectral image (HSI) classification problem due to better extracting spectral and spatial information. However, CNN can only perform convolution calculations on Euclidean datasets. To solve this problem, recently, t...

Full description

Bibliographic Details
Main Authors: Pan Yang, Lei Tong, Bin Qian, Zheng Gao, Jing Yu, Chuangbai Xiao
Format: Article
Language:English
Published: IEEE 2021-01-01
Series:IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing
Subjects:
Online Access:https://ieeexplore.ieee.org/document/9286488/
Description
Summary:Convolutional neural networks (CNN) have achieved excellent performance for the hyperspectral image (HSI) classification problem due to better extracting spectral and spatial information. However, CNN can only perform convolution calculations on Euclidean datasets. To solve this problem, recently, the graph convolutional neural network (GCN) is proposed, which can be applied to the semisupervised HSI classification problem. However, the GCN is a direct push learning method, which requires all nodes to participate in the training process to get the node embedding. This may bring great computational cost for the hyperspectral data with a large number of pixels. Therefore, in this article, we propose an inductive learning method to solve the problem. It constructs the graph by sampling and aggregating (GraphSAGE) feature from a node's local neighborhood. This could greatly reduce the space complexity. Moreover, to enhance the classification performance, we also construct the graph using spectral and spatial information (spectra-spatial GraphSAGE). Experiments on several hyperspectral image datasets show that the proposed method can achieve better classification performance compared with state-of-the-art HSI classification methods.
ISSN:2151-1535