A Statistical View of Column Subset Selection

Anav Sood, Trevor Hastie
7/24/2023
stat.MEcs.DScs.LG

Official code: AnavSood/CSS (2 stars, last commit 2025-04-20).

Abstract

We consider the problem of selecting a small subset of representative variables from a large dataset. In the computer science literature, this dimensionality reduction problem is typically formalized as Column Subset Selection (CSS). Meanwhile, the typical statistical formalization is to find an information-maximizing set of Principal Variables. This paper shows that these two approaches are equivalent, and moreover, both can be viewed as maximum likelihood estimation within a certain semi-parametric model. Within this model, we establish suitable conditions under which the CSS estimate is consistent in high dimensions, specifically in the proportional asymptotic regime where the number of variables over the sample size converges to a constant. Using these connections, we show how to efficiently (1) perform CSS using only summary statistics from the original dataset; (2) perform CSS in the presence of missing and/or censored data; and (3) select the subset size for CSS in a hypothesis testing framework.

View on arXivView PDF

Code Implementations(1)

AnavSood/CSSOfficial
21Python, Jupyter NotebookFeb 4, 20231 years ago

Similar Papers

Cite this paper

@article{sood2023statistical,
  title  = {A Statistical View of Column Subset Selection},
  author = {Anav Sood and Trevor Hastie},
  year   = {2023},
  eprint = {2307.12892},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2307.12892v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2307.12892.svg)](https://codeofpaper.com/paper/2307.12892)

Discussion