NUK - logo
E-viri
Celotno besedilo
Recenzirano
  • LiteDepthwiseNet: A Lightwe...
    Cui, Benlei; Dong, Xue-Mei; Zhan, Qiaoqiao; Peng, Jiangtao; Sun, Weiwei

    IEEE transactions on geoscience and remote sensing, 2022, Letnik: 60
    Journal Article

    Deep learning methods have shown considerable potential for hyperspectral image (HSI) classification, which can achieve high accuracy compared with traditional methods. However, they often need a large number of training samples and have a lot of parameters and high computational overhead. To solve these problems, this article proposes new network architecture, LiteDepthwiseNet, for HSI classification. Based on 3-D depthwise convolution, LiteDepthwiseNet can decompose standard convolution into depthwise convolution and pointwise convolution, which can achieve high classification performance with minimal parameters. Moreover, we remove the ReLU layer and batch normalization layer in the original 3-D depthwise convolution, which is likely to improve the overfitting phenomenon of the model on small-sized data sets. In addition, focal loss is used as the loss function to improve the model's attention on difficult samples and unbalanced data, and its training performance is significantly better than that of cross-entropy loss or balanced cross-entropy loss. Experiment results on five benchmark hyperspectral data sets show that LiteDepthwiseNet achieves state-of-the-art performance with a very small number of parameters and low computational cost.