arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.09712cs.LGmath.CO

Wasserstein Learning of Determinantal Point Processes

INSA Rouen(鲁昂国立应用科学学院);Criteo AI Lab(Criteo人工智能实验室);University of Rouen(鲁昂大学);Sorbonne University(索邦大学)

Lucas Anquetil, Mike Gartrell, Alain Rakotomamonjy, Ugo Tanielian, Clément Calauzènes

英文摘要:

Determinantal point processes (DPPs) have received significant attention as an elegant probabilistic model for discrete subset selection. Most prior work on DPP learning focuses on maximum likelihood estimation (MLE). While efficient and scalable, MLE approaches do not leverage any subset similarity information and may fail to recover the true generative distribution of discrete data. In this work, by deriving a differentiable relaxation of a DPP sampling algorithm, we present a novel approach for learning DPPs that minimizes the Wasserstein distance between the model and data composed of observed subsets. Through an evaluation on a real-world dataset, we show that our Wasserstein learning approach provides significantly improved predictive performance on a generative task compared to DPPs trained using MLE.

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

↑