mlsanityData to 7 Oct 2026

Optimizing Data Shapley Interaction Calculation from O(2^n) to O(t n^2) for KNN models

arXiv 2304.01224 ↗First posted 2 Apr 2023cs.LG cs.AI cs.GTcited by 4

Not on the map. Why?

Abstract (arXiv)

With the rapid growth of data availability and usage, quantifying the added value of each training data point has become a crucial process in the field of artificial intelligence. The Shapley values have been recognized as an effective method for data valuation, enabling efficient training set summarization, acquisition, and outlier removal. In this paper, we introduce "STI-KNN", an innovative algorithm that calculates the exact pair-interaction Shapley values for KNN models in O(t n2) time, which is a significant improvement over the O(2n) time complexity of baseline methods. By using STI-KNN, we can efficiently and accurately evaluate the value of individual data points, leading to improved training outcomes and ultimately enhancing the effectiveness of artificial intelligence applications.

Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 6 of its 12 references that are in the map. First posted on arXiv 2 Apr 2023.

Builds on

6 of 12 references.

References · 6

  1. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
  2. 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 617
  3. 2017Interpretation of Neural Networks is Fragilecited by 443
  4. 2021Beta Shapley: a Unified and Noise-reduced Data Valuation Framework for Machine Learningcited by 162
  5. 2019The Shapley Taylor Interaction Indexcited by 143
  6. 2019Data Valuation using Reinforcement Learningcited by 115

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 3; 2025: 0; 2026: 1.

4 papers.