Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

LinRec: Linear Attention Mechanism for Long-term Sequential Recommender Systems

About

Transformer models have achieved remarkable success in sequential recommender systems (SRSs). However, computing the attention matrix in traditional dot-product attention mechanisms results in a quadratic complexity with sequence lengths, leading to high computational costs for long-term sequential recommendation. Motivated by the above observation, we propose a novel L2-Normalized Linear Attention for the Transformer-based Sequential Recommender Systems (LinRec), which theoretically improves efficiency while preserving the learning capabilities of the traditional dot-product attention. Specifically, by thoroughly examining the equivalence conditions of efficient attention mechanisms, we show that LinRec possesses linear complexity while preserving the property of attention mechanisms. In addition, we reveal its latent efficiency properties by interpreting the proposed LinRec mechanism through a statistical lens. Extensive experiments are conducted based on two public benchmark datasets, demonstrating that the combination of LinRec and Transformer models achieves comparable or even superior performance than state-of-the-art Transformer-based SRS models while significantly improving time and memory efficiency.

Langming Liu, Xiangyu Zhao, Chi Zhang, Jingtong Gao, Wanyu Wang, Wenqi Fan, Yiqi Wang, Ming He, Zitao Liu, Qing Li• 2024

Related benchmarks

TaskDatasetResultRank
RecommendationGowalla
Recall@200.7186
100
Sequential RecommendationML 1M
NDCG@100.6078
49
Sequential RecommendationAmazon Beauty
Recall@1046.74
48
Sequential RecommendationKuaiRand
HR@107.99
22
Sequential RecommendationBeauty
Memory (MB)5.27e+3
18
Sequential RecommendationML-20M
Memory (MB)3.02e+3
18
Sequential RecommendationAmazon Video-Games
NDCG@100.5313
13
RecommendationGowalla
Runtime110
12
RecommendationDIGINETICA
MRR@1015.16
7
RecommendationYoochoose
MRR@1034.67
7
Showing 10 of 17 rows

Other info

Follow for update