ISTA-Net
ISTA-Net: Interpretable Optimization-Inspired Deep Network for Image Compressive Sensing. With the aim of developing a fast yet accurate algorithm for compressive sensing (CS) reconstruction of natural images, we combine in this paper the merits of two existing categories of CS methods: the structure insights of traditional optimization-based methods and the speed of recent network-based ones. Specifically, we propose a novel structured deep network, dubbed ISTA-Net, which is inspired by the Iterative Shrinkage-Thresholding Algorithm (ISTA) for optimizing a general ℓ1 norm CS reconstruction model. To cast ISTA into deep network form, we develop an effective strategy to solve the proximal mapping associated with the sparsity-inducing regularizer using nonlinear transforms. All the parameters in ISTA-Net (eg nonlinear transforms, shrinkage thresholds, step sizes, etc.) are learned end-to-end, rather than being hand-crafted. Moreover, considering that the residuals of natural images are more compressible, an enhanced version of ISTA-Net in the residual domain, dubbed {ISTA-Net}+, is derived to further improve CS reconstruction. Extensive CS experiments demonstrate that the proposed ISTA-Nets outperform existing state-of-the-art optimization-based and network-based CS methods by large margins, while maintaining fast computational speed. Our source codes are available: extsl{http://jianzhang.tech/projects/ISTA-Net}.
Keywords for this software
References in zbMATH (referenced in 7 articles )
Showing results 1 to 7 of 7.
Sorted by year (- Hou, Ruizhi; Li, Fang: IDPCNN: iterative denoising and projecting CNN for MRI reconstruction (2022)
- Bubba, Tatiana A.; Galinier, Mathilde; Lassas, Matti; Prato, Marco; Ratti, Luca; Siltanen, Samuli: Deep neural networks for inverse problems with pseudodifferential operators: an application to limited-angle tomography (2021)
- Chen, Yunmei; Liu, Hongcheng; Ye, Xiaojing; Zhang, Qingchao: Learnable descent algorithm for nonsmooth nonconvex image reconstruction (2021)
- Pesquet, Jean-Christophe; Repetti, Audrey; Terris, Matthieu; Wiaux, Yves: Learning maximally monotone operators for image recovery (2021)
- Wang, Lizhi; Zhang, Shipeng; Huang, Hua: Adaptive dimension-discriminative low-rank tensor recovery for computational hyperspectral imaging (2021)
- Bertocchi, Carla; Chouzenoux, Emilie; Corbineau, Marie-Caroline; Pesquet, Jean-Christophe; Prato, Marco: Deep unfolding of a proximal interior point method for image restoration (2020)
- Tian, Chunwei; Fei, Lunke; Zheng, Wenxian; Xu, Yong; Zuo, Wangmeng; Lin, Chia-Wen: Deep learning on image denoising: an overview (2020)