Hyperspectral Image Classification Based on A Multi-Scale Weighted Kernel Network

SUN Le,XU Bin,LU Zhenyu
DOI: https://doi.org/10.1049/cje.2021.00.130
IF: 1.019
2022-01-01
Chinese Journal of Electronics
Abstract:Recently,many deep learning models have shown excellent performance in hyperspectral image(HSI)classification.Among them,networks with mul-tiple convolution kernels of different sizes have been proved to achieve richer receptive fields and extract more representative features than those with a single convolu-tion kernel.However,in most networks,different-sized convolution kernels are usually used directly on multi-branch structures,and the image features extracted from them are fused directly and simply.In this paper,to fully and adaptively explore the multiscale information in both spectral and spatial domains of HSI,a novel multi-scale weighted kernel network(MSWKNet)based on an adapt-ive receptive field is proposed.First,the original HSI cu-bic patches are transformed to the input features by com-bining the principal component analysis and one-dimen-sional spectral convolution.Then,a three-branch net-work with different convolution kernels is designed to convolve the input features,and adaptively adjust the size of the receptive field through the attention mechanism of each branch.Finally,the features extracted from each branch are fused together for the task of classification.Experiments on three well-known hyperspectral data sets show that MSWKNet outperforms many deep learning networks in HSI classification.
What problem does this paper attempt to address?