Reversible Autoencoder: A CNN-Based Nonlinear Lifting Scheme for Image Reconstruction

Shaohui Li,Wenrui Dai,Ziyang Zheng,Chenglin Li,Junni Zou,Hongkai Xiong
DOI: https://doi.org/10.1109/tsp.2021.3082465
IF: 4.875
2021-01-01
IEEE Transactions on Signal Processing
Abstract:Emerging deep learning approaches have facilitated image reconstruction at the expense of excessive model complexities and lack of theoretical guarantees of stability. In this paper, we propose a theoretically sound deep architecture, named reversible autoencoder (Rev-AE), from the perspective of well-developed frame theory for image reconstruction. Reversible blocks (Rev-Blocks) are developed as fundamental building blocks to realize nonlinear dimension invariant transform, up-sampling (with dimension growth) and down-sampling (with dimension reduction) in Rev-AE. The Rev-Block integrates CNN-based operators with the two-stream structure of lifting scheme to introduce nonlinearity and achieve reversibility. Therefore, Rev-AE only requires training and storage of encoder parameters. Furthermore, Rev-AE bridges existing nonlinear CNNs and well-established traditional signal processing tools to realize end-to-end optimized reversible networks with a guarantee of stability for image reconstruction for the first time. Theoretical analysis demonstrates that reversible blocks guarantee perfect reconstruction with derived frame bounds and are robust to noise perturbation under restricted energy propagation of noise. Moreover, we prove that Rev-AE achieves stable reconstruction with bounded projection angle under the presence of down-sampling and up-sampling. The stability and robustness of Rev-Blocks are also validated with numerical evaluations. Experimental results demonstrate that Rev-AE achieves comparable performance with halved parameters in comparison to recent leading methods (e.g., 3-D CNN based autoencoder and ISTA-Net$^+$) in the applications of image compression and compressive sensing.
What problem does this paper attempt to address?