通过反事实数据增强提升图神经网络公平性
Improving Fairness in Graph Neural Networks via Counterfactual Debiasing
- 用反事实构建多样化邻居,避免敏感信息误删
- 在三个主流GNN模型上显著提升公平性指标
- 适合关注算法公平性的图学习研究者使用
图神经网络(GNN)在建模图结构数据方面表现优异,但其预测可能因种族、性别等属性产生偏见。这种偏见会因图结构和消息传递机制而加剧。现有方法通过边删除或特征掩码过滤敏感信息,但可能同时移除非敏感特征,损害准确率与公平性的平衡。为此,本文提出一种新方法Fair-ICD,利用反事实数据增强在消息传递前生成多样化的节点邻域,使模型学习无偏表示。随后,采用对抗判别器进一步减少分类器的偏见。在标准数据集上使用三种GNN骨干网络的实验表明,Fair-ICD显著提升公平性指标,同时保持高预测性能。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) have been successful in modeling graph-structured data. However, similar to other machine learning models, GNNs can exhibit bias in predictions based on attributes like race and gender. Moreover, bias in GNNs can be exacerbated by the graph structure and message-passing mechanisms. Recent cutting-edge methods propose mitigating bias by filtering out sensitive information from input or representations, like edge dropping or feature masking. Yet, we argue that such strategies may unintentionally eliminate non-sensitive features, leading to a compromised balance between predictive accuracy and fairness. To tackle this challenge, we present a novel approach utilizing counterfactual data augmentation for bias mitigation. This method involves creating diverse neighborhoods using counterfactuals before message passing, facilitating unbiased node representations learning from the augmented graph. Subsequently, an adversarial discriminator is employed to diminish bias in predictions by conventional GNN classifiers. Our proposed technique, Fair-ICD, ensures the fairness of GNNs under moderate conditions. Experiments on standard datasets using three GNN backbones demonstrate that Fair-ICD notably enhances fairness metrics while preserving high predictive performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。