Stochastic Chebyshev Gradient Descent for Spectral Optimization
This work addresses the computational bottleneck in machine learning for tasks involving spectral optimization, offering a novel method for faster and more stable convergence in stochastic gradient descent.
The paper tackled the problem of efficiently optimizing spectral functions, such as log-determinant and nuclear norm, by developing unbiased stochastic gradients for spectral-sums, which reduced computational complexity from cubic to faster convergence rates as demonstrated in numerical experiments.
A large class of machine learning techniques requires the solution of optimization problems involving spectral functions of parametric matrices, e.g. log-determinant and nuclear norm. Unfortunately, computing the gradient of a spectral function is generally of cubic complexity, as such gradient descent methods are rather expensive for optimizing objectives involving the spectral function. Thus, one naturally turns to stochastic gradient methods in hope that they will provide a way to reduce or altogether avoid the computation of full gradients. However, here a new challenge appears: there is no straightforward way to compute unbiased stochastic gradients for spectral functions. In this paper, we develop unbiased stochastic gradients for spectral-sums, an important subclass of spectral functions. Our unbiased stochastic gradients are based on combining randomized trace estimators with stochastic truncation of the Chebyshev expansions. A careful design of the truncation distribution allows us to offer distributions that are variance-optimal, which is crucial for fast and stable convergence of stochastic gradient methods. We further leverage our proposed stochastic gradients to devise stochastic methods for objective functions involving spectral-sums, and rigorously analyze their convergence rate. The utility of our methods is demonstrated in numerical experiments.