English

Efficient Banzhaf-Based Data Valuation for $k$-Nearest Neighbors Classification

Machine Learning 2026-05-21 v1 Data Structures and Algorithms

Abstract

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 kk-nearest neighbor (kkNN) 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 kkNN 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)O(Wkn^2) time complexity for weighted kkNN classifiers, where WW is the maximum sum of top-kk weights, and a specialized algorithm for unweighted kkNN that achieves O(nk2)O(nk^2) 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.

Keywords

Cite

@article{arxiv.2605.21033,
  title  = {Efficient Banzhaf-Based Data Valuation for $k$-Nearest Neighbors Classification},
  author = {Guangyi Zhang and Lutz Oettershagen and Lixu Wang and Aristides Gionis},
  journal= {arXiv preprint arXiv:2605.21033},
  year   = {2026}
}

Comments

To appear at VLDB 2026