Leveraged volume sampling for linear regression

TL;DR

This paper introduces a rescaled volume sampling method for linear regression that achieves near-optimal unbiased estimates with fewer samples, improving previous bounds and offering a new efficient algorithm with broader applications.

Contribution

The paper develops a rescaled volume sampling technique that provides unbiased estimates with improved sample complexity for linear regression.

Findings

Rescaled volume sampling achieves $k=O(d ext{log}d + d/ extepsilon)$ sample size for $(1+ extepsilon)$-approximate loss.

The new method outperforms previous unbiased estimators requiring $k=O(d^2/ extepsilon)$ samples.

Introduces a determinantal rejection sampling algorithm applicable to determinantal point processes.

Abstract

Suppose an $n \times d$ design matrix in a linear regression problem is given, but the response for each point is hidden unless explicitly requested. The goal is to sample only a small number $k \ll n$ of the responses, and then produce a weight vector whose sum of squares loss over all points is at most $1+\epsilon$ times the minimum. When $k$ is very small (e.g., $k=d$), jointly sampling diverse subsets of points is crucial. One such method called volume sampling has a unique and desirable property that the weight vector it produces is an unbiased estimate of the optimum. It is therefore natural to ask if this method offers the optimal unbiased estimate in terms of the number of responses $k$ needed to achieve a $1+\epsilon$ loss approximation. Surprisingly we show that volume sampling can have poor behavior when we require a very accurate approximation -- indeed worse than some i.i.d. sampling techniques whose estimates are biased, such as leverage score sampling. We then develop a new rescaled variant of volume sampling that produces an unbiased estimate which avoids this bad behavior and has at least as good a tail bound as leverage score sampling: sample size $k=O(d\log d + d/\epsilon)$ suffices to guarantee total loss at most $1+\epsilon$ times the minimum with high probability. Thus, we improve on the best previously known sample size for an unbiased estimator, $k=O(d^2/\epsilon)$. Our rescaling procedure leads to a new efficient algorithm for volume sampling which is based on a determinantal rejection sampling technique with potentially broader applications to determinantal point processes. Other contributions include introducing the combinatorics needed for rescaled volume sampling and developing tail bounds for sums of dependent random matrices which arise in the process.