Lightweight Tensorized Neural Networks for Hyperspectral Image Classification
Tian-Yu Ma,Heng-Chao Li,Rui Wang,Qian Du,Xiuping Jia,Antonio Plaza
DOI: https://doi.org/10.1109/tgrs.2022.3225438
IF: 8.2
2022-12-17
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Deep learning methods have demonstrated excellent performance in hyperspectral image (HSI) classification. However, these methods mainly focus on improving the classification accuracy while ignoring their high complexity. By considering that the data formats of both HSIs and network weights can be represented in the form of tensors, we develop a new lightweight tensorized neural network (TNN) for HSI classification that takes advantage of low-rank tensor decomposition techniques to reduce complexity. First, inspired by tensor train (TT)-based tensorized convolutional layers, a new tensorized 2D convolutional layer based on chain calculation (with better expression ability) is introduced. Based on this innovation, a new lightweight 2D-TNN is designed for HSI classification. Furthermore, to better preserve the intrinsic structure of HSI data, a new lightweight 3D-TNN is proposed by extending the tensorized 2D convolutional layers to their 3D versions. Quantitative and comparative experiments on three widely used datasets show that the proposed models are able to achieve the state-of-the-art performance (with a low number of model parameters) for different training sample sizes, especially for very small training sets.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics