Distributionally Robust Graph Out-of-Distribution Recommendation via Diffusion Model
Chu Zhao, Enneng Yang, Yuliang Liang, Jianzhe Zhao, Guibing Guo, Xingwei Wang
Abstract
The distributionally robust optimization (DRO)-based graph neural network methods improve recommendation systems' out-ofdistribution (OOD) generalization by optimizing the model's worstcase performance. However, these studies fail to consider the impact of noisy samples in the training data, which results in diminished generalization capabilities and lower accuracy. Through experimental and theoretical analysis, this paper reveals that current DRO-based graph recommendation methods assign greater weight to noise distribution, leading to model parameter learning being dominated by it. When the model overly focuses on fitting noise samples in the training data, it may learn irrelevant or meaningless features that cannot be generalized to OOD data. To address this challenge, we design a Distributionally Robust Graph model for OOD recommendation (DRGO). Specifically, our method first employs a simple and effective diffusion paradigm to alleviate the noisy effect in the latent space. Additionally, an entropy regularization term is introduced in the DRO objective function to avoid extreme sample weights in the worst-case distribution. Finally, we provide a theoretical proof of the generalization error bound of DRGO as well as a theoretical analysis of how our approach mitigates noisy sample effects, which helps to better understand the proposed framework from a theoretical perspective. We conduct extensive experiments on four datasets to evaluate the effectiveness of our framework against three typical distribution shifts, and the results demonstrate its superiority in both independently and identically distributed distributions (IID) and OOD. Our code is available at https://github.com/user683/DRGO . CCS CONCEPTS • Information systems → Recommender systems.
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
- Fading to Grow: Growing Preference Ratios via Preference Fading Discrete Diffusion for RecommendationGuoqing Hu, An Zhang, Shuchang Liu, Wenyu Mao et al.NeurIPS 2025 · 4 citations
- Causal Direct Preference Optimization for Distributionally Robust Generative RecommendationChu Zhao, Enneng Yang, Jianzhe Zhao, Guibing GuoICML 2026
- Uncertainty-Constrained Trustworthiness for Graph LearningChunhui Zhang, Pengqi Li, Lizhong Ding, Ye Yuan et al.ICML 2026
- Post-hoc Popularity Bias Correction in GNN-based Collaborative FilteringMd Aminul Islam, Elena Zheleva, Ren WangWWW 2026
- From Distribution to Geometry: Stable Graph Generalization via Invariant BarycentersHangyuan Du, Rong Wang, Weihong Zhang, Lu Bai et al.ICML 2026
Builds on22
- 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
- LightGCL: Simple Yet Effective Graph Contrastive Learning for RecommendationXuheng Cai, Chao Huang, Lianghao Xia, Xubin RenICLR 2023 · 99 citations
Related papers
- Distributionally Robust Graph-based Recommendation SystemBohao Wang, Jiawei Chen, Changdong Li, Sheng Zhou et al.WWW 2024 · 42 citations
- Graph Representation Learning via Causal Diffusion for Out-of-Distribution RecommendationChu Zhao, Enneng Yang, Yuliang Liang, Pengxiang Lan et al.WWW 2025 · 21 citations
- Geometry-Calibrated DRO: Combating Over-Pessimism with Free Energy ImplicationsJiashuo Liu, Jiayun Wu, Tianyu Wang, Hao Zou et al.ICML 2024 · 5 citations
- Distributionally Robust Optimization via Generative Ambiguity ModelingJiaqi Wen, Jianyi YangICLR 2026 · 3 citations
- A Generic Learning Framework for Sequential Recommendation with Distribution ShiftsZhengyi Yang, Xiangnan He, Jizhi Zhang, Jiancan Wu et al.SIGIR 2023 · 55 citations
