arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.01931cs.IR
收藏

面向冷启动用户的锐度感知跨域推荐

Sharpness-Aware Cross-Domain Recommendation to Cold-Start Users

Guohang Zeng, Qian Zhang, Guangquan Zhang, Jie Lu

AI总结:

本文针对跨域推荐中重叠用户稀少导致模型收敛到尖锐极小值、泛化差的问题,提出锐度感知跨域推荐方法SCDR,通过同时优化推荐损失与损失锐度,显著提升冷启动推荐性能并增强对抗鲁棒性。

摘要:

跨域推荐(CDR)是一种受迁移学习启发、用于解决推荐系统中冷启动问题的有前景的范式。现有最先进的CDR方法训练一个显式映射函数,将冷启动用户从数据丰富的源域迁移到目标域。然而,这些方法的一个局限在于,映射函数是在跨域重叠用户上训练的,而可用于训练的重叠用户数量很少。通过可视化现有CDR模型的损失景观,我们发现,在少量重叠用户上训练会导致模型收敛到尖锐极小值,从而造成泛化能力较差。基于这一观察,我们利用基于损失几何的机器学习方法,提出了一种新的CDR方法,称为锐度感知CDR(SCDR)。我们提出的方法同时优化推荐损失和损失锐度,从而在具有理论保证的情况下获得更好的泛化能力。在真实世界数据集上的实证研究表明,SCDR在冷启动推荐任务上显著优于其他CDR模型,同时增强了模型对对抗攻击的鲁棒性。

英文摘要:

Cross-Domain Recommendation (CDR) is a promising paradigm inspired by transfer learning to solve the cold-start problem in recommender systems. Existing state-of-the-art CDR methods train an explicit mapping function to transfer the cold-start users from a data-rich source domain to a target domain. However, a limitation of these methods is that the mapping function is trained on overlapping users across domains, while only a small number of overlapping users are available for training. By visualizing the loss landscape of the existing CDR model, we find that training on a small number of overlapping users causes the model to converge to sharp minima, leading to poor generalization. Based on this observation, we leverage loss-geometry-based machine learning approach and propose a novel CDR method called Sharpness-Aware CDR (SCDR). Our proposed method simultaneously optimizes recommendation loss and loss sharpness, leading to better generalization with theoretical guarantees. Empirical studies on real-world datasets demonstrate that SCDR significantly outperforms the other CDR models for cold-start recommendation tasks, while concurrently enhancing the model's robustness to adversarial attacks.

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

↑