GAPMSF-Net: Generalized Alternating Projection With Multi-Stage Fusion Network for Snapshot Compressive Imaging

Xin Liu,Zhangxin Chen,Ce Zhu,Yipeng Liu
DOI: https://doi.org/10.1109/tci.2023.3304472
IF: 5.4
2023-08-25
IEEE Transactions on Computational Imaging
Abstract:Snapshot compressive imaging (SCI) aims to fully capture 3-dimensional spatio-temporal data using 2D sensors. It takes different masks to modulate the high-speed scene during exposure to get compressed frames. In SCI systems, model-based and learning-based methods have been used to reconstruct video frames from compressed samples, but most of these methods cannot achieve a good balance between accuracy and speed. For SCI reconstruction, generalized alternating projection (GAP) is one of the most effective methods. To enhance the SCI reconstruction performance from training, we propose a deep unfolding network called GAPMSF-Net whose stages are unfolded from the iterations of the GAP algorithm to allow learnable parameters in reconstruction. First, motivated by the fact the accelerated computation process of the original GAP algorithm can significantly improve the convergence speed of the algorithm, we design a deep acceleration network based on its computational principle, which can learn a rule to adaptively adjust the linear manifold in a data-driven manner based on the reconstruction results of the previous phase. In addition, to enhance the information interaction between different reconstruction stages in the deep unrolling algorithm, we design a multi-stage fusion of deep priors. At each reconstruction stage, the deep prior fuses and interacts with the hidden states that take all the previous information. The superiority of the proposed method is demonstrated on simulated and real datasets in terms of accuracy and computational complexity.
engineering, electrical & electronic,imaging science & photographic technology
What problem does this paper attempt to address?