通过对比学习提升推荐系统在噪声数据下的鲁棒性
Contrastive Matrix Completion with Denoising and Augmented Graph Views for Robust Recommendation
- 构建双视图图表示:一个去噪,一个分布对齐
- 在真实数据集上使RMSE降低0.8%,排名指标提升36%
- 适合处理含噪声的推荐场景,尤其对过拟合敏感任务
矩阵补全是推荐系统中的常用框架,通过预测用户-物品评分矩阵中的缺失条目来全面理解用户偏好。然而,当前基于图神经网络(GNN)的方法因消息传递机制对噪声或无关边高度敏感,易过拟合,限制了泛化能力。为此,我们提出一种新方法MCCL:为每个交互提取局部邻域子图,生成两个不同图表示。第一种通过融合GNN与注意力机制实现去噪;第二种采用图变分自编码器,使特征分布逼近标准先验。训练中使用互学习损失函数逐步对齐两种表示,从而捕捉共性模式,显著提升模型泛化性。在多个真实数据集上的实验表明,该方法不仅提升了预测得分的数值精度(RMSE最高降低0.8%),还使排名指标最高提升36%。
原文摘要 · Abstract (English)
Matrix completion is a widely adopted framework in recommender systems, as predicting the missing entries in the user-item rating matrix enables a comprehensive understanding of user preferences. However, current graph neural network (GNN)-based approaches are highly sensitive to noisy or irrelevant edges--due to their inherent message-passing mechanisms--and are prone to overfitting, which limits their generalizability. To overcome these challenges, we propose a novel method called Matrix Completion using Contrastive Learning (MCCL). Our approach begins by extracting local neighborhood subgraphs for each interaction and subsequently generates two distinct graph representations. The first representation emphasizes denoising by integrating GNN layers with an attention mechanism, while the second is obtained via a graph variational autoencoder that aligns the feature distribution with a standard prior. A mutual learning loss function is employed during training to gradually harmonize these representations, enabling the model to capture common patterns and significantly enhance its generalizability. Extensive experiments on several real-world datasets demonstrate that our approach not only improves the numerical accuracy of the predicted scores--achieving up to a 0.8% improvement in RMSE--but also produces superior rankings with improvements of up to 36% in ranking metrics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。