arXiv:2502.02221cs.LGcs.AI2025-02KDD被引 4

提出新偏差检测方法MSD,高效识别数据分布中的子群体差异。

Bias Detection via Maximum Subgroup Discrepancy

  • 基于特征子群的最大差异度量,捕捉分布间细微偏差
  • 样本复杂度仅线性依赖特征数,实际应用可行
  • 可解释性强,适合调试和修复模型偏差

偏差评估是可信AI的基础,既可用于检验数据质量,也可用于评估AI系统输出。传统度量如总变差和Wasserstein距离样本复杂度高,在实际场景中常无法有效区分。本文提出最大子群差异(MSD)这一新度量:若所有特征子群的差异均低,则两分布接近。尽管子群数量可能呈指数级增长,但本文证明其样本复杂度仅线性依赖于特征数,具备实际可行性。同时,提出基于混合整数优化(MIO)的实用算法。该度量具有强可解释性,有助于定位并修正偏差。进一步构建了名为MSDD的自然偏差检测框架,验证了MSD与其高度契合。在真实数据集上,实验表明MSD优于其他度量,并展现出良好性质。

原文摘要 · Abstract (English)

Bias evaluation is fundamental to trustworthy AI, both in terms of checking data quality and in terms of checking the outputs of AI systems. In testing data quality, for example, one may study the distance of a given dataset, viewed as a distribution, to a given ground-truth reference dataset. However, classical metrics, such as the Total Variation and the Wasserstein distances, are known to have high sample complexities and, therefore, may fail to provide a meaningful distinction in many practical scenarios. In this paper, we propose a new notion of distance, the Maximum Subgroup Discrepancy (MSD). In this metric, two distributions are close if, roughly, discrepancies are low for all feature subgroups. While the number of subgroups may be exponential, we show that the sample complexity is linear in the number of features, thus making it feasible for practical applications. Moreover, we provide a practical algorithm for evaluating the distance based on Mixed-integer optimization (MIO). We also note that the proposed distance is easily interpretable, thus providing clearer paths to fixing the biases once they have been identified. Finally, we describe a natural general bias detection framework, termed MSDD distances, and show that MSD aligns well with this framework. We empirically evaluate MSD by comparing it with other metrics and by demonstrating the above properties of MSD on real-world datasets.

偏差检测分布比较可解释性子群分析

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