arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2311.02446cs.IR

通过置信软标签学习鲁棒的序列推荐模型

Learning Robust Sequential Recommenders through Confident Soft Labels

Shiguang Wu, Xin Xin, Pengjie Ren, Zhumin Chen, Jun Ma, Maarten de Rijke, Zhaochun Ren

AI总结:

针对序列推荐中独热标签稀疏导致训练有偏的问题,提出含教师-学生模块的CSRec框架,通过三类教师模块生成置信软标签结合独热标签训练,在四个基准数据集上验证了其提升推荐性能的有效性。

摘要:

基于隐式反馈训练的序列推荐模型通常采用基于softmax的损失函数,以独热(one-hot)类别标签将其建模为多分类任务进行训练。然而,独热训练标签具有稀疏性,可能导致训练有偏、性能未达最优。已有研究表明,稠密软标签有助于提升推荐性能,但如何从用户与物品间含噪声的序列交互中生成高质量、高置信的软标签,仍是一个待解决的问题。\n 我们提出一种新的序列推荐学习框架CSRec,该框架引入置信软标签,为从用户-物品交互中学习的过程提供鲁棒的指导。CSRec包含一个教师模块(用于生成高质量、高置信的软标签)和一个学生模块(作为目标推荐模型,基于稠密软标签与稀疏独热标签的组合进行训练)。\n 我们提出并对比了三种构建教师模块的方法:(1)模型级、(2)数据级、(3)训练级。为评估CSRec的有效性与泛化能力,我们在四个基准数据集上,以多种当前最优的序列推荐模型作为目标学生模块开展实验。实验结果表明,CSRec可有效训练出性能更优的序列推荐模型。

英文摘要:

Sequential recommenders that are trained on implicit feedback are usually learned as a multi-class classification task through softmax-based loss functions on one-hot class labels. However, one-hot training labels are sparse and may lead to biased training and sub-optimal performance. Dense, soft labels have been shown to help improve recommendation performance. But how to generate high-quality and confident soft labels from noisy sequential interactions between users and items is still an open question. We propose a new learning framework for sequential recommenders, CSRec, which introduces confident soft labels to provide robust guidance when learning from user-item interactions. CSRec contains a teacher module that generates high-quality and confident soft labels and a student module that acts as the target recommender and is trained on the combination of dense, soft labels and sparse, one-hot labels. We propose and compare three approaches to constructing the teacher module: (i) model-level, (ii) data-level, and (iii) training-level. To evaluate the effectiveness and generalization ability of CSRec, we conduct experiments using various state-of-the-art sequential recommendation models as the target student module on four benchmark datasets. Our experimental results demonstrate that CSRec is effective in training better performing sequential recommenders.

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

↑