Scalable Learning of Determinantal Point Processes with Kernel-Structured Interactions
Organizations: ZOZO Research · Institute of Statistical Mathematics · Waseda University
Abstract
Determinantal point processes (DPPs) on a finite set are probability models that exhibit repulsion where inter-item similarities are encoded by a positive definite kernel matrix. Scaling DPP learning to large datasets typically requires imposing structural constraints on the kernel matrix, such as a low-rank-plus-diagonal form. However, structural assumptions introduce a severe trade-off between expressive power and computational scalability. Our study addresses this challenge by (a) modeling the full-rank interactions via a nonlinear kernel function and latent features, (b) designing a surrogate function for the maximum likelihood estimator, and (c) adopting stochastic optimization using random features of the nonlinear kernel. Numerical experiments on real-world datasets demonstrate that our method significantly improves this trade-off, achieving better generalization performance with remarkable parameter efficiency.
Figures & tables
| Method | N=200 | N=500 | N=1000 | N=2000 | N=3000 |
|---|---|---|---|---|---|
| MM | |||||
| LR+D r=32 | |||||
| LR+D r=r max | |||||
| KS-DPP d=32 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | N | ||||
|---|---|---|---|---|---|
| 200 | 500 | 1000 | 2000 | 3000 | |
| MM | |||||
| LR+D r=4 | |||||
| LR+D r=8 | |||||
| LR+D r=16 | |||||
| LR+D r=32 | |||||
Determinantal Point Process
Determinantal point processes model and sample diverse subsets using determinant-based probabilities.
See all 18 papers