Combining Low-Rank and Deep Plug-and-Play Priors for Snapshot Compressive Imaging

被引:28
作者
Chen, Yong [1 ]
Gui, Xinfeng [1 ]
Zeng, Jinshan [1 ]
Zhao, Xi-Le [2 ]
He, Wei [3 ]
机构
[1] Jiangxi Normal Univ, Sch Comp & Informat Engn, Nanchang 330022, Jiangxi, Peoples R China
[2] Univ Elect Sci & Technol China, Sch Math Sci, Chengdu 610051, Peoples R China
[3] Wuhan Univ, State Key Lab Informat Engn Surveying Mapping & R, Wuhan 430079, Peoples R China
基金
中国国家自然科学基金;
关键词
Image reconstruction; Electronics packaging; Imaging; Learning systems; Training; Data models; Correlation; Deep image prior (DIP); hyperspectral compressive imaging; low-rank representation; plug-and-play (PnP) prior;
D O I
10.1109/TNNLS.2023.3294262
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Snapshot compressive imaging (SCI) is a promising technique that captures a 3-D hyperspectral image (HSI) by a 2-D detector in a compressed manner. The ill-posed inverse process of reconstructing the HSI from their corresponding 2-D measurements is challenging. However, current approaches either neglect the underlying characteristics, such as high spectral correlation, or demand abundant training datasets, resulting in an inadequate balance among performance, generalizability, and interpretability. To address these challenges, in this article, we propose a novel approach called LR2DP that integrates the model-driven low-rank prior and data-driven deep priors for SCI reconstruction. This approach not only captures the spectral correlation and deep spatial features of HSI but also takes advantage of both model-based and learning-based methods without requiring any extra training datasets. Specifically, to preserve the strong spectral correlation of the HSI effectively, we propose that the HSI lies in a low-rank subspace, thereby transforming the problem of reconstructing the HSI into estimating the spectral basis and spatial representation coefficient. Inspired by the mutual promotion of unsupervised deep image prior (DIP) and trained deep denoising prior (DDP), we integrate the unsupervised network and pre-trained deep denoiser into the plug-and-play (PnP) regime to estimate the representation coefficient together, aiming to explore the internal target image prior (learned by DIP) and the external training image prior (depicted by pre-trained DDP) of the HSI. An effective half-quadratic splitting (HQS) technique is employed to optimize the proposed HSI reconstruction model. Extensive experiments on both simulated and real datasets demonstrate the superiority of the proposed method over the state-of-the-art approaches.
引用
收藏
页码:16396 / 16408
页数:13
相关论文
共 57 条
[1]   Hyperspectral Unmixing Overview: Geometrical, Statistical, and Sparse Regression-Based Approaches [J].
Bioucas-Dias, Jose M. ;
Plaza, Antonio ;
Dobigeon, Nicolas ;
Parente, Mario ;
Du, Qian ;
Gader, Paul ;
Chanussot, Jocelyn .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2012, 5 (02) :354-379
[2]  
Cai Y., 2022, Advances in Neural Information Processing Systems, V35, P37749
[3]   Coarse-to-Fine Sparse Transformer for Hyperspectral Image Reconstruction [J].
Cai, Yuanhao ;
Lin, Jing ;
Hu, Xiaowan ;
Wang, Haoqian ;
Yuan, Xin ;
Zhang, Yulun ;
Timofte, Radu ;
Van Gool, Luc .
COMPUTER VISION - ECCV 2022, PT XVII, 2022, 13677 :686-704
[4]   Mask-guided Spectral-wise Transformer for Efficient Hyperspectral Image Reconstruction [J].
Cai, Yuanhao ;
Lin, Jing ;
Hu, Xiaowan ;
Wang, Haoqian ;
Yuan, Xin ;
Zhang, Yulun ;
Timofte, Radu ;
Van Gool, Luc .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, :17481-17490
[5]   Computational Snapshot Multispectral Cameras Toward dynamic capture of the spectral world [J].
Cao, Xun ;
Yue, Tao ;
Lin, Xing ;
Lin, Stephen ;
Yuan, Xin ;
Dai, Qionghai ;
Carin, Lawrence ;
Brady, David J. .
IEEE SIGNAL PROCESSING MAGAZINE, 2016, 33 (05) :95-108
[6]  
Chen Y., 2022, IEEE Trans. Geosci. Remote Sens., V60
[7]   Hyperspectral and Multispectral Image Fusion Using Factor Smoothed Tensor Ring Decomposition [J].
Chen, Yong ;
Zeng, Jinshan ;
He, Wei ;
Zhao, Xi-Le ;
Huang, Ting-Zhu .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[8]   Hyperspectral Image Restoration Using Weighted Group Sparsity-Regularized Low-Rank Tensor Decomposition [J].
Chen, Yong ;
He, Wei ;
Yokoya, Naoto ;
Huang, Ting-Zhu .
IEEE TRANSACTIONS ON CYBERNETICS, 2020, 50 (08) :3556-3570
[9]   Recurrent Neural Networks for Snapshot Compressive Imaging [J].
Cheng, Ziheng ;
Chen, Bo ;
Lu, Ruiying ;
Wang, Zhengjue ;
Zhang, Hao ;
Meng, Ziyi ;
Yuan, Xin .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (02) :2264-2281
[10]   BIRNAT: Bidirectional Recurrent Neural Networks with Adversarial Training for Video Snapshot Compressive Imaging [J].
Cheng, Ziheng ;
Lu, Ruiying ;
Wang, Zhengjue ;
Zhang, Hao ;
Chen, Bo ;
Meng, Ziyi ;
Yuan, Xin .
COMPUTER VISION - ECCV 2020, PT XXIV, 2020, 12369 :258-275