Convolutional neural networks for hyperspectral image classification

NEUROCOMPUTING

卷:219

Article

Yu, Shiqi¹  Jia, Sen¹  Xu, Chunyan² 

[1] Shenzhen Univ, Coll Comp Sci & Software Engn, Shenzhen, Peoples R China

[2] Nanjing Univ Sci & Technol, Sch Comp Sci & Technol, Nanjing, Jiangsu, Peoples R China

关键词: Hyperspectral image classification;    Convolutional neural networks;    Deep learning;   

DOI  :  10.1016/j.neucom.2016.09.010

来源: Elsevier

【 摘 要 】

As a powerful visual model, convolutional neural networks (CNNs) have demonstrated remarkable performance in various visual recognition problems, and attracted considerable attention in recent years. However, due to the highly correlated bands and insufficient training samples of hyperspectral image data, it still remains a challenging problem to effectively apply the CNN models on hyperspectral images. In this paper, an efficient CNN architecture has been proposed to boost its discriminative capability for hyperspectral image classification, in which the original data is used as the input and the final CNN outputs are the predicted class-related results. The proposed CNN infrastructure has several distinct advantages. Firstly, different from traditional classification methods those need hand-crafted features, the CNN model used here is designed to deal with the problem of hyperspectral image analysis in an end-to-end way. Secondly, the parameters of the CNN model are optimized from a small training set, while the over-fitting problem of the neural network has been alleviated to some extent. Finally, in order to better deal with the hyperspectral image information, 1 x 1 convolutional layers have been adopted, and an average pooling layer and larger dropout rates have also been employed in the whole CNN procedure. The experiments on three benchmark data sets have demonstrated that the proposed CNN architecture considerably outperforms other state-of-the-art methods. (C) 2016 Elsevier B.V. All rights reserved.

【 授权许可】

Free