Fast determinantal point processes via distortion-free intermediate sampling

TL;DR

This paper introduces a fast, efficient algorithm for sampling from determinantal point processes with reduced preprocessing and sampling time, enabling scalable applications in machine learning and data analysis.

Contribution

The paper proposes a novel regularized DPP with a distortion-free intermediate distribution, achieving near-linear preprocessing and polynomial-time sampling independent of dataset size.

Findings

Preprocessing runs in near-linear time based on non-zero entries of the data matrix.

Sampling complexity is polynomial in the feature dimension, independent of dataset size.

The method maintains exact probabilities without distortion, ensuring accurate sampling.

Abstract

Given a fixed $n\times d$ matrix $\mathbf{X}$, where $n\gg d$, we study the complexity of sampling from a distribution over all subsets of rows where the probability of a subset is proportional to the squared volume of the parallelepiped spanned by the rows (a.k.a. a determinantal point process). In this task, it is important to minimize the preprocessing cost of the procedure (performed once) as well as the sampling cost (performed repeatedly). To that end, we propose a new determinantal point process algorithm which has the following two properties, both of which are novel: (1) a preprocessing step which runs in time $O(\text{number-of-non-zeros}(\mathbf{X})\cdot\log n)+\text{poly}(d)$, and (2) a sampling step which runs in $\text{poly}(d)$ time, independent of the number of rows $n$. We achieve this by introducing a new regularized determinantal point process (R-DPP), which serves as an intermediate distribution in the sampling procedure by reducing the number of rows from $n$ to $\text{poly}(d)$. Crucially, this intermediate distribution does not distort the probabilities of the target sample. Our key novelty in defining the R-DPP is the use of a Poisson random variable for controlling the probabilities of different subset sizes, leading to new determinantal formulas such as the normalization constant for this distribution. Our algorithm has applications in many diverse areas where determinantal point processes have been used, such as machine learning, stochastic optimization, data summarization and low-rank matrix reconstruction.