Sparse Convex Regression
Dimitris Bertsimas, Nishanth Mundru
INFORMS Journal on Computing Β· 2021 Β· Volume 33 Β· Issue 1
Abstract
We consider the problem of best π k -subset convex regression using π n observations in π d variables. For the case without sparsity, we develop a scalable algorithm for obtaining high quality solutions in practical times that compare favorably with other state of the art methods. We show that by using a cutting plane method, the least squares convex regression problem can be solved for sizes ( π , π ) = ( 1 0 4 , 1 0 ) ( n , d ) = ( 10 4 , 10 ) in minutes and ( π , π ) = ( 1 0 5 , 1 β’ 0 2 ) ( n , d ) = ( 10 5 , 1 β’ 0 2 ) in hours. Our algorithm can be adapted to solve variants such as finding the best convex or concave functions with coordinate-wise monotonicity, norm-bounded subgradients, and minimize the β 1 β 1 lossβall with similar scalability to the least squares convex regression problem. Under sparsity, we propose algorithms which iteratively solve for the best subset of features based on first order and cutting plane methods. We show that our methods scale for sizes ( π , π , π = 1 0 4 , 1 β’ 0 2 , 1 0 ) ( n , d , k = 10 4 , 1 β’ 0 2 , 10 ) in minutes and ( π , π , π = 1 0 5 , 1 β’ 0 2 , 1 0 ) ( n , d , k = 10 5 , 1 β’ 0 2 , 10 ) in hours. We demonstrate that these methods control for the false discovery rate effectively.
Keywords
statistics, regression, linear programming, large-scale systems, applications
Enable JavaScript for citation tools and additional paper interactions.