Multi-layer feature fusion for hyperspectral image classification
Siyuan Hao,Rui Li,Min Zhang,Yuanxin Ye,Siyuan HaoRui LiMin ZhangYuanxin Yea College of Information and Control Engineering,Qingdao University of Technology,Qingdao,Chinab Faculty of Geosciences and Environmental Engineering,Southwest Jiaotong University,Chengdu,China
DOI: https://doi.org/10.1080/01431161.2023.2270109
IF: 3.531
2023-11-04
International Journal of Remote Sensing
Abstract:Convolutional Neural Network (CNN) has been widely used in hyperspectral image (HSI) classification. It has obvious advantages in extracting spatial and spectral features. However, the classification performance depends not only on spatial and spectral information, but also on the sequence properties between spectral features. Transformer is an excellent model for establishing sequence relationship, however making spatial correlation loss. On the basis of preserving spatial correlation as much as possible, we propose a multi-layer feature fusion module (MLFF) to fully mine and fuse the representative spectral and spatial features and sequence relationship. The proposed MLFF is composed of Feature Transformation, Tensor-based Sobel and Transformer Fusion. Feature Transformation is implemented by transpose and convolution operations, which can get the high-level contextual features and increase the diversity of feature information. At the same time, we designed a Tensor-based Sobel module to better participate in the subsequent fusion of HSI edge and texture features. Finally, in the Transformer Fusion module, these features are fused by one feature per layer through the improved encoder and decoder. Besides, we add skip structure to maintain a good relationship between each layer and strengthen cross-layer connections. Through the above three modules, we can better acquire and fuse HSI spectral, spatial and edge features to achieve good classification accuracy. The experiments conducted on three public HSI datasets demonstrate that the proposed method has better performance compared with existing methods.
imaging science & photographic technology,remote sensing