BeMap: Balanced Message Passing for Fair Graph Neural Network

Xiao Lin, Jian Kang, Weilin Cong, Hanghang Tong

Research output: Contribution to journalConference articlepeer-review

Abstract

Fairness in graph neural networks has been actively studied recently. However, existing works often do not explicitly consider the role of message passing in introducing or amplifying the bias. In this paper, we first investigate the problem of bias amplification in message passing. We empirically and theoretically demonstrate that message passing could amplify the bias when the 1-hop neighbors from different demographic groups are unbalanced. Guided by such analyses, we propose BeMap, a fair message passing method, that leverages a balance-aware sampling strategy to balance the number of the 1-hop neighbors of each node among different demographic groups. Extensive experiments on node classification demonstrate the efficacy of BeMap in mitigating bias while maintaining classification accuracy.

Original languageEnglish (US)
Pages (from-to)371-3725
Number of pages3355
JournalProceedings of Machine Learning Research
Volume231
StatePublished - 2023
Event2nd Learning on Graphs Conference, LOG 2023 - Virtual, Online
Duration: Nov 27 2023Nov 30 2023

ASJC Scopus subject areas

  • Artificial Intelligence
  • Software
  • Control and Systems Engineering
  • Statistics and Probability

Fingerprint

Dive into the research topics of 'BeMap: Balanced Message Passing for Fair Graph Neural Network'. Together they form a unique fingerprint.

Cite this