LightSANs
概述
LightSANs(Lightweight Self-Attentive Network),2021 年提出的轻量级自注意力序列推荐模型,在保持 SASRec 性能的同时显著减少参数量和计算开销。
关键内容
-
论文信息:LightSANs(Lightweight Self-Attentive Network for Sequential Recommendation),发表于 SIGIR 2021。
-
核心创新:针对 SASRec 中 $O(n^2 d)$ 的自注意力复杂度问题,提出轻量化的自注意力机制。通过低秩近似或核方法减少注意力计算的参数量和计算量。
-
方法:使用用户兴趣表示作为 Query,物品表示作为 Key/Value,将注意力计算从序列级别压缩到兴趣级别,大幅减少计算开销。
-
效果:在多个数据集上达到与 SASRec 相当的性能,同时参数量和推理速度显著优化,更适合资源受限的部署场景。