arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.08417cs.LGstat.ML

Scalable Sampling for Nonsymmetric Determinantal Point Processes

Yale University(耶鲁大学);Criteo AI Lab(Criteo人工智能实验室);Google Research(谷歌研究院);Facebook AI Research(脸书人工智能研究院)

Insu Han, Mike Gartrell, Jennifer Gillenwater, Elvis Dohmatob, Amin Karbasi

英文摘要:

A determinantal point process (DPP) on a collection of $M$ items is a model, parameterized by a symmetric kernel matrix, that assigns a probability to every subset of those items. Recent work shows that removing the kernel symmetry constraint, yielding nonsymmetric DPPs (NDPPs), can lead to significant predictive performance gains for machine learning applications. However, existing work leaves open the question of scalable NDPP sampling. There is only one known DPP sampling algorithm, based on Cholesky decomposition, that can directly apply to NDPPs as well. Unfortunately, its runtime is cubic in $M$, and thus does not scale to large item collections. In this work, we first note that this algorithm can be transformed into a linear-time one for kernels with low-rank structure. Furthermore, we develop a scalable sublinear-time rejection sampling algorithm by constructing a novel proposal distribution. Additionally, we show that imposing certain structural constraints on the NDPP kernel enables us to bound the rejection rate in a way that depends only on the kernel rank. In our experiments we compare the speed of all of these samplers for a variety of real-world tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑