arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2101.04849cs.IRcs.LG
收藏

Top-K推荐的自适应间隔概率度量学习

Probabilistic Metric Learning with Adaptive Margin for Top-K Recommendation

McGill University(麦吉尔大学);Huawei Noah’s Ark Lab(华为诺亚方舟实验室)

Chen Ma, Liheng Ma, Yingxue Zhang, Ruiming Tang, Xue Liu, Mark Coates

摘要:

随着越来越多的内容和服务变得可用,并且用户很难确定他们可能感兴趣的内容,个性化推荐系统正在发挥越来越重要的作用。虽然基于矩阵分解和深度学习的方法在用户偏好建模中被证明是有效的,但它们违反了三角不等式,无法捕获细粒度的偏好信息。为了解决这个问题,我们开发了一个基于距离的推荐模型,具有几个新的方面:(i)每个用户和项目参数化的高斯分布,以捕捉学习的不确定性;(ii)提出了一个自适应的边缘生成方案,以生成关于不同的训练三元组的边缘;(iii)明确的用户-用户/项目-项目相似性建模纳入目标函数。由于Wasserstein距离服从三角不等式,可以度量概率分布之间的距离,因此被用来确定偏好。通过使用五个真实世界数据集和最先进方法进行比较,该模型在Top-K推荐的@K召回率方面比现有最佳模型高出4-22%。

英文摘要:

Personalized recommender systems are playing an increasingly important role as more content and services become available and users struggle to identify what might interest them. Although matrix factorization and deep learning based methods have proved effective in user preference modeling, they violate the triangle inequality and fail to capture fine-grained preference information. To tackle this, we develop a distance-based recommendation model with several novel aspects: (i) each user and item are parameterized by Gaussian distributions to capture the learning uncertainties; (ii) an adaptive margin generation scheme is proposed to generate the margins regarding different training triplets; (iii) explicit user-user/item-item similarity modeling is incorporated in the objective function. The Wasserstein distance is employed to determine preferences because it obeys the triangle inequality and can measure the distance between probabilistic distributions. Via a comparison using five real-world datasets with state-of-the-art methods, the proposed model outperforms the best existing models by 4-22% in terms of recall@K on Top-K recommendation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑