中文

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

机器学习 2026-05-21 v1 数据结构与算法

摘要

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.

关键词

引用

@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}
}

备注

To appear at VLDB 2026