Solving Large-Scale Sparse PCA to Certifiable (Near) Optimality
Abstract
Sparse principal component analysis (PCA) is a popular dimensionality reduction technique for obtaining principal components which are linear combinations of a small subset of the original features. Existing approaches cannot supply certifiably optimal principal components with more than of variables. By reformulating sparse PCA as a convex mixed-integer semidefinite optimization problem, we design a cutting-plane method which solves the problem to certifiable optimality at the scale of selecting k=5 covariates from p=300 variables, and provides small bound gaps at a larger scale. We also propose a convex relaxation and greedy rounding scheme that provides bound gaps of in practice within minutes for s or hours for s and is therefore a viable alternative to the exact method at scale. Using real-world financial and medical datasets, we illustrate our approach's ability to derive interpretable principal components tractably at scale.
Cite
@article{arxiv.2005.05195,
title = {Solving Large-Scale Sparse PCA to Certifiable (Near) Optimality},
author = {Dimitris Bertsimas and Ryan Cory-Wright and Jean Pauphilet},
journal= {arXiv preprint arXiv:2005.05195},
year = {2022}
}
Comments
Revision submitted to JMLR