Accurate Bundle Matching and Generation via Multitask Learning with Partially Shared Parameters

TL;DR

This paper introduces BundleMage, a multi-task learning model that improves the accuracy of bundle matching and generation in e-commerce by effectively handling heterogeneous data and personalizing bundle recommendations.

Contribution

BundleMage employs an adaptive gating mechanism and partially shared parameters in multi-task learning to enhance bundle matching and generation accuracy.

Findings

Up to 6.6% higher nDCG in bundle matching.

6.3x higher nDCG in bundle generation.

Effective personalization of generated bundles.

Abstract

How can we recommend existing bundles to users accurately? How can we generate new tailored bundles for users? Recommending a bundle, or a group of various items, has attracted widespread attention in e-commerce owing to the increased satisfaction of both users and providers. Bundle matching and bundle generation are two representative tasks in bundle recommendation. The bundle matching task is to correctly match existing bundles to users while the bundle generation is to generate new bundles that users would prefer. Although many recent works have developed bundle recommendation models, they fail to achieve high accuracy since they do not handle heterogeneous data effectively and do not learn a method for customized bundle generation. In this paper, we propose BundleMage, an accurate approach for bundle matching and generation. BundleMage effectively mixes user preferences of items and bundles using an adaptive gate technique to achieve high accuracy for the bundle matching. BundleMage also generates a personalized bundle by learning a generation module that exploits a user preference and the characteristic of a given incomplete bundle to be completed. BundleMage further improves its performance using multi-task learning with partially shared parameters. Through extensive experiments, we show that BundleMage achieves up to 6.6% higher nDCG in bundle matching and 6.3x higher nDCG in bundle generation than the best competitors. We also provide qualitative analysis that BundleMage effectively generates bundles considering both the tastes of users and the characteristics of target bundles.