A Stochastic Proximal Alternating Minimization for Nonsmooth and Nonconvex Optimization
Derek Driggs,Junqi Tang,Jingwei Liang,Mike Davies,Carola-Bibiane Schonlieb
DOI: https://doi.org/10.1137/20m1387213
IF: 1.938
2021-01-01
SIAM Journal on Imaging Sciences
Abstract:In this work, we introduce a novel stochastic proximal alternating linearized minimization algorithm [J. Bolte, S. Sabach, and M. Teboulle, Math. Program., 146 (2014), pp. 459--494] for solving a class of nonsmooth and nonconvex optimization problems. Large-scale imaging problems are becoming increasingly prevalent due to the advances in data acquisition and computational capabilities. Motivated by the success of stochastic optimization methods, we propose a stochastic variant of proximal alternating linearized minimization. We provide global convergence guarantees, demonstrating that our proposed method with variance-reduced stochastic gradient estimators, such as SAGA [A. Defazio, F. Bach, and S. Lacoste-Julien, Advances in Neural Information Processing Systems, 2014, pp. 1646--1654] and SARAH [L. M. Nguyen, J. Liu, K. Scheinberg, and M. Taka'\c<^>\, Proceedings of the 34th International Conference on Machine Learning, PMLR 70, 2017, pp. 2613-2621], achieves stateof-the-art oracle complexities. We also demonstrate the efficacy of our algorithm via several numerical examples including sparse nonnegative matrix factorization, sparse principal component analysis, and blind image-deconvolution.