Graph Representation Learning via Causal Diffusion for Out-of-Distribution Recommendation
Chu Zhao, Enneng Yang, Yuliang Liang, Pengxiang Lan, Yuting Liu, Jianzhe Zhao, Guibing Guo, Xingwei Wang
Abstract
Graph Neural Networks (GNNs)-based recommendation algorithms typically assume that training and testing data are drawn from independent and identically distributed (IID) spaces. However, this assumption often fails in the presence of out-of-distribution (OOD) data, resulting in significant performance degradation. In this study, we construct a Structural Causal Model (SCM) to analyze interaction data, revealing that environmental confounders (e.g., the COVID-19 pandemic) lead to unstable correlations in GNN-based models, thus impairing their generalization to OOD data. To address this issue, we propose a novel approach, graph representation learning via causal diffusion (CausalDiffRec) for OOD recommendation. This method enhances the model's generalization on OOD data by eliminating environmental confounding factors and learning invariant graph representations. Specifically, we use backdoor adjustment and variational inference to infer the real environmental distribution, thereby eliminating the impact of environmental confounders. This inferred distribution is then used as prior knowledge to guide the representation learning in the reverse phase of the diffusion process to learn the invariant representation. In addition,we provide a theoretical derivation that proves optimizing the objective function of CausalDiffRec can encourage the model to learn environment-invariant graph representations, thereby achieving excellent generalization performance in recommendations under distribution shifts. Our extensive experiments validate the effectiveness of CausalDiffRec in improving the generalization of OOD data, and the average improvement is up to 10.69% on Food, 18.83% on KuaiRec, 22.41% on Yelp2018, and 11.65% on Douban datasets.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers5
- Hyperbolic Diffusion Recommender ModelMeng Yuan, Yutian Xiao, Wei Chen, Chou Zhao et al.WWW 2025 · 13 citations
- Distributionally Robust Graph Out-of-Distribution Recommendation via Diffusion ModelChu Zhao, Enneng Yang, Yuliang Liang, Jianzhe Zhao et al.WWW 2025 · 11 citations
- Modeling Item-Level Dynamic Variability with Residual Diffusion for Bundle RecommendationDong Zhang, Lin Li, Ming Li, Amran Bhuiyan et al.AAAI 2026 · 2 citations
- How Does Topology Bias Distort Message Passing in Graph Recommender? A Dirichlet Energy PerspectiveYanbiao Ji, Yue Ding, Dan Luo, Chang Liu et al.NeurIPS 2025 · 2 citations
- Post-hoc Popularity Bias Correction in GNN-based Collaborative FilteringMd Aminul Islam, Elena Zheleva, Ren WangWWW 2026
Builds on30
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- LightGCN: Simplifying and Powering Graph Convolution Network for RecommendationXiangnan He, Kuan Deng, Xiang Wang, Yan Li et al.SIGIR 2020 · 4,448 citations
- Self-supervised Graph Learning for RecommendationJiancan Wu, Xiang Wang, Fuli Feng, Xiangnan He et al.SIGIR 2021 · 1,476 citations
- Are Graph Augmentations Necessary?: Simple Graph Contrastive Learning for RecommendationJunliang Yu, Hongzhi Yin, Xin Xia, Tong Chen et al.SIGIR 2022 · 658 citations
- Disentangled Graph Collaborative FilteringXiang Wang, Hongye Jin, An Zhang, Xiangnan He et al.SIGIR 2020 · 621 citations
Related papers
- Graph Out-of-Distribution Generalization via Causal InterventionQitian Wu, Fan Nie, Chenxiao Yang, Tianyi Bao et al.WWW 2024 · 58 citations
- Causal Direct Preference Optimization for Distributionally Robust Generative RecommendationChu Zhao, Enneng Yang, Jianzhe Zhao, Guibing GuoICML 2026
- Invariant Learning on Heterogeneous Graphs via Subgraph Environment InferenceYanghui Fu, Yunfei Wang, Hao Zou, Yue He et al.WWW 2026
- GraphDE: A Generative Framework for Debiased Learning and Out-of-Distribution Detection on GraphsZenan Li, Qitian Wu, Fan Nie, Junchi YanNeurIPS 2022 · 75 citations
- Improving Generalization of Dynamic Graph Learning via Environment PromptKuo Yang, Zhengyang Zhou, Qihe Huang, Limin Li et al.NeurIPS 2024 · 14 citations
