arXiv:2502.03664cs.IRcs.LG2025-02被引 19

用对比学习提升冷启动推荐效果,自适应融合多源特征。

Contrastive Learning for Cold Start Recommendation with Adaptive Feature Fusion

  • 自适应特征选择模块动态调整关键特征权重。
  • 在MovieLens-1M上各项指标均显著优于主流方法。
  • 适合冷启动、跨域等数据稀缺场景的推荐系统应用。

本文提出一种融合对比学习的冷启动推荐模型,旨在解决用户与物品交互数据稀少导致推荐性能下降的问题。该模型通过自适应特征选择模块动态调整关键特征权重,并结合多模态特征融合机制,有效整合用户属性、物品元信息和上下文特征,从而提升推荐效果。同时引入对比学习机制,通过构建正负样本对增强特征表示的鲁棒性与泛化能力。在MovieLens-1M数据集上的实验表明,所提模型在HR、NDCG、MRR和Recall等指标上显著优于矩阵分解、LightGBM、DeepFM和AutoRec等主流方法,尤其在冷启动场景下表现突出。消融实验验证了各模块的关键作用,学习率敏感性分析表明适度的学习率对模型优化至关重要。本研究不仅为冷启动问题提供新解法,也为对比学习在推荐系统中的应用提供重要参考。未来有望拓展至实时推荐与跨域推荐等更广泛场景。

原文摘要 · Abstract (English)

This paper proposes a cold start recommendation model that integrates contrastive learning, aiming to solve the problem of performance degradation of recommendation systems in cold start scenarios due to the scarcity of user and item interaction data. The model dynamically adjusts the weights of key features through an adaptive feature selection module and effectively integrates user attributes, item meta-information, and contextual features by combining a multimodal feature fusion mechanism, thereby improving recommendation performance. In addition, the model introduces a contrastive learning mechanism to enhance the robustness and generalization ability of feature representation by constructing positive and negative sample pairs. Experiments are conducted on the MovieLens-1M dataset. The results show that the proposed model significantly outperforms mainstream recommendation methods such as Matrix Factorization, LightGBM, DeepFM, and AutoRec in terms of HR, NDCG, MRR, and Recall, especially in cold start scenarios. Ablation experiments further verify the key role of each module in improving model performance, and the learning rate sensitivity analysis shows that a moderate learning rate is crucial to the optimization effect of the model. This study not only provides a new solution to the cold start problem but also provides an important reference for the application of contrastive learning in recommendation systems. In the future, this model is expected to play a role in a wider range of scenarios, such as real-time recommendation and cross-domain recommendation.

冷启动对比学习特征融合

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