arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2009.06192cs.LGcs.CYstat.ML

A Principled Approach to Data Valuation for Federated Learning

Harvard University(哈佛大学);ETH Zurich(苏黎世联邦理工学院);Virginia Tech(弗吉尼亚理工大学);UC Berkeley(加州大学伯克利分校)

Tianhao Wang, Johannes Rausch, Ce Zhang, Ruoxi Jia, Dawn Song

英文摘要:

Federated learning (FL) is a popular technique to train machine learning (ML) models on decentralized data sources. In order to sustain long-term participation of data owners, it is important to fairly appraise each data source and compensate data owners for their contribution to the training process. The Shapley value (SV) defines a unique payoff scheme that satisfies many desiderata for a data value notion. It has been increasingly used for valuing training data in centralized learning. However, computing the SV requires exhaustively evaluating the model performance on every subset of data sources, which incurs prohibitive communication cost in the federated setting. Besides, the canonical SV ignores the order of data sources during training, which conflicts with the sequential nature of FL. This paper proposes a variant of the SV amenable to FL, which we call the federated Shapley value. The federated SV preserves the desirable properties of the canonical SV while it can be calculated without incurring extra communication cost and is also able to capture the effect of participation order on data value. We conduct a thorough empirical study of the federated SV on a range of tasks, including noisy label detection, adversarial participant detection, and data summarization on different benchmark datasets, and demonstrate that it can reflect the real utility of data sources for FL and has the potential to enhance system robustness, security, and efficiency. We also report and analyze "failure cases" and hope to stimulate future research.

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

↑