作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程 ›› 2026, Vol. 52 ›› Issue (6): 132-140. doi: 10.19678/j.issn.1000-3428.0070193

• 计算智能与模式识别 • 上一篇    下一篇

基于随机自注意力和动量对比学习的自监督序列推荐方法

余正涛1,2,*(), 孙资钦1,2, 张勇丙1,2, 高盛祥1,2, 黄于欣1,2, 谭凯文1,2   

  1. 1. 昆明理工大学信息工程与自动化学院, 云南 昆明 650000
    2. 云南省人工智能重点实验室, 云南 昆明 650000
  • 收稿日期:2024-08-05 修回日期:2024-10-27 出版日期:2026-06-15 发布日期:2026-06-02
  • 通讯作者: 余正涛
  • 作者简介:

    余正涛,男,教授、博士,主研方向为人工智能、自然语言处理

    孙资钦,硕士

    张勇丙,博士

    高盛祥,教授、博士

    黄于欣,教授、博士

    谭凯文,副教授、博士

  • 基金资助:
    国家自然科学基金联合基金重点项目(U23A20388); 国家自然科学基金(U21B2027); 国家自然科学基金(62376111); 国家自然科学基金(62266028); 国家自然科学基金(62266027); 云南省重点研发计划(202303AP140008); 云南省重点研发计划(202401BC070021); 云南省重点研发计划(202103AA080015); 云南省科技人才与平台计划(202105AC160018); 云南省基础研究项目(202301AT070393)

Self-Supervised Sequence Recommendation Method Based on Random Self-Attention and Momentum Contrastive Learning

YU Zhengtao1,2,*(), SUN Ziqin1,2, ZHANG Yongbing1,2, GAO Shengxiang1,2, HUANG Yuxin1,2, TAN Kaiwen1,2   

  1. 1. School of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650000, Yunnan, China
    2. Yunnan Key Laboratory of Artificial Intelligence, Kunming 650000, Yunnan, China
  • Received:2024-08-05 Revised:2024-10-27 Online:2026-06-15 Published:2026-06-02
  • Contact: YU Zhengtao

摘要:

序列推荐利用用户历史序列行为建模用户兴趣并进行内容推荐, 常被用于新闻、广告和电子商务等领域。基于对比学习的自监督序列推荐是当前研究热点, 然而, 真实的序列数据具有动态不确定, 且对比学习中存在采样偏移问题, 限制了推荐的性能。为了缓解这些问题, 提出基于随机自注意力和动量对比学习的自监督序列推荐方法, 其中随机自注意力用于缓解序列动态不确定问题, 动量对比学习用于缓解对比学习中存在采样偏移问题。为验证模型性能, 在Beauty、Office、Yelp和Toys 4个常用数据集上的实验结果表明, 该方法在HR@K、NDCG@K等多个指标上均优于其他基线模型, 展示了该方法在准确性和鲁棒性方面的显著提升。

关键词: 序列推荐, 动量对比学习, Wasserstein距离, 自监督学习, 自注意力

Abstract:

Sequence recommendation utilizes user historical sequence behavior to model user interests and provide content recommendations, and is commonly employed in sectors such as news, advertising, and e-commerce. Self-supervised sequence recommendation based on contrastive learning is a current research hotspot. However, real sequence data are dynamically uncertain, and sampling biases exist in contrastive learning, which limit the performance of recommendations. To mitigate these issues, this paper proposes a self-supervised sequence recommendation method based on stochastic self-attention and momentum contrastive learning. Stochastic self-attention is used to alleviate the uncertainty of sequence dynamics, and momentum contrastive learning is used to mitigate the sampling bias problem in contrastive learning. To validate the performance of the model, experiments are conducted on three datasets: Beauty, Office, Yelp, and Toys. The results demonstrate that the proposed method outperforms other baseline models across several metrics, including HR@K and NDCG@K, indicating significant improvements in both accuracy and robustness.

Key words: sequence recommendation, momentum contrastive learning, Wasserstein distance, self-supervised learning, self-attention