Theoretical Linear Convergence of Unfolded ISTA and Its Practical Weights and Thresholds

Chen, Xiaohan, Liu, Jialin, Wang, Zhangyang, Yin, Wotao

Neural Information Processing Systems 

In recent years, unfolding iterative algorithms as neural networks has become an empirical success in solving sparse recovery problems. However, its theoretical understanding is still immature, which prevents us from fully utilizing the power of neural networks. In this work, we study unfolded ISTA (Iterative Shrinkage Thresholding Algorithm) for sparse signal recovery. We introduce a weight structure that is necessary for asymptotic convergence to the true sparse signal. With this structure, unfolded ISTA can attain a linear convergence, which is better than the sublinear convergence of ISTA/FISTA in general cases.