Basket Completion with Multi-task Determinantal Point Processes

Determinantal point processes (DPPs) have received significant attention in the recent years as an elegant model for a variety of machine learning tasks, due to their ability to elegantly model set diversity and item quality or popularity. Recent work has shown that DPPs can be effective models for product recommendation and basket completion tasks. We present an enhanced DPP model that is specialized for the task of basket completion, the multi-task DPP. We view the basket completion problem as a multi-class classification problem, and leverage ideas from tensor factorization and multi-class classification to design the multi-task DPP model. We evaluate our model on several real-world datasets, and find that the multi-task DPP provides significantly better predictive quality than a number of state-of-the-art models.

[1]  Michal Valko,et al.  Zonotope Hit-and-run for Efficient Sampling from Projection DPPs , 2017, ICML.

[2]  Steffen Rendle,et al.  Factorization Machines , 2010, 2010 IEEE International Conference on Data Mining.

[3]  Wynne Hsu,et al.  Integrating Classification and Association Rule Mining , 1998, KDD.

[4]  Dilan Görür,et al.  Diverse Personalization with Determinantal Point Process , 2013 .

[5]  Ulrich Paquet,et al.  Low-Rank Factorization of Determinantal Point Processes , 2017, AAAI.

[6]  Suvrit Sra,et al.  Fixed-point algorithms for determinantal point processes , 2015, ArXiv.

[7]  Ben Taskar,et al.  Determinantal Point Processes for Machine Learning , 2012, Found. Trends Mach. Learn..

[8]  Yifan Hu,et al.  Collaborative Filtering for Implicit Feedback Datasets , 2008, 2008 Eighth IEEE International Conference on Data Mining.

[9]  Sooyoung Kim,et al.  Classification-based collaborative filtering using market basket data , 2005, Expert Syst. Appl..

[10]  Thomas Reutterer,et al.  An improved collaborative filtering approach for predicting cross-category purchases based on binary market basket data , 2003 .

[11]  Abraham Bernstein,et al.  Blockbusters and Wallflowers: Accurate, Diverse, and Scalable Recommendations with Random Walks , 2015, RecSys.

[12]  Kun Guo,et al.  Data mining for the online retail industry: A case study of RFM model-based customer segmentation using data mining , 2012 .

[13]  Yves Grandvalet,et al.  A Coverage-Based Approach to Recommendation Diversity On Similarity Graph , 2016, RecSys.

[14]  Ben Taskar,et al.  Expectation-Maximization for Learning Determinantal Point Processes , 2014, NIPS.

[15]  Geert Wets,et al.  Using association rules for product assortment decisions: a case study , 1999, KDD '99.

[16]  A. Vershik,et al.  Asymptotic combinatorics with applications to mathematical physics : a European Mathematical Summer School held at the Euler Institute, St. Petersburg, Russia, July 9-20, 2001 , 2003 .

[17]  Choon Hui Teo,et al.  Adaptive, Personalized Diversity for Visual Discovery , 2016, RecSys.

[18]  Ulrich Paquet,et al.  Bayesian Low-Rank Determinantal Point Processes , 2016, RecSys.

[19]  Greg Linden,et al.  Amazon . com Recommendations Item-to-Item Collaborative Filtering , 2001 .

[20]  Das Amrita,et al.  Mining Association Rules between Sets of Items in Large Databases , 2013 .

[21]  Ben Taskar,et al.  Discovering Diverse and Salient Threads in Document Collections , 2012, EMNLP.

[22]  Ulrich Paquet,et al.  One-class collaborative filtering with random graphs , 2013, WWW '13.

[23]  Abraham Bernstein,et al.  Fewer Flops at the Top: Accuracy, Diversity, and Regularization in Two-Class Collaborative Filtering , 2017, RecSys.

[24]  Ben Taskar,et al.  Learning the Parameters of Determinantal Point Process Kernels , 2014, ICML.

[25]  Jinwoo Shin,et al.  Faster Greedy MAP Inference for Determinantal Point Processes , 2017, ICML.

[26]  David M. Blei,et al.  Scalable Recommendation with Poisson Factorization , 2013, ArXiv.

[27]  Saul Vargas,et al.  Improving sales diversity by recommending users to items , 2014, RecSys '14.

[28]  Ulrich Güntzer,et al.  Algorithms for association rule mining — a general survey and comparison , 2000, SKDD.

[29]  Hans-Peter Kriegel,et al.  A Three-Way Model for Collective Learning on Multi-Relational Data , 2011, ICML.