Researchers extend Shapley value to uncertain data with variance control
When datasets are random, Shapley value becomes a random variable—here's how to measure it.
Data valuation in machine learning often relies on the Shapley value to fairly distribute rewards based on contribution. But existing formulations assume each participant contributes a fixed, deterministic dataset. In practice, data owners typically provide samples drawn from underlying probabilistic distributions—making marginal contributions stochastic and the Shapley value itself a random variable. This paper closes that gap by defining a probabilistic Shapley value that captures both the expected contribution and the variance introduced by random sampling.
Zhuofan Jia and Jian Pei derive unbiased estimators for the expectation and variance of this probabilistic Shapley value, analyzing their statistical properties. They also introduce three Monte Carlo estimation algorithms: a baseline estimator using independent samples, a pooled estimator that reuses samples for greater efficiency, and a stratified pooled estimator that adaptively allocates sampling budget based on each player's variability. Experiments on synthetic and real datasets show the stratified pooled approach achieves strong accuracy-efficiency trade-offs, delivering substantial variance reduction at minimal added cost. This work provides both theoretical rigor and practical tools for reliable, fair data valuation in modern stochastic data-sharing environments.
- Introduces unbiased estimators for the expectation and variance of probabilistic Shapley values
- Three Monte Carlo algorithms: baseline, pooled (sample reuse), and stratified pooled (adaptive budget allocation)
- Stratified pooled estimator reduces variance substantially with minimal extra computational cost
Why It Matters
Enables fair compensation in data marketplaces and federated learning when data is inherently stochastic, not fixed.