SureMap通过融合外部数据,精准估算模型在多个群体中的表现。
SureMap: Simultaneous Mean Estimation for Single-Task and Multi-Task Disaggregated Evaluation
- 将多任务群体评估转为结构化高斯均值联合估计
- 在真实数据上相比基线方法显著提升评估精度
- 适合需要公平性验证的AI系统部署者使用
disaggregated evaluation(细分群体评估)是评估机器学习模型性能与群体公平性的核心任务。主要挑战在于数据稀缺,尤其是由种族、性别、年龄等属性交叉形成的子群体规模极小。当前多个客户从同一开发者处采购同一模型,各自需在自身数据中进行细分评估,形成多任务细分评估问题。本文提出SureMap方法,可同时高效准确地完成单任务与多任务黑箱模型的细分评估。其核心思想是将问题转化为结构化的联合高斯均值估计,并引入外部数据(如模型开发者或其它客户的数据)。方法结合最大后验(MAP)估计与无需交叉验证的Stein无偏风险估计(SURE)实现参数优化。在多个领域任务上的实验表明,SureMap相比多种强基线方法均有显著精度提升。
原文摘要 · Abstract (English)
Disaggregated evaluation -- estimation of performance of a machine learning model on different subpopulations -- is a core task when assessing performance and group-fairness of AI systems. A key challenge is that evaluation data is scarce, and subpopulations arising from intersections of attributes (e.g., race, sex, age) are often tiny. Today, it is common for multiple clients to procure the same AI model from a model developer, and the task of disaggregated evaluation is faced by each customer individually. This gives rise to what we call the multi-task disaggregated evaluation problem, wherein multiple clients seek to conduct a disaggregated evaluation of a given model in their own data setting (task). In this work we develop a disaggregated evaluation method called SureMap that has high estimation accuracy for both multi-task and single-task disaggregated evaluations of blackbox models. SureMap's efficiency gains come from (1) transforming the problem into structured simultaneous Gaussian mean estimation and (2) incorporating external data, e.g., from the AI system creator or from their other clients. Our method combines maximum a posteriori (MAP) estimation using a well-chosen prior together with cross-validation-free tuning via Stein's unbiased risk estimate (SURE). We evaluate SureMap on disaggregated evaluation tasks in multiple domains, observing significant accuracy improvements over several strong competitors.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。