School of Information Science and Engineering, Yunnan University, Kunming 650091, China.
School of Mathematics, Southeast University, Nanjing 210096, China.
Neural Netw. 2020 Apr;124:308-318. doi: 10.1016/j.neunet.2020.01.021. Epub 2020 Jan 30.
In recommender systems, matrix factorization and its variants have grown up to be dominant in collaborative filtering due to their simplicity and effectiveness. In matrix factorization based methods, dot product which is actually used as a measure of distance from users to items, does not satisfy the inequality property, and thus may fail to capture the inner grained preference information and further limits the performance of recommendations. Metric learning produces distance functions that capture the essential relationships among rating data and has been successfully explored in collaborative recommendations. However, without the global statistical information of user-user pairs and item-item pairs, it makes the model easy to achieve a suboptimal metric. For this, we present a co-occurrence embedding regularized metric learning model (CRML) for collaborative recommendations. We consider the optimization problem as a multi-task learning problem which includes optimizing a primary task of metric learning and two auxiliary tasks of representation learning. In particular, we develop an effective approach for learning the embedding representations of both users and items, and then exploit the strategy of soft parameter sharing to optimize the model parameters. Empirical experiments on four datasets demonstrate that the CRML model can enhance the naive metric learning model and significantly outperforms the state-of-the-art methods in terms of accuracy of collaborative recommendations.
在推荐系统中,由于其简单性和有效性,矩阵分解及其变体在协同过滤中已经发展成为主导方法。在基于矩阵分解的方法中,实际上用作用户与项目之间距离度量的点积不满足不等式性质,因此可能无法捕获细微的偏好信息,并进一步限制推荐的性能。度量学习生成的距离函数可以捕获评分数据之间的基本关系,并已在协同推荐中得到成功探索。然而,没有用户-用户对和项目-项目对的全局统计信息,这使得模型容易达到次优度量。为此,我们提出了一种协同出现嵌入正则化度量学习模型(CRML)用于协同推荐。我们将优化问题视为多任务学习问题,包括优化度量学习的主要任务和表示学习的两个辅助任务。特别地,我们开发了一种有效的方法来学习用户和项目的嵌入表示,然后利用软参数共享策略来优化模型参数。在四个数据集上的实验表明,CRML 模型可以增强朴素度量学习模型,并在协同推荐的准确性方面显著优于最新方法。