CausalRec: A CausalBoost Attention Model for Sequential Recommendation
Yunbo Hou, Tianle Yang, Ruijie Li, Li He, Liang Wang, Weiping Li, Bo Zheng, Guojie Song
TL;DR
CausalRec introduces a causal-augmented attention framework for sequential recommendation, combining a Causal Discovery Block with identifiability guarantees and a CausalBooster that amplifies causally relevant signals in the attention mechanism. The model learns item-level causal graphs from sequences via covariance-based estimation under a linear SEM, imposes acyclicity and sparsity constraints, and integrates the learned causality into a transformer-style attention network. Empirical results on four real-world datasets show consistent improvements in NDCG@10 and HR@10 over state-of-the-art baselines, with notable gains on the Foursquare dataset, and ablations confirm the contributions of causal discovery, sparsity, and attention. The work advances the use of causality in recommender systems, offering improved predictive accuracy and explanations while addressing identifiability concerns and ensuring practical scalability.
Abstract
Recent advances in correlation-based sequential recommendation systems have demonstrated substantial success. Specifically, the attention-based model outperforms other RNN-based and Markov chains-based models by capturing both short- and long-term dependencies more effectively. However, solely focusing on item co-occurrences overlooks the underlying motivations behind user behaviors, leading to spurious correlations and potentially inaccurate recommendations. To address this limitation, we present a novel framework that integrates causal attention for sequential recommendation, CausalRec. It incorporates a causal discovery block and a CausalBooster. The causal discovery block learns the causal graph in user behavior sequences, and we provide a theory to guarantee the identifiability of the learned causal graph. The CausalBooster utilizes the discovered causal graph to refine the attention mechanism, prioritizing behaviors with causal significance. Experimental evaluations on real-world datasets indicate that CausalRec outperforms several state-of-the-art methods, with average improvements of 7.21% in Hit Rate (HR) and 8.65% in Normalized Discounted Cumulative Gain (NDCG). To the best of our knowledge, this is the first model to incorporate causality through the attention mechanism in sequential recommendation, demonstrating the value of causality in generating more accurate and reliable recommendations.
