Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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.

Yukuo Cen, Jianwei Zhang, Xu Zou, Chang Zhou, Hongxia Yang, Jie Tang• 2020

Related benchmarks

TaskDatasetResultRank
RankingAMAZON
Recall88.7
12
RetrievalMovieLens (test)
IC@200.844
12
RetrievalTaobao (test)
IC@200.284
12
Recommendation RetrievalTaobao Hit@2k
Base@1k93
5
Recommendation RetrievalKuairand Hit@2k
Base@1k1.17
5
Recommendation RetrievalKuairand Hit@4k
Base@2k1.67
5
Recommendation RetrievalTaobao Hit@4k
Base@1k1.44
5
Showing 7 of 7 rows

Other info

Follow for update