Passer à la navigation principale Passer à la recherche Passer au contenu principal

SCALABLE LEARNING AND MAP INFERENCE FOR NONSYMMETRIC DETERMINANTAL POINT PROCESSES

  • Criteo AI Lab
  • Korea Advanced Institute of Science and Technology
  • Google Inc.

Résultats de recherche: Contribution à une conférencePapierRevue par des pairs

Résumé

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.

langue originaleAnglais
étatPublié - 1 janv. 2021
Evénement9th International Conference on Learning Representations, ICLR 2021 - Virtual, Online, Autriche
Durée: 3 mai 20217 mai 2021

Une conférence

Une conférence9th International Conference on Learning Representations, ICLR 2021
Pays/TerritoireAutriche
La villeVirtual, Online
période3/05/217/05/21

Empreinte digitale

Examiner les sujets de recherche de « SCALABLE LEARNING AND MAP INFERENCE FOR NONSYMMETRIC DETERMINANTAL POINT PROCESSES ». Ensemble, ils forment une empreinte digitale unique.

Contient cette citation