KAIROS: Scalable Model-Agnostic Data Valuation
Not on the map. Why?
Abstract (arXiv)
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/N2) 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.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 48 of its 83 references that are in the map. First posted on arXiv 30 Jun 2025.
Builds on
48 of 83 references.
References · 48
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,316
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,741
- 2015Learning Transferable Features with Deep Adaptation Networkscited by 1,329
- 2017BadNets: Identifying Vulnerabilities in the Machine Learning Model Supply Chaincited by 1,106
- 2021Deduplicating Training Data Makes Language Models Bettercited by 714
- 2019Data Shapley: Equitable Valuation of Data for Machine Learningcited by 629
- 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 580
- 2018Poison Frogs! Targeted Clean-Label Poisoning Attacks on Neural Networkscited by 519
- 2023Studying Large Language Model Generalization with Influence Functionscited by 326
- 2023TRAK: Attributing Model Behavior at Scalecited by 283
- 2017Joint Distribution Optimal Transportation for Domain Adaptationcited by 282
36 more
- 2019Towards Efficient Data Valuation Based on the Shapley Valuecited by 281
- 2021Beta Shapley: a Unified and Noise-reduced Data Valuation Framework for Machine Learningcited by 165
- 2021Mind the Style of Text! Adversarial and Backdoor Attacks Based on Text Style Transfercited by 158
- 2023D4: Improving LLM Pretraining via Document De-Duplication and Diversificationcited by 150
- 2019Efficient Task-Specific Data Valuation for Nearest Neighbor Algorithmscited by 148
- 2022Data Banzhaf: A Robust Data Valuation Framework for Machine Learningcited by 132
- 2019Data Valuation using Reinforcement Learningcited by 116
- 2016Theoretical Analysis of Domain Adaptation with Optimal Transportcited by 105
- 2024The rising costs of training frontier AI modelscited by 95
- 2024What is Your Data Worth to GPT? LLM-Scale Data Valuation with Influence Functionscited by 94
- 2024Lisa: Lazy Safety Alignment for Large Language Models against Harmful Fine-tuning Attackcited by 92
- 2020A Measure-Theoretic Approach to Kernel Conditional Mean Embeddingscited by 86
- 2024Data Shapley in One Training Runcited by 81
- 2020A Distributional Framework for Data Valuationcited by 80
- 2017Data Noising as Smoothing in Neural Network Language Modelscited by 78
- 2023LAVA: Data Valuation without Pre-Specified Learning Algorithmscited by 69
- 2023OpenDataVal: a Unified Benchmark for Data Valuationcited by 55
- 2023Model Agnostic Sample Reweighting for Out-of-Distribution Learningcited by 44
- 2024Robustifying Safety-Aligned Large Language Models through Clean Data Curationcited by 44
- 2021Revisiting Methods for Finding Influential Examplescited by 39
- 2023Data-OOB: Out-of-bag Estimate as a Simple and Efficient Data Valuecited by 38
- 2023MMD-FUSE: Learning and Combining Kernels for Two-Sample Testing Without Data Splittingcited by 33
- 2022CS-Shapley: Class-wise Shapley Values for Data Valuation in Classificationcited by 32
- 2024Rethinking Data Shapley for Data Selection Tasks: Misleads and Meritscited by 29
- 2016Conditional Generative Moment-Matching Networkscited by 24
- 2024Stochastic Amortization: A Unified Approach to Accelerate Feature and Data Attributioncited by 21
- 2021Incentivizing Collaboration in Machine Learning via Synthetic Data Rewardscited by 20
- 2022Fundamentals of Task-Agnostic Data Valuationcited by 19
- 2015Adaptivity and Computation-Statistics Tradeoffs for Kernel and Distance based High Dimensional Two Sample Testingcited by 14
- 2020Truthful Data Acquisition via Peer Predictioncited by 14
- 2022On the Uniqueness of Kantorovich Potentialscited by 7
- 2025IF-GUIDE: Influence Function-Guided Detoxification of LLMscited by 7
- 2024Data Distribution Valuationcited by 5
- 20242D-OOB: Attributing Data Contribution Through Joint Valuation Frameworkcited by 4
- 2024SAVA: Scalable Learning-Agnostic Data Valuationcited by 1
- 2024LossVal: Efficient Data Valuation for Neural Networkscited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.