arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.09288cs.LG

EcoVal:一种面向机器学习的高效数据估值框架

EcoVal: An Efficient Data Valuation Framework for Machine Learning

RespAI Lab(RespAI实验室);KIIT Bhubaneswar(布巴内斯瓦尔KIIT学院);NUS Business School(新加坡国立大学商学院);National University of Singapore(新加坡国立大学);Adam Smith Business School(亚当·斯密商学院);University of Glasgow(格拉斯哥大学);School of Computing(计算机学院)

Ayush K Tarun, Vikram S Chundawat, Murari Mandal, Hong Ming Tan, Bowei Chen, Mohan Kankanhalli

AI总结:

针对现有基于Shapley值的机器学习数据估值框架计算成本高的问题,提出EcoVal框架,通过簇级估值+价值传播、结合生产函数估算数据内外在价值,实现高效数据估值,且在分布内和样本外数据上均有效。

摘要:

在机器学习工作流中量化数据价值,对机器学习项目制定更具战略性的决策至关重要。现有基于Shapley值(沙普利值)的机器学习数据估值框架计算成本极高,因为需要对模型进行大量重复训练才能得到Shapley值。本文提出了一种高效数据估值框架EcoVal,可快速且实用地估计机器学习模型的数据价值。该框架不直接处理单个数据样本,而是先确定一组相似数据点构成的簇的价值,再将该价值传播到簇内所有成员数据点。研究表明,可通过估计每个数据的内在价值与外在价值来确定数据的整体价值,这一过程通过将模型性能建模为“生产函数”实现——生产函数是传统自由经济市场中用于根据劳动力、资本等要素估算产出量的常用概念。本文为所提出的估值技术提供了形式化证明,阐明了其实现加速性能的原理与机制。通过验证该方法在分布内数据和样本外数据上的有效性,展示了其实际应用价值。本工作解决了机器学习模型大规模高效数据估值的核心挑战之一,代码可在https://github.com/respai-lab/ecoval 获取。

英文摘要:

Quantifying the value of data within a machine learning workflow can play a pivotal role in making more strategic decisions in machine learning initiatives. The existing Shapley value based frameworks for data valuation in machine learning are computationally expensive as they require considerable amount of repeated training of the model to obtain the Shapley value. In this paper, we introduce an efficient data valuation framework EcoVal, to estimate the value of data for machine learning models in a fast and practical manner. Instead of directly working with individual data sample, we determine the value of a cluster of similar data points. This value is further propagated amongst all the member cluster points. We show that the overall value of the data can be determined by estimating the intrinsic and extrinsic value of each data. This is enabled by formulating the performance of a model as a \textit{production function}, a concept which is popularly used to estimate the amount of output based on factors like labor and capital in a traditional free economic market. We provide a formal proof of our valuation technique and elucidate the principles and mechanisms that enable its accelerated performance. We demonstrate the real-world applicability of our method by showcasing its effectiveness for both in-distribution and out-of-sample data. This work addresses one of the core challenges of efficient data valuation at scale in machine learning models. The code is available at \underline{https://github.com/respai-lab/ecoval}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑