超越自注意力的序列推荐归纳偏置
机器学习
2024-02-20 v2 人工智能
信息检索
摘要
基于 Transformer 的序列推荐 (SR) 模型已取得显著成功。Transformer 在计算机视觉和自然语言处理中的自注意力机制存在过平滑问题,即隐藏表示变得与 token 相似。在 SR 领域,我们首次证明了同样问题的存在。我们进行了开创性研究,揭示了 SR 中自注意力的低通滤波特性,该特性导致了过平滑。为此,我们提出了一种名为eyond elf-ttention for Sequential ommendation (BSARec) 的新方法,该方法利用傅里叶变换来:i) 通过考虑细粒度序列模式注入归纳偏置;ii) 整合低频和高频信息以缓解过平滑。我们的发现展示了 SR 领域的重大进步,有望弥合现有基于 Transformer 的 SR 模型之间的差距。我们在 6 个基准数据集上通过大量实验测试了所提出的方法。实验结果表明,我们的模型在推荐性能方面优于 7 种基线方法。我们的代码可在 https://github.com/yehjin-shin/BSARec 获取。
引用
@article{arxiv.2312.10325,
title = {An Attentive Inductive Bias for Sequential Recommendation beyond the Self-Attention},
author = {Yehjin Shin and Jeongwhan Choi and Hyowon Wi and Noseong Park},
journal= {arXiv preprint arXiv:2312.10325},
year = {2024}
}
备注
Accepted by AAAI 2024. Yehjin Shin and Jeongwhan Choi are co-first authors with equal contribution