想象一下你是一名法官,在法庭上,你的职责不是审判个人,而是根据一份庞大的过往案例账本来评判群体。你的目标是做出预测:“这个人会成功吗?”或者“这个人会再次犯罪吗?”
你提供的这份论文——《迈向可证明公平的机器学习》(Towards Provably Fair Machine Learning)——认为现代大多数计算机程序(机器学习模型)在面对特定的小众群体时,并不是优秀的法官。它们往往会做出一些与眼前确凿证据相矛盾的自信猜测。
以下是该论文论点的拆解,使用了简单的类比。
1. 问题所在:“自信的蠢货”
标准的机器学习模型就像是一个背下了考试答案但并不理解逻辑的学生。
- 问题: 当数据量巨大时(比如一个城市的人口),这些模型表现良好。但当它们观察一个极其微小的特定群体时(比如“左撇子、红发、上夜班的女性”),整个数据库中可能只有 5 个人属于这个群体。
- 错误: 标准模型仍会尝试进行猜测。它们通常会为了简化数学计算而“抹平”细节。这就像一位老师为了让全班平均分好看,而忽略了少数几个学生面临的具体困难。
- 结果: 模型做出的预测在统计学上是不可能的。例如,如果 100 个完全相同的人中恰好有 50% 的成功率,模型可能会自信地断言“100% 会成功”或“0% 会成功”。这两者都是错的,但模型依然会这么说,因为它试图表现得果断。
2. 解决方案:“诚实的侦探”
作者提出了一种名为公平贝叶斯(Fair Bayesian, FB)分类器的新方法。请不要把它看作是一个试图拿“A”的学生,而要把它看作是一个诚实的侦探——它拒绝在证据不确凿时进行猜测。
这位侦探遵循两条严格的规则:
- 双胞胎规则(确定性): 如果两个人的细节完全相同(同样的职业、年龄、经历),他们必须得到完全相同的预测。你不能区别对待这对一模一样的双胞胎。
- 证据规则(统计一致性): 侦探只有在数据能够“证明”某种可能性时才会做出预测。如果证据太弱,或者证据证明“是”与“否”都是错的,侦探将拒绝猜测。
3. 魔法技巧:“弃权”
这是论文中最独特的部分。在现实世界中,我们通常认为计算机应该始终给出一个答案。但本文认为,有时唯一的公平答案是**“我不知道”**。
- 类比: 想象一次硬币投掷。如果你投了 3 次,其中 2 次是正面,你可能会猜下一次也是正面。但如果你投了 1,000 次,且正好是 500 次正面和 500 次反面,你就知道这枚硬币是公平的。如果你被迫对下一次投掷做出预测,你就是在瞎猜。但如果你被迫对这 1,000 次投掷中的每一次都做出“正面”或“反面”的预测,你就是在撒谎。
- 论文的方法: 公平贝叶斯分类器会观察一个群体。如果数据显示预测“是”是错误的,且预测“否”也是错误的(因为该群体比例为 50/50,且样本量足够大以至于可以确定这一点),模型就会弃权。它会说:“对于这个特定的群体,我无法做出公平且一致的预测。”
4. 为什么这关乎公平性
论文指出了当前 AI 中一个残酷的讽刺现象:
- 少数群体往往处于小众群体中。 因为他们很稀少,所以他们往往会陷入那些标准模型最容易出错的“微小子集”中。
- 标准模型伤害少数群体。 为了优化数学计算,标准模型经常将这些小群体并入更大的、通用的群体中。这抹杀了他们的独特性,并将他们强行纳入一个并不符合他们的预测之中。
- 公平贝叶斯的修复方案: 通过逐一观察每一个微小的群体,并在证据不稳固时拒绝猜测,这种新方法保护了这些处于弱势地位的小众群体。它承认自己信息不足,而不是编造一个会对他们造成伤害的故事。
5. 结果:“零错误”规则
作者将他们的“诚实侦探”与标准模型(如决策树和神经网络)在三个著名的数据库(关于收入、刑事司法和银行营销)上进行了对比测试。
- 标准模型: 它们对大量的微小群体做出了与实际数据相矛盾的预测。它们是“自信地犯错”。
- 公平贝叶斯模型:
- 它在所有预测中,零次出现了与数据相矛盾的情况。
- 在它做出预测的群体中,它的准确率实际上比其他模型更高。
- 它标记出了那些它无法决定的群体(即“我不知道”的群体),这是一种功能,而非缺陷。
总结
该论文声称,真正的公平不仅在于做出正确的猜测,更在于做出一个确实有证据支撑的猜测。
如果证据过于单薄,或者证据证明一个简单的“是/否”答案是不可能的,一个公平的系统应当停下来并说:“我需要更多信息。”作者构建了一个能做到这一点的系统,确保没有人会被一个连数据本身都证明是不可能的预测所审判。
技术摘要:迈向可证明公平的学习:用于一致且透明预测的贝叶斯方法
1. 问题陈述
部署在高度敏感领域(金融、刑事司法、医疗保健)中的机器学习模型经常产生与观测数据系统性不一致的预测,特别是在由多个特征交织定义的细粒度子群中。这种不一致性源于标准的频率派方法将观测到的样本比例视为可靠的真实概率估计,而忽略了样本量的大小。在大型数据集中,虽然总数据量很高,但在最细分辨率下(即共享所有特征相同值的个体)的数据往往由规模较小的子群组成。
两个主要问题加剧了这一现象:
- 小样本推断失效: 标准模型未能考虑到小规模子群带来的高不确定性,导致其产生过度自信但缺乏统计支持的预测。
- 正则化偏差: 为了防止过拟合,标准机器学习技术(例如剪枝决策树、神经网络中的正则化)会将小规模子群合并为较大的聚合体。这不成比例地影响了少数群体,因为他们往往集中在这些细小的、具有交织特征的子群中,从而有效地抹除了他们独特的行为模式。
- 缺乏原则性的弃权机制: 现有的模型(包括旨在实现公平性的模型)被迫为每个输入都给出预测。它们缺乏一种机制,在证据过于稀疏不足以支持置信预测,或者相反,证据足以排除所有确定性预测(例如,一个目标率恰好为 50% 且样本量很大的群体)时进行弃权。
现有的公平性方法(群体公平性、多校准性)通常依赖于预设的受保护群体,无法扩展到所有可能的交集,并且没有解决预测与观测证据之间的根本统计不一致性问题。
2. 方法论:公平贝叶斯 (FB) 分类器
作者提出了 公平贝叶斯 (Fair Bayesian, FB) 分类器,该框架将分类视为一个关于统计合理性的问题,而非针对总体准确率的优化问题。该方法建立在应用于人口概率而非模型参数的经典贝叶斯推断之上。
核心要求
该框架对任何预测强制执行两个严格要求:
- 确定性: 相同的个体(共享所有特征值)必须获得相同的预测。
- 统计一致性: 子群的预测必须与通过显著性水平 α 评估的从观测样本推断出的目标分布在统计上保持一致。
技术框架
- 子群定义: 数据被划分为 d-节点(具有所有 m 个属性相同值的群体)和 v-节点(由属性子集定义的群体,留有其他属性未指定)。
- 贝叶斯推断: 对于每个 d-节点,未知的总体概率 p 使用 Beta 后验进行建模(使用均匀的 Beta(1,1) 先验)。未来观测值的预测分布遵循 Beta-Binomial 分布。
- 一致性测试: 对于给定的具有 Nd 个观测值和 Td 个正向目标的 d-节点,会对确定性预测(全为正或全为负)进行测试。
- 如果观测数据在水平 α 下排除了“全为正”的预测,则该预测被拒绝。
- 如果数据排除了“全为负”的预测,则该预测被拒绝。
- 节点分类:
- d0:仅“全为负”是一致的。
- d1:仅“全为正”是一致的。
- damb:两者皆一致(模糊);其解析依赖于父级 v-节点的约束。
- dnf (无公平性):两者确定性预测均不一致(例如,一个目标率为 50% 且样本量很大的群体)。
- 弃权机制: 与基于置信度的弃权不同,FB 分类器在 dnf 节点上选择弃权,因为证据积极地排除了两种确定性选项。这是防止发布明显不一致预测的一种保障措施。
- 全局一致性 (v-节点): 预测还必须满足所有 v-节点(d-节点的聚合)的一致性约束。由于 v-节点的分布由于异质性无法通过闭式解计算,作者通过从其子级 d-节点的分布中采样来近似它们。
- 约束满足: 该问题被表述为一个约束满足问题,其中要求子级 d-节点的预测之和必须落在父级 v-节点的 [Vmin,Vmax] 边界内。作者使用 Gurobi 优化器寻找一个可行赋值,以最大化基于后验对数几率的线性目标,随后使用 v-节点对数似然进行二次评分步骤,从可行解池中选择最佳解。
- 异质性处理: 为了防止在可能包含未观测到的时间或来源变化的动态数据集中出现过度自信,对 Beta 后验应用了方差底限(τ=10−5),从而限制了有效样本量。
3. 主要贡献
- 预测一致性框架: 提供了一个基于贝叶斯推断的预测一致性的正式定义,并将其详尽地应用于每一个可能的子群(任何特征组合的交集),而非仅仅是预设的受保护群体。
- 公平贝叶斯分类器: 该分类器保证了与观测证据的一致性,并实现了原则性的弃权。它仅在数据排除了所有确定性预测时才进行弃权,而不是在模型置信度较低时弃权。
- 不一致性的实证证据: 证明了标准模型(决策树、神经网络)以及公平感知后处理器(比例多校准)对于相当大比例的子群而言,其预测在统计上是不一致的,即使是在它们所训练的数据集上也是如此。
- 性能结果: FB 分类器通过构建实现了零一致性误差,并且在它做出预测的子群上,其准确率超过了所有基准模型。它还通过强制执行一致性,获得了具有竞争力的多校准性。
4. 结果
作者在三个基准数据集(Adult, COMPAS, Bank Marketing)上将 FB 分类器与决策树 (DT)、神经网络 (NN) 和比例多校准 (PMC) 进行了评估:
- 一致性误差: 标准模型在大量子群中产生了统计不一致的预测。例如,在 Bank Marketing 数据集上,PMC 显示出 16.97% 的 d0/d1 一致性误差和 43.46% 的 v-节点一致性误差。FB 分类器通过设计实现了所有指标 0.00% 的误差。
- 弃权: FB 分类器在 dnf 节点上进行弃权。在 Adult 数据集中,50.4% 的实例属于 dnf 节点,这凸显了对于一半的人口而言,鉴于现有特征,不存在一致的确定性预测是可能的。
- 准确率: 在 FB 分类器做出预测的数据子集(排除 dnf 节点)上,它优于所有基准模型。值得注意的是,在 COMPAS 上,FB 实现了 77.6% 的准确率,而基准模型约为 68%。
- 多校准性: FB 分类器在没有进行显式优化的情况下实现了具有竞争力的多校准性得分,这表明统计一致性是跨细粒度子群进行校准的一个强有力的代理指标。
5. 重要性与主张
论文认为,统计一致性为预测质量提供了一个原则性的基础,并对算法公平性具有直接影响。作者主张:
- 少数群体不成比例地集中在数据推断最不可靠的小规模子群中。解决这一推断问题是实现公平机器学习的必要步骤。
- 通过在数据支持的最细分辨率级别强制执行贝叶斯一致性,可以实现实践中的详尽子群公平性和原则性的弃权。
- 能够识别出在何种情况下不存在公平的确定性预测(通过 dnf 节点)是一个关键的保障措施,防止系统发布与证据相矛盾的预测。
- 该方法将范式从“学习关联并后期调整公平性”转变为“将分类视为统计合理性”,确保每项预测都能透明地源自该特定子群所拥有的证据。
作者指出,虽然由于子群数量呈指数级增长,该框架在计算上非常密集,但对于目前的基准数据集而言是可行的,并且为启发式的公平性调整提供了一个严谨的替代方案。他们承认,处理未见数据(新 d-节点中的测试实例)需要进一步开发,尽管已提出了一个初步机制。
每周获取最佳 machine learning 论文。
受到斯坦福、剑桥和法国科学院研究人员的信赖。
请查收邮箱确认订阅。
出了点问题,再试一次?
无垃圾邮件,随时退订。