Basket Completion with Multi-task Determinantal Point Processes

05/24/2018
by   Romain Warlop, et al.
0

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.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
06/22/2022

Multi-task twin support vector machine with Universum data

Multi-task learning (MTL) has emerged as a promising topic of machine le...
research
03/25/2020

Adversarial Multi-Binary Neural Network for Multi-class Classification

Multi-class text classification is one of the key problems in machine le...
research
05/02/2023

Multi-Task Multi-Behavior MAP-Elites

We propose Multi-Task Multi-Behavior MAP-Elites, a variant of MAP-Elites...
research
04/03/2017

Multi-Task Learning of Keyphrase Boundary Classification

Keyphrase boundary classification (KBC) is the task of detecting keyphra...
research
02/07/2023

Multi-Task Recommendations with Reinforcement Learning

In recent years, Multi-task Learning (MTL) has yielded immense success i...
research
05/22/2017

Multi-output Polynomial Networks and Factorization Machines

Factorization machines and polynomial networks are supervised polynomial...
research
04/13/2018

Comparatives, Quantifiers, Proportions: A Multi-Task Model for the Learning of Quantities from Vision

The present work investigates whether different quantification mechanism...

Please sign up or login with your details

Forgot password? Click here to reset