Hyperspectral classification based on lightweight 3D-CNN with transfer learning.

Authors Organisations
  • Haokui Zhang(Author)
    Northwestern Polytechnical University
  • Ying Li(Author)
    Northwestern Polytechnical University
  • Yenan Jiang(Author)
    Northwestern Polytechnical University
  • Peng Wang(Author)
    Northwestern Polytechnical University
  • Qiang Shen(Author)
  • Chunhua Shen(Author)
    University of Adelaide
Type Article
Original languageEnglish
Pages (from-to)5813-5828
Number of pages16
JournalIEEE Transactions on Geoscience and Remote Sensing
Volume57
Issue number8
Early online date11 Apr 2019
DOI
Publication statusPublished - 01 Aug 2019
Show download statistics
View graph of relations
Citation formats

Abstract

Recently, hyperspectral image (HSI) classification approaches based on deep learning (DL) models have been proposed and shown promising performance. However, due to very limited available training samples and massive model parameters, deep learning methods may suffer from over-fitting. In this
paper, we propose an end-to-end 3D lightweight convolutional neural network (CNN) (abbreviated as 3D-LWNet) for limited samples based HSI classification. Compared with conventional 3D-CNN models, the proposed 3D-LWNet has deeper network structure, less parameters and lower computation cost, while it
results in better classification performance. To further alleviate the small sample problem, we also propose two transfer learning strategies: 1) cross-sensor strategy. We pretrain a 3D model in the source HSI datasets containing a greater number of labeled samples and then transfer it to the target HSI datasets; and
2) cross-modal strategy. We pretrain a 3D model in the 2D RGB image datasets containing a large number of samples and then transfer it to the target HSI datasets. In contrast to previous approaches, we do not impose restrictions over the source datasets in that they do not have to be collected by the same senors as the target datasets. Experiments on three public HSI datasets captured by different sensors demonstrate that our model achieves competitive performance for HSI classification compared to several state-of-the-art methods

Keywords

  • hyperspectral classification, deep learning, 3D lightweight convolutional network, transfer learning