Controllable Multi-Interest Framework for Recommendation
About
Recently, neural networks have been widely used in e-commerce recommender systems, owing to the rapid development of deep learning. We formalize the recommender system as a sequential recommendation problem, intending to predict the next items that the user might be interacted with. Recent works usually give an overall embedding from a user's behavior sequence. However, a unified user embedding cannot reflect the user's multiple interests during a period. In this paper, we propose a novel controllable multi-interest framework for the sequential recommendation, called ComiRec. Our multi-interest module captures multiple interests from user behavior sequences, which can be exploited for retrieving candidate items from the large-scale item pool. These items are then fed into an aggregation module to obtain the overall recommendation. The aggregation module leverages a controllable factor to balance the recommendation accuracy and diversity. We conduct experiments for the sequential recommendation on two real-world datasets, Amazon and Taobao. Experimental results demonstrate that our framework achieves significant improvements over state-of-the-art models. Our framework has also been successfully deployed on the offline Alibaba distributed cloud platform.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multi-objective Re-ranking | ML 1M | HR@570.05 | 13 | |
| Multi-objective Re-ranking | Grocery | HR@538 | 13 | |
| Multi-objective Re-ranking | Beauty | Hit Rate @ 535.73 | 13 | |
| Sequential Recommendation | Yelp | Recall@206.25 | 13 | |
| Sequential Recommendation | Beauty | Recall@206.5 | 13 | |
| Sequential Recommendation | Retail Rocket | Recall@2010.35 | 13 | |
| Sequential Recommendation | Gowalla | Recall@206.23 | 13 | |
| Sequential Recommendation | Amazon Books | -- | 13 | |
| Ranking | AMAZON | Recall88.7 | 12 | |
| Retrieval | MovieLens (test) | IC@200.844 | 12 |