With the capacity to capture high-order collaborative signals, Graph Neural Networks (GNNs) have emerged as powerful methods in Recommender Systems (RS). However, their efficacy often hinges on the assumption that training and testing data share the same distribution (a.k.a. IID assumption), and exhibits significant declines under distribution shifts. Distribution shifts commonly arises in RS, often attributed to the dynamic nature of user preferences or ubiquitous biases during data collection in RS. Despite its significance, researches on GNN-based recommendation against distribution shift are still sparse. To bridge this gap, we propose Distributionally Robust GNN (DR-GNN) that incorporates Distributional Robust Optimization (DRO) into the GNN-based recommendation. DR-GNN addresses two core challenges: 1) To enable DRO to cater to graph data intertwined with GNN, we reinterpret GNN as a graph smoothing regularizer, thereby facilitating the nuanced application of DRO; 2) Given the typically sparse nature of recommendation data, which might impede robust optimization, we introduce slight perturbations in the training distribution to expand its support. Notably, while DR-GNN involves complex optimization, it can be implemented easily and efficiently. Our extensive experiments validate the effectiveness of DR-GNN against three typical distribution shifts. The code is available at https://github.com/WANGBohaO-jpg/DR-GNN .
翻译:摘要:凭借捕捉高阶协同信号的能力,图神经网络(GNN)已成为推荐系统(RS)中的强大方法。然而,其有效性往往依赖于训练数据和测试数据具有相同分布的假设(即独立同分布假设),并且在分布偏移下性能显著下降。分布偏移在推荐系统中普遍存在,通常归因于用户偏好的动态性或数据收集过程中的普遍偏差。尽管其重要性,针对分布偏移的基于GNN的推荐研究仍较为稀缺。为填补这一空白,我们提出分布鲁棒图神经网络(DR-GNN),将分布鲁棒优化(DRO)融入基于GNN的推荐中。DR-GNN解决两个核心挑战:1)为使DRO适用于与GNN交织的图数据,我们将GNN重新解释为图平滑正则化项,从而促进DRO的精细化应用;2)鉴于推荐数据通常具有稀疏性(可能阻碍鲁棒优化),我们在训练分布中引入轻微扰动以扩展其支撑集。值得注意的是,尽管DR-GNN涉及复杂优化,但其实现简便高效。大量实验验证了DR-GNN在三种典型分布偏移下的有效性。代码已开源至https://github.com/WANGBohaO-jpg/DR-GNN。