We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:

References & Citations


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Statistics > Methodology

Title: Probabilistic Best Subset Selection via Gradient-Based Optimization

Abstract: In high-dimensional statistics, variable selection recovers the latent sparse patterns from all possible covariate combinations. This paper proposes a novel optimization method to solve the exact L0-regularized regression problem, which is also known as the best subset selection. We reformulate the optimization problem from a discrete space to a continuous one via probabilistic reparameterization. The new objective function is differentiable but its gradient often cannot be computed in a closed form. Then we propose a family of unbiased gradient estimators to optimize the best subset selection objectives by the stochastic gradient descent. Within this family, we identify the estimator with uniformly minimum variance. Theoretically, we study the general conditions under which the method is guaranteed to converge to the ground truth in expectation. The proposed method can find the true regression model from thousands of covariates in seconds. In a wide variety of synthetic and semi-synthetic data, the proposed method outperforms existing variable selection tools based on the relaxed penalties, coordinate descent, and mixed integer optimization in both sparse pattern recovery and out-of-sample prediction.
Subjects: Methodology (stat.ME)
Cite as: arXiv:2006.06448 [stat.ME]
  (or arXiv:2006.06448v4 [stat.ME] for this version)

Submission history

From: Mingzhang Yin [view email]
[v1] Thu, 11 Jun 2020 13:57:29 GMT (217kb,D)
[v2] Mon, 22 Jun 2020 18:28:46 GMT (217kb,D)
[v3] Fri, 7 Aug 2020 04:23:34 GMT (154kb,D)
[v4] Wed, 1 Jun 2022 01:59:07 GMT (215kb,D)

Link back to: arXiv, form interface, contact.