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

Scalable Learning and MAP Inference for Nonsymmetric Determinantal Point Processes

Criteo AI Lab(Criteo AI实验室);KAIST(韩国科学技术院);Google Research(谷歌研究院);ENSAE ParisTech(巴黎高等统计与经济管理学院)

Mike Gartrell, Insu Han, Elvis Dohmatob, Jennifer Gillenwater, Victor-Emmanuel Brunel

英文摘要:

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 $M$, existing NDPP learning and inference algorithms require memory quadratic in $M$ and runtime cubic (for learning) or quadratic (for inference) in $M$, making them impractical for many typical subset selection tasks. In this work, we develop a learning algorithm with space and time requirements linear in $M$ 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.

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

↑