English

Scalable Learning and MAP Inference for Nonsymmetric Determinantal Point Processes

Machine Learning 2021-04-14 v2 Machine Learning

Abstract

Determinantal point processes (DPPs) have attracted significant attention in machine learning for their ability to model subsets drawn from a large item collection. Recent work shows that nonsymmetric DPP (NDPP) kernels have significant advantages over symmetric kernels in terms of modeling power and predictive performance. However, for an item collection of size MM, existing NDPP learning and inference algorithms require memory quadratic in MM and runtime cubic (for learning) or quadratic (for inference) in MM, making them impractical for many typical subset selection tasks. In this work, we develop a learning algorithm with space and time requirements linear in MM by introducing a new NDPP kernel decomposition. We also derive a linear-complexity NDPP maximum a posteriori (MAP) inference algorithm that applies not only to our new kernel but also to that of prior work. Through evaluation on real-world datasets, we show that our algorithms scale significantly better, and can match the predictive performance of prior work.

Keywords

Cite

@article{arxiv.2006.09862,
  title  = {Scalable Learning and MAP Inference for Nonsymmetric Determinantal Point Processes},
  author = {Mike Gartrell and Insu Han and Elvis Dohmatob and Jennifer Gillenwater and Victor-Emmanuel Brunel},
  journal= {arXiv preprint arXiv:2006.09862},
  year   = {2021}
}

Comments

ICLR 2021

R2 v1 2026-06-23T16:24:15.080Z