arXiv:2602.21509stat.MLcs.LG2026-02AAAI

提出可高效扩展的公平聚类方法,解决传统算法参数随数据量增长的问题。

Fair Model-based Clustering

  • 基于有限混合模型构建聚类框架,参数量与样本规模无关
  • 支持小批量学习,实现大规模数据上的近似公平聚类
  • 适用于非度量数据,只要似然函数定义良好即可

公平聚类的目标是使每个簇中敏感属性(如性别、种族等)的比例与整个数据集相似。现有公平聚类算法通常在标准K均值基础上修改以满足公平约束,但存在一个关键局限:每个数据点的簇分配需与簇中心同时优化,导致可学习参数数量与样本量成正比,难以扩展。本文提出一种基于有限混合模型的新公平聚类算法——公平模型基聚类(Fair Model-based Clustering, FMC)。FMC的核心优势在于其可学习参数数量独立于样本规模,因而易于扩展。特别地,支持小批量学习,可在大规模数据上获得近似公平的聚类结果。此外,FMC可应用于非度量数据(如分类数据),只要对应的似然函数有明确定义。论文提供了理论和实证证据,证明该算法的优越性。

原文摘要 · Abstract (English)

The goal of fair clustering is to find clusters such that the proportion of sensitive attributes (e.g., gender, race, etc.) in each cluster is similar to that of the entire dataset. Various fair clustering algorithms have been proposed that modify standard K-means clustering to satisfy a given fairness constraint. A critical limitation of several existing fair clustering algorithms is that the number of parameters to be learned is proportional to the sample size because the cluster assignment of each datum should be optimized simultaneously with the cluster center, and thus scaling up the algorithms is difficult. In this paper, we propose a new fair clustering algorithm based on a finite mixture model, called Fair Model-based Clustering (FMC). A main advantage of FMC is that the number of learnable parameters is independent of the sample size and thus can be scaled up easily. In particular, mini-batch learning is possible to obtain clusters that are approximately fair. Moreover, FMC can be applied to non-metric data (e.g., categorical data) as long as the likelihood is well-defined. Theoretical and empirical justifications for the superiority of the proposed algorithm are provided.

聚类公平性模型基可扩展

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