Current browse context:
stat.ML
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: CrossWalk: Fairness-enhanced Node Representation Learning
(Submitted on 6 May 2021 (v1), last revised 25 Mar 2022 (this version, v2))
Abstract: The potential for machine learning systems to amplify social inequities and unfairness is receiving increasing popular and academic attention. Much recent work has focused on developing algorithmic tools to assess and mitigate such unfairness. However, there is little work on enhancing fairness in graph algorithms. Here, we develop a simple, effective and general method, CrossWalk, that enhances fairness of various graph algorithms, including influence maximization, link prediction and node classification, applied to node embeddings. CrossWalk is applicable to any random walk based node representation learning algorithm, such as DeepWalk and Node2Vec. The key idea is to bias random walks to cross group boundaries, by upweighting edges which (1) are closer to the groups' peripheries or (2) connect different groups in the network. CrossWalk pulls nodes that are near groups' peripheries towards their neighbors from other groups in the embedding space, while preserving the necessary structural properties of the graph. Extensive experiments show the effectiveness of our algorithm to enhance fairness in various graph algorithms, including influence maximization, link prediction and node classification in synthetic and real networks, with only a very small decrease in performance.
Submission history
From: Ahmad Khajehnejad [view email][v1] Thu, 6 May 2021 14:45:34 GMT (12474kb,D)
[v2] Fri, 25 Mar 2022 13:19:23 GMT (12558kb,D)
Link back to: arXiv, form interface, contact.