Simple one-pass algorithm for penalized linear regression with cross-validation on MapReduce

Abstract

In this paper, we propose a one-pass algorithm on MapReduce for penalized linear regression \[f_ ( , ) = \|Y - 1 - X \|_2^2 + p_ ( )\] where is the intercept which can be omitted depending on application; is the coefficients and p_ is the penalized function with penalizing parameter . f_(, ) includes interesting classes such as Lasso, Ridge regression and Elastic-net. Compared to latest iterative distributed algorithms requiring multiple MapReduce jobs, our algorithm achieves huge performance improvement; moreover, our algorithm is exact compared to the approximate algorithms such as parallel stochastic gradient decent. Moreover, what our algorithm distinguishes with others is that it trains the model with cross validation to choose optimal instead of user specified one. Key words: penalized linear regression, lasso, elastic-net, ridge, MapReduce

Tasks

Reproductions