高级检索

融合超网络和项目属性的联邦推荐模型

Federated Recommendation Models Incorporating Hypernetworks and Item Attributes

  • 摘要: 传统的推荐系统可以利用机器学习来预测用户对项目的评分,但这类方法需要用户共享他们的数据,可能会引发潜在的隐私泄露问题。将联邦学习引入推荐系统中可以保护用户隐私,但目前的联邦推荐模型需要多次使用用户的项目交互数据来学习全局参数。因此,该文提出了一种融合超网络和项目属性的联邦推荐模型。首先,使用超网络快速初始化客户端的推荐模型,并将用户偏好嵌入作为超网络的输入,以获得用户的个性化偏好;然后,在客户端推荐模型中,使用项目属性的内容嵌入作为全局信息,以提供额外的上下文信息;最后,使用注意力残差块来学习不同项目属性的重要性。实验表明,该方法在Movielens1M、Hetrec-movielens和Douban数据集上与其他模型相比具有良好的推荐性能,平均MAE、RMSE和Accuracy分别提升约1.4%、2.0%和4.1%。

     

    Abstract: Traditional recommendation systems requires users to share their data, which can potentially lead to privacy concerns. Introducing federated learning into recommendation systems can protect user privacy, but current federated recommendation models require multiple interactions of user-item data to learn global parameters. This paper proposes a federated recommendation model that integrates hypernetworks and item attributes. Firstly, hypernetworks are used to rapidly initialize recommendation models on client devices, with user preferences embedded as input to obtain personalized user preferences. Then, in the client-side recommendation model, content embeddings of item attributes are used as global information to provide additional contextual information. Finally, attention residual blocks are employed to learn the importance of different item attributes. Experiments on the Movielens1M, Hetrec-movielens, and Douban datasets demonstrate that this approach outperforms other models with an average improvement of approximately 1.4% in MAE, 2.0% in RMSE, and 4.1% in accuracy.

     

/

返回文章
返回