More Efficient Accelerated Proximal Algorithm for Nonconvex Problems

Quanming Yao,James Tin-Yau Kwok
2016-01-01
Abstract:The accelerated proximal gradient (APG) algorithm is a popular optimization tool in machine learning. Though originally designed for convex problems, there is growing interest in extending it for nonconvex problems. Recent examples include the monotone and nonmonotone APG algorithms in cite{li2015accelerated}. However, each iteration requires more than one proximal steps. This can be computationally demanding when the proximal step has no closed-form solution or when the solution is expensive. In this paper, we propose an efficient APG algorithm for nonconvex problems. In each iteration, it requires only one inexact (less expensive) proximal step. Convergence to a critical point is still guaranteed, and a $O(1/k)$ convergence rate is derived. Experimental results on various machine learning problems demonstrate that the proposed algorithm has comparable prediction performance as the state-of-the-art, but is much faster.
What problem does this paper attempt to address?