KAIROS: Scalable Model-Agnostic Data Valuation

Jiongli Zhu, Parjanya Prajakta Prashant, Alex Cloninger, Babak Salimi
6/30/2025
cs.LG

No official code found. Best implementation: lodino/kairos (2 stars, MIT, last commit 2025-10-23).

Abstract

Training data increasingly shapes not only model accuracy but also regulatory compliance and market valuation of AI assets. Yet existing valuation methods remain inadequate: model-based techniques depend on a single fitted model and inherit its biases, while algorithm-based approaches such as Data Shapley require costly retrainings at web scale. Recent Wasserstein-based model-agnostic methods rely on approximations that misrank examples relative to their true leave-one-out (LOO) utility. We introduce KAIROS, a scalable, model-agnostic valuation framework that assigns each example a distributional influence score: its contribution to the Maximum Mean Discrepancy (MMD) between the empirical training distribution and a clean reference set. Unlike Wasserstein surrogates, our MMD-based influence admits a closed-form solution that faithfully approximates the exact LOO ranking within $O(1/N^2)$ error, requires no retraining, and naturally extends to conditional kernels for unified label- and feature-error detection. Moreover, KAIROS supports efficient online updates: when a new batch of size m arrives, all scores can be updated in $O(mN)$ time, delivering up to 50x speedup without compromising ranking quality. Empirical evaluations on noise, mislabeling, and poisoning benchmarks show that KAIROS consistently outperforms state-of-the-art model-, Shapley-, and Wasserstein-based baselines in both accuracy and runtime. We provide rigorous theoretical guarantees, including symmetry for reproducible rankings and density-separation for interpretable thresholds.

View on arXivView PDF

Code Implementations(1)

lodino/kairosCommunity67%
MIT

Implementation for paper: KAIROS: Scalable Model-Agnostic Data Valuation

21PythonMay 14, 202512 months agoMIT

Similar Papers

Cite this paper

@article{zhu2025kairos,
  title  = {KAIROS: Scalable Model-Agnostic Data Valuation},
  author = {Jiongli Zhu and Parjanya Prajakta Prashant and Alex Cloninger and Babak Salimi},
  year   = {2025},
  eprint = {2506.23799},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2506.23799v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2506.23799.svg)](https://codeofpaper.com/paper/2506.23799)

Discussion