Deep Neural Network Based Hyperspectral Pixel Classification With Factorized Spectral-Spatial Feature Representation

被引:9
作者
Chen, Jingzhou [1 ]
Chen, Siyu [1 ]
Zhou, Peilin [2 ]
Qian, Yuntao [1 ]
机构
[1] Zhejiang Univ, Coll Comp Sci, Inst Artificial Intelligence, Hangzhou 310027, Zhejiang, Peoples R China
[2] Huawei Co, Shanghai, Peoples R China
基金
中国国家自然科学基金;
关键词
Hyperspectral pixel classification; deep neural networks; spectral-spatial feature factorization; IMAGE CLASSIFICATION; FEATURE-EXTRACTION; BAND SELECTION; INFORMATION; SPARSE;
D O I
10.1109/ACCESS.2019.2923776
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Deep learning has been widely used for hyperspectral pixel classification due to its ability to generate deep feature representation. However, how to construct an efficient and powerful network suitable for hyperspectral data is still under exploration. In this paper, a novel neural network model is designed for taking full advantage of the spectral-spatial structure of hyperspectral data. First, we extract pixel-based intrinsic features from rich yet redundant spectral bands by a subnetwork with the supervised pretraining scheme. Second, in order to utilize the local spatial correlation among pixels, we share the previous subnetwork as a spectral feature extractor for each pixel in a patch of the image, after which the spectral features of all pixels in a patch are combined and fed into the subsequent classification subnetwork. Finally, the whole network is further fine-tuned to improve its classification performance. Especially, the spectralspatial factorization scheme is applied in our model architecture, making the network size and the number of parameters great less than the existing spectral-spatial deep networks for hyperspectral image classification. Compared with other state-of-the-art deep learning methods, experiments on the hyperspectral data sets show that our method achieves 0.18%-7.6%, 0.1%-3.58%, and 0.21%-3.09% improvement on overall accuracy (OA), average accuracy (AA), and kappa, respectively, while having smaller network size and fewer parameters.
引用
收藏
页码:81407 / 81418
页数:12
相关论文
共 51 条
[1]  
Alec R., 2016, 4 INT C LEARN REPR I, P1
[2]  
[Anonymous], PROC CVPR IEEE
[3]  
[Anonymous], 2016, P ICLR
[4]  
[Anonymous], 2017, ARXIV170404861
[5]  
[Anonymous], 2017, COMMUN ACM, DOI DOI 10.1145/3065386
[6]  
[Anonymous], PROC CVPR IEEE
[7]  
[Anonymous], 2017, NEURIPS
[8]  
[Anonymous], 2017, ARXIV171205689
[9]   Improved manifold coordinate representations of large-scale hyperspectral scenes [J].
Bachmann, Charles M. ;
Ainsworth, Thomas L. ;
Fusina, Robert A. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2006, 44 (10) :2786-2803
[10]   Statistical Inference in PCA for Hyperspectral Images [J].
Bajorski, Peter .
IEEE JOURNAL OF SELECTED TOPICS IN SIGNAL PROCESSING, 2011, 5 (03) :438-445