arXiv:2410.23757cs.IR2024-10NeurIPS被引 13

无需预设分组数,自动发现用户群并自监督推荐

Identify Then Recommend: Towards Unsupervised Group Recommendation

  • 先无监督聚类识别动态用户群,再通过自训练任务优化推荐
  • 在两个数据集上群体推荐指标提升超22%(NDCG@5)
  • 适合工业场景中实时变化的用户群推荐需求

群体推荐(GR)旨在为用户群体推荐项目,是推荐系统中极具前景的方向。现有方法存在两大问题:一是预设固定数量的用户群,难以适应工业场景中动态变化的群体分布;二是依赖昂贵的用户-群组和群组-项目标签,标注成本高。为此,本文提出新颖的无监督群体推荐框架ITR,先无监督地识别用户群(无需预设群数),再设计两项自训练任务进行自监督推荐。具体而言,在群组识别阶段,通过估计用户点的自适应密度,将高密度区域识别为群中心,并采用启发式合并与分裂策略发现群组及边界;在自监督学习阶段,提出拉-推预训练任务优化用户-群组分布,并设计伪群推荐任务辅助推荐。大量实验表明,ITR在用户推荐(如NDCG@5提升22.22%)和群体推荐(如提升22.95%)上均显著优于现有方法。此外,已在工业推荐系统中部署并取得良好效果。

原文摘要 · Abstract (English)

Group Recommendation (GR), which aims to recommend items to groups of users, has become a promising and practical direction for recommendation systems. This paper points out two issues of the state-of-the-art GR models. (1) The pre-defined and fixed number of user groups is inadequate for real-time industrial recommendation systems, where the group distribution can shift dynamically. (2) The training schema of existing GR methods is supervised, necessitating expensive user-group and group-item labels, leading to significant annotation costs. To this end, we present a novel unsupervised group recommendation framework named \underline{I}dentify \underline{T}hen \underline{R}ecommend (\underline{ITR}), where it first identifies the user groups in an unsupervised manner even without the pre-defined number of groups, and then two pre-text tasks are designed to conduct self-supervised group recommendation. Concretely, at the group identification stage, we first estimate the adaptive density of each user point, where areas with higher densities are more likely to be recognized as group centers. Then, a heuristic merge-and-split strategy is designed to discover the user groups and decision boundaries. Subsequently, at the self-supervised learning stage, the pull-and-repulsion pre-text task is proposed to optimize the user-group distribution. Besides, the pseudo group recommendation pre-text task is designed to assist the recommendations. Extensive experiments demonstrate the superiority and effectiveness of ITR on both user recommendation (e.g., 22.22\% NDCG@5 $\uparrow$) and group recommendation (e.g., 22.95\% NDCG@5 $\uparrow$). Furthermore, we deploy ITR on the industrial recommender and achieve promising results.

群体推荐无监督学习自监督

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。