Shapley-Based Data Valuation for Weighted -Nearest Neighbors
Guangyi Zhang, Qiyu Liu, Aristides Gionis
Abstract
Data valuation quantifies the impact of individual data points on model performance, and Shapley values provide a principled approach to this important task due to their desirable axiomatic properties, albeit with high computational complexity. Recent breakthroughs have enabled fast computation of exact Shapley values for unweighted k -nearest neighbor ( k NN) classifiers. However, extending this to weighted k NN models has remained a significant open challenge. The state-of-the-art methods either require quadratic time complexity or resort to approximation via sampling. In this paper, we show that a conceptually simple but overlooked approach — data duplication — can be applied to this problem, yielding a natural variant of weighted k NN-Shapley. However, a straightforward application of the data-duplication idea leads to increased data size and prohibitive computational and memory costs. We develop an efficient algorithm that avoids materializing the duplicated dataset by exploiting the structural properties of weighted k NN models, reducing the complexity to near-linear time in the original data size. Besides, we establish theoretical foundations for this approach through axiomatic characterization of the resulting values, and empirically validate the effectiveness and efficiency of our method.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 8074dbce-8721-4b51-b77b-ed880968434dBuilds on7
- Estimating Training Data Influence by Tracing Gradient DescentGarima Pruthi, Frederick Liu, Satyen Kale, Mukund SundararajanNeurIPS 2020 · 784 citations
- What Neural Networks Memorize and Why: Discovering the Long Tail via Influence EstimationVitaly Feldman, Chiyuan ZhangNeurIPS 2020 · 674 citations
- Scaling Up Influence FunctionsAndrea Schioppa, Polina Zablotskaia, David Vilar, Artem SokolovAAAI 2022 · 149 citations
- If You Like Shapley Then You'll Love the CoreTom Yan, Ariel D. ProcacciaAAAI 2021 · 85 citations
- Datamodels: Understanding Predictions with Data and Data with PredictionsAndrew Ilyas, Sung Min Park, Logan Engstrom, Guillaume Leclerc et al.ICML 2022 · 66 citations
Related papers
- Localized Data Shapley: Accelerating Valuation for Nearest Neighbor AlgorithmsGuangyi Zhang, Yanhao Wang, Chengliang Chai, Qiyu Liu et al.NeurIPS 2025 · 1 citation
- Efficient Banzhaf-Based Data Valuation for k-Nearest Neighbors ClassificationGuangyi Zhang, Lutz Oettershagen, Lixu Wang, Aristides GionisVLDB 2026 · 1 citation
- A Privacy-Friendly Approach to Data ValuationJiachen T. Wang, Yuqing Zhu, Yu-Xiang Wang, Ruoxi Jia et al.NeurIPS 2023 · 12 citations
- Shapley Value Approximation Based on k-Additive GamesGuilherme Dean Pelegrina, Patrick Kolpaczki, Eyke HüllermeierAAAI 2026
- Data-OOB: Out-of-bag Estimate as a Simple and Efficient Data ValueYongchan Kwon, James ZouICML 2023 · 54 citations
