Hyperspectral Image Denoising via Matrix Factorization and Deep Prior Regularization

被引:54
作者
Lin, Baihong [1 ,2 ]
Tao, Xiaoming [1 ,2 ]
Lu, Jianhua [1 ,2 ]
机构
[1] Tsinghua Univ, Dept Elect Engn, Beijing 100084, Peoples R China
[2] Tsinghua Univ, Beijing Natl Res Ctr Informat Sci & Technol BNRis, Beijing 100084, Peoples R China
基金
中国国家自然科学基金;
关键词
Noise reduction; Training; Sparse matrices; Optimization; Hyperspectral imaging; Computational modeling; Image denoising; Hyperspectral image denoising; nonnegative matrix factorization (NMF); deep prior regularization (DPR); convolutional neural networks (CNN); RESTORATION; SPARSE;
D O I
10.1109/TIP.2019.2928627
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep learning has been successfully introduced for 2D-image denoising, but it is still unsatisfactory for hyperspectral image (HSI) denoising due to the unacceptable computational complexity of the end-to-end training process and the difficulty of building a universal 3D-image training dataset. In this paper, instead of developing an end-to-end deep learning denoising network, we propose an HSI denoising framework for the removal of mixed Gaussian impulse noise, in which the denoising problem is modeled as a convolutional neural network (CNN) constrained non-negative matrix factorization problem. Using the proximal alternating linearized minimization, the optimization can be divided into three steps: the update of the spectral matrix, the update of the abundance matrix, and the estimation of the sparse noise. Then, we design the CNN architecture and proposed two training schemes, which can allow the CNN to be trained with a 2D-image dataset. Compared with the state-of-the-art denoising methods, the proposed method has a relatively good performance on the removal of the Gaussian and mixed Gaussian impulse noises. More importantly, the proposed model can be only trained once by a 2D-image dataset but can be used to denoise HSIs with different numbers of channel bands.
引用
收藏
页码:565 / 578
页数:14
相关论文
共 53 条
[1]  
Abadi M, 2016, ACM SIGPLAN NOTICES, V51, P1, DOI [10.1145/2951913.2976746, 10.1145/3022670.2976746]
[2]   Hyperspectral Image Denoising Using Spatio-Spectral Total Variation [J].
Aggarwal, Hemant Kumar ;
Majumdar, Angshul .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2016, 13 (03) :442-446
[3]  
Agostinelli F., 2013, ADV NEURAL INFORM PR, P1493
[4]  
[Anonymous], 2012, P 25 INT C NEUR INF
[5]  
[Anonymous], 2013, INT C MACHINE LEARNI
[6]  
[Anonymous], 2009, Advances in Neural Information Processing Systems
[7]  
[Anonymous], P INT C NEUR INF PRO
[8]  
[Anonymous], THESIS
[9]  
[Anonymous], 2016, PROC IEEE C COMPUT V, DOI DOI 10.1109/CVPR.2016.187
[10]  
[Anonymous], 2014, PROC 7 INT C IMAGE S