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 | |
|---|---|---|---|---|
| Ranking | AMAZON | Recall88.7 | 12 | |
| Retrieval | MovieLens (test) | IC@200.844 | 12 | |
| Retrieval | Taobao (test) | IC@200.284 | 12 | |
| Recommendation Retrieval | Taobao Hit@2k | Base@1k93 | 5 | |
| Recommendation Retrieval | Kuairand Hit@2k | Base@1k1.17 | 5 | |
| Recommendation Retrieval | Kuairand Hit@4k | Base@2k1.67 | 5 | |
| Recommendation Retrieval | Taobao Hit@4k | Base@1k1.44 | 5 |