arXiv:2501.00595cs.LGcs.AI2025-01

通过生成式子图扩散消除图神经网络偏见,提升公平性。

Unbiased GNN Learning via Fairness-Aware Subgraph Diffusion

  • 从大图中采样小子图,用随机微分方程进行公平感知的生成扩散去偏。
  • 在正向扩散中引入对抗性偏见扰动,训练得分模型学习数据中的偏见动态。
  • 通过反向扩散重构无偏子图,再用标准GNN实现公平预测,适合高敏感场景。

图神经网络(GNN)在众多图任务中表现优异,但其对年龄、性别等敏感节点属性易产生偏见。这种偏见因消息传递机制被放大,使公平预测难实现。本文提出生成式公平感知子图扩散(FASD)方法:先从原图中采样小子图,再基于随机微分方程(SDEs)进行生成式公平扩散去偏。正向过程中引入对抗性偏见扰动,训练得分模型预测这些扰动,以学习数据中的偏见动态;反向过程中利用训练好的模型对子图进行去偏重构。最终在去偏子图上运行标准GNN,实现公平节点预测。实验表明,FASD在多个基准数据集上优于现有最先进公平GNN方法。

原文摘要 · Abstract (English)

Graph Neural Networks (GNNs) have demonstrated remarkable efficacy in tackling a wide array of graph-related tasks across diverse domains. However, a significant challenge lies in their propensity to generate biased predictions, particularly with respect to sensitive node attributes such as age and gender. These biases, inherent in many machine learning models, are amplified in GNNs due to the message-passing mechanism, which allows nodes to influence each other, rendering the task of making fair predictions notably challenging. This issue is particularly pertinent in critical domains where model fairness holds paramount importance. In this paper, we propose a novel generative Fairness-Aware Subgraph Diffusion (FASD) method for unbiased GNN learning. The method initiates by strategically sampling small subgraphs from the original large input graph, and then proceeds to conduct subgraph debiasing via generative fairness-aware graph diffusion processes based on stochastic differential equations (SDEs). To effectively diffuse unfairness in the input data, we introduce additional adversary bias perturbations to the subgraphs during the forward diffusion process, and train score-based models to predict these applied perturbations, enabling them to learn the underlying dynamics of the biases present in the data. Subsequently, the trained score-based models are utilized to further debias the original subgraph samples through the reverse diffusion process. Finally, FASD induces fair node predictions on the input graph by performing standard GNN learning on the debiased subgraphs. Experimental results demonstrate the superior performance of the proposed method over state-of-the-art Fair GNN baselines across multiple benchmark datasets.

图神经网络公平性去偏生成模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。