Distributionally Robust Graph-based Recommendation System

Bohao Wang,Jiawei Chen,Changdong Li,Sheng Zhou,Qihao Shi,Yang Gao,Yan Feng,Chun Chen,Can Wang
DOI: https://doi.org/10.1145/3589334.3645598
2024-01-01
Abstract:With the capacity to capture high-order collaborative signals, Graph NeuralNetworks (GNNs) have emerged as powerful methods in Recommender Systems (RS).However, their efficacy often hinges on the assumption that training andtesting data share the same distribution (a.k.a. IID assumption), and exhibitssignificant declines under distribution shifts. Distribution shifts commonlyarises in RS, often attributed to the dynamic nature of user preferences orubiquitous biases during data collection in RS. Despite its significance,researches on GNN-based recommendation against distribution shift are stillsparse. To bridge this gap, we propose Distributionally Robust GNN (DR-GNN)that incorporates Distributional Robust Optimization (DRO) into the GNN-basedrecommendation. DR-GNN addresses two core challenges: 1) To enable DRO to caterto graph data intertwined with GNN, we reinterpret GNN as a graph smoothingregularizer, thereby facilitating the nuanced application of DRO; 2) Given thetypically sparse nature of recommendation data, which might impede robustoptimization, we introduce slight perturbations in the training distribution toexpand its support. Notably, while DR-GNN involves complex optimization, it canbe implemented easily and efficiently. Our extensive experiments validate theeffectiveness of DR-GNN against three typical distribution shifts. The code isavailable at https://github.com/WANGBohaO-jpg/DR-GNN .
What problem does this paper attempt to address?