Fast Relative-Error Approximation Algorithm for Ridge Regression

Ridge regression is one of the most popular and effective regularized regression methods, and one case of particular interest is that the number of features p is much larger than the number of samples n, i. e. p n. In this case, the standard optimization algorithm for ridge regression computes the optimal solution x⇤ in O(n2 p + n3 ) time. In this paper, we propose a fast relativeerror approximation algorithm for ridge regression. More specifically, our algorithm outputs a solution x̃ satisfying kx̃ x⇤ k2  ✏kx⇤ k2 with high probability and runs in Õ(nnz(A) + n3 /✏2 ) time, where nnz(A) is the number of non-zero entries of matrix A. To the best of our knowledge, this is the first algorithm for ridge regression that runs in o(n2 p) time with provable relative-error approximation bound on the output vector. In addition, we analyze the risk inflation bound of our algorithm and apply our techniques to two generalizations of ridge regression, including multiple response ridge regression and a non-linear ridge regression problem. Finally, we show empirical results on both synthetic and real datasets.

Authors

• Shouyuan Chen
• Yang Liu 0003
• Michael R. Lyu
• Irwin King
• Shengyu Zhang 0002

Keywords

No keywords are indexed for this paper.