Chen Zan, Guo Wenlong, Feng Yuanjing, Li Yongqiang, Zhao Changchen, Ren Yi, Shao Ling
IEEE Trans Image Process. 2021;30:7112-7126. doi: 10.1109/TIP.2021.3088611. Epub 2021 Aug 12.
Deep learning has recently been intensively studied in the context of image compressive sensing (CS) to discover and represent complicated image structures. These approaches, however, either suffer from nonflexibility for an arbitrary sampling ratio or lack an explicit deep-learned regularization term. This paper aims to solve the CS reconstruction problem by combining the deep-learned regularization term and proximal operator. We first introduce a regularization term using a carefully designed residual-regressive net, which can measure the distance between a corrupted image and a clean image set and accurately identify to which subspace the corrupted image belongs. We then address a proximal operator with a tailored dilated residual channel attention net, which enables the learned proximal operator to map the distorted image into the clean image set. We adopt an adaptive proximal selection strategy to embed the network into the loop of the CS image reconstruction algorithm. Moreover, a self-ensemble strategy is presented to improve CS recovery performance. We further utilize state evolution to analyze the effectiveness of the designed networks. Extensive experiments also demonstrate that our method can yield superior accurate reconstruction (PSNR gain over 1 dB) compared to other competing approaches while achieving the current state-of-the-art image CS reconstruction performance. The test code is available at https://github.com/zjut-gwl/CSDRCANet.
深度学习最近在图像压缩感知(CS)的背景下得到了深入研究,以发现和表示复杂的图像结构。然而,这些方法要么在任意采样率下缺乏灵活性,要么缺少一个明确的深度学习正则化项。本文旨在通过结合深度学习正则化项和近端算子来解决CS重建问题。我们首先使用精心设计的残差回归网络引入一个正则化项,它可以测量受损图像与干净图像集之间的距离,并准确识别受损图像所属的子空间。然后,我们用一个定制的扩张残差通道注意力网络来处理近端算子,使学习到的近端算子能够将失真图像映射到干净图像集。我们采用自适应近端选择策略将网络嵌入到CS图像重建算法的循环中。此外,还提出了一种自集成策略来提高CS恢复性能。我们进一步利用状态演化来分析所设计网络的有效性。大量实验还表明,与其他竞争方法相比,我们的方法可以产生更优的精确重建效果(PSNR增益超过1dB),同时实现当前的图像CS重建性能的最优水平。测试代码可在https://github.com/zjut-gwl/CSDRCANet获取。