Efficient Banzhaf-Based Data Valuation for k-Nearest Neighbors Classification
Not on the map. Why?
Abstract (arXiv)
Data valuation, the task of quantifying the contribution of individual data points to model performance, has emerged as a fundamental challenge in machine learning. Game-theoretic approaches, such as the Banzhaf value, offer principled frameworks for fair data valuation; however, they suffer from exponential computational complexity. We address this challenge by developing efficient algorithms specifically tailored for computing Banzhaf values in k-nearest neighbor (kNN) classifiers. We first establish the theoretical hardness of the problem by proving that it is #P-hard. Despite this intractability, we exploit the locality properties of kNN classifiers to develop practical exact algorithms. Our main contribution is a dynamic programming framework that achieves significant computational improvements: we present a pseudo-polynomial algorithm with O(Wkn2) time complexity for weighted kNN classifiers, where W is the maximum sum of top-k weights, and a specialized algorithm for unweighted kNN that achieves O(nk2) time complexity, that is, linear in the number of data points. We also offer efficient Monte Carlo estimation methods. Extensive experiments on real-world datasets demonstrate the practical efficiency of our approach and its effectiveness in data valuation applications.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 18 of its 41 references that are in the map. First posted on arXiv 20 May 2026.
Builds on
18 of 41 references.
References · 18
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,741
- 2021Deduplicating Training Data Makes Language Models Bettercited by 714
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 629
- 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 580
- 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 552
- 2014OpenML: networked science in machine learningcited by 526
- 2020What Neural Networks Memorize and Why: Discovering the Long Tail via Influence Estimationcited by 497
- 2021Interpretable Machine Learning: Fundamental Principles and 10 Grand Challengescited by 367
- 2019Efficient Task-Specific Data Valuation for Nearest Neighbor Algorithmscited by 148
- 2022Training Data Influence Analysis and Estimation: A Surveycited by 135
- 2022Data Banzhaf: A Robust Data Valuation Framework for Machine Learningcited by 132
- 2021Sampling Permutations for Shapley Value Estimationcited by 102
6 more
- 2024Data Shapley in One Training Runcited by 81
- 2023LAVA: Data Valuation without Pre-Specified Learning Algorithmscited by 69
- 2013Bounding the Estimation Error of Sampling-based Shapley Value Approximationcited by 58
- 2023OpenDataVal: a Unified Benchmark for Data Valuationcited by 55
- 2024Efficient Data Shapley for Weighted Nearest Neighbor Algorithmscited by 18
- 2023A Note on "Efficient Task-Specific Data Valuation for Nearest Neighbor Algorithms"cited by 11
Built on by
Not cited yet.