• Photonics Research
  • Vol. 10, Issue 8, 1848 (2022)
Lishun Wang1,2, Zongliang Wu3, Yong Zhong1,2,4,*, and Xin Yuan3,5,*
Author Affiliations
  • 1Chengdu Institute of Computer Application, Chinese Academy of Sciences, Chengdu 610041, China
  • 2University of Chinese Academy of Sciences, Beijing 100049, China
  • 3Research Center for Industries of the Future and School of Engineering, Westlake University, Hangzhou 310030, China
  • 4e-mail: zhongyong@casit.com.cn
  • 5e-mail: xyuan@westlake.edu.cn
  • show less
    DOI: 10.1364/PRJ.458231 Cite this Article Set citation alerts
    Lishun Wang, Zongliang Wu, Yong Zhong, Xin Yuan, "Snapshot spectral compressive imaging reconstruction using convolution and contextual Transformer," Photonics Res. 10, 1848 (2022) Copy Citation Text show less

    Abstract

    Spectral compressive imaging (SCI) is able to encode a high-dimensional hyperspectral image into a two-dimensional snapshot measurement, and then use algorithms to reconstruct the spatio-spectral data-cube. At present, the main bottleneck of SCI is the reconstruction algorithm, and state-of-the-art (SOTA) reconstruction methods generally face problems of long reconstruction times and/or poor detail recovery. In this paper, we propose a hybrid network module, namely, a convolution and contextual Transformer (CCoT) block, that can simultaneously acquire the inductive bias ability of convolution and the powerful modeling ability of Transformer, which is conducive to improving the quality of reconstruction to restore fine details. We integrate the proposed CCoT block into a physics-driven deep unfolding framework based on the generalized alternating projection (GAP) algorithm, and further propose the GAP-CCoT network. Finally, we apply the GAP-CCoT algorithm to SCI reconstruction. Through experiments on a large amount of synthetic data and real data, our proposed model achieves higher reconstruction quality (>2dB in peak signal-to-noise ratio on simulated benchmark datasets) and a shorter running time than existing SOTA algorithms by a large margin. The code and models are publicly available at https://github.com/ucaswangls/GAP-CCoT.
    Gu,v==1nλFu,v,l+Zu,v,

    View in Article

    H=[D1,,Dnλ]Rnx(ny+nλ1)×nxnynλ,

    View in Article

    g=Hf+z.

    View in Article

    f^=argminf12gHf2+λΩ(f),

    View in Article

    (f^,v^)=argminf,v12fv22+λΩ(v),s.t.g=Hf.

    View in Article

    A=Conv1(Conv2([K1,Q]3)),

    View in Article

    LMSE(Θ)=1nλ=1nλF^F22,

    View in Article

    Lishun Wang, Zongliang Wu, Yong Zhong, Xin Yuan, "Snapshot spectral compressive imaging reconstruction using convolution and contextual Transformer," Photonics Res. 10, 1848 (2022)
    Download Citation