Over the past few years, the application of few-shot learning (FSL) in hyperspectral image (HSI) classification has emerged as a popular research area. Therefore, this paper proposes a lightweight network named UNet-PRFN (U-Net with principal component analysis and rotate-flip-noise augmentation) based on 3D U-Net. Firstly, we reduce redundant data and enhance sample diversity through dimensionality reduction by using principal component analysis (PCA) and data augmentation by using rotate-flip-noise (RFN). Then, we design a 3D-U-Net-like structure aimed at generating more discriminative features by employing segmentation and recombination strategies. Finally, we apply small-sample loss function and comparative loss function to minimize the sample distribution entropy. Our experiments demonstrate that the proposed approach surpasses multiple current leading models across four publicly accessible datasets.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

UNet-PRFN: A Lightweight Network Based on 3D U-Net for Few-Shot Hyperspectral Image Classification

  • Yafeng Wang,
  • Yi Liu,
  • Caihong Mu

摘要

Over the past few years, the application of few-shot learning (FSL) in hyperspectral image (HSI) classification has emerged as a popular research area. Therefore, this paper proposes a lightweight network named UNet-PRFN (U-Net with principal component analysis and rotate-flip-noise augmentation) based on 3D U-Net. Firstly, we reduce redundant data and enhance sample diversity through dimensionality reduction by using principal component analysis (PCA) and data augmentation by using rotate-flip-noise (RFN). Then, we design a 3D-U-Net-like structure aimed at generating more discriminative features by employing segmentation and recombination strategies. Finally, we apply small-sample loss function and comparative loss function to minimize the sample distribution entropy. Our experiments demonstrate that the proposed approach surpasses multiple current leading models across four publicly accessible datasets.