提出高效公平图聚类方法FairAD,解决大规模图中群体公平性问题。
FairAD: Computationally Efficient Fair Graph Clustering via Algebraic Distance
- 基于代数距离构建新相似矩阵,嵌入公平约束
- 图粗化找代表性节点,实现40倍加速
- 适合需快速公平聚类的大型图数据场景
随着对机器学习模型在特定人口群体上不公平行为的关注增加,公平性成为研究热点,推动了图聚类中的公平性研究。公平图聚类旨在将图中节点划分为k个互斥簇,使每个簇内受保护群体的比例与整体数据集中的比例一致。然而,将公平性约束融入现有图聚类算法对大规模图而言计算复杂。为此,我们提出FairAD,一种计算高效的公平图聚类方法。该方法首先基于代数距离构建新亲和矩阵以施加公平约束;随后对该矩阵进行图粗化,识别对应k个簇的代表性节点;最后求解带约束的最小化问题,获得公平聚类结果。在修改的随机块模型及六个公开数据集上的实验表明,FairAD在实现公平聚类的同时,相较当前最优方法速度提升最高达40倍。
原文摘要 · Abstract (English)
Due to the growing concern about unsavory behaviors of machine learning models toward certain demographic groups, the notion of 'fairness' has recently drawn much attention from the community, thereby motivating the study of fairness in graph clustering. Fair graph clustering aims to partition the set of nodes in a graph into $k$ disjoint clusters such that the proportion of each protected group within each cluster is consistent with the proportion of that group in the entire dataset. It is, however, computationally challenging to incorporate fairness constraints into existing graph clustering algorithms, particularly for large graphs. To address this problem, we propose FairAD, a computationally efficient fair graph clustering method. It first constructs a new affinity matrix based on the notion of algebraic distance such that fairness constraints are imposed. A graph coarsening process is then performed on this affinity matrix to find representative nodes that correspond to $k$ clusters. Finally, a constrained minimization problem is solved to obtain the solution of fair clustering. Experiment results on the modified stochastic block model and six public datasets show that FairAD can achieve fair clustering while being up to 40 times faster compared to state-of-the-art fair graph clustering algorithms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。