arXiv:cs.AI· Yunting Song, Matthew Watson, Moustafa Farid Alzantot, Peter Grabowski, Jun Qin·· 13 小时前
First-Order In-Context Shapley:面向 LLM 对齐与评估的可扩展数据审计
First-Order In-Context Shapley: Scalable Data Auditing for LLM Alignment and Evaluation
AI 导读
研究者提出 First-Order In-Context Shapley,一种仅需推理的数据估值流程,在语义 k-NN 邻域上以 O(k) 次前向计算完成,无需更新参数,与多轮 Monte-Carlo in-context Shapley 在 100% 目标上正相关,算力降低 19 倍。
正文
Abstract:Post-training alignment of Large Language Models is increasingly bottlenecked by data quality: preference and instruction-tuning corpora accumulate label noise, safety inversions, and inter-annotator contradictions that deduplication misses and full-corpus LLM judging cannot afford. While Data Shapley offers a principled valuation framework, exact computation requires intractable retraining, and embedding surrogates like KNN-Shapley fail to separate high- from low-utility demonstrations ($\rho = 0.0003$ with generative utility). We introduce an inference-only valuation pipeline that computes the first-order term of an in-context cooperative game over semantic $k$-NN neighborhoods in $O(k)$ forward passes without parameter updates, correlating positively with multi-shot Monte-Carlo in-context Shapley on 100% of targets at 19x less compute. Projecting pairwise log-likelihood shifts onto a directed influence graph characterizes each record via three signals: outgoing Exemplifying Advantage, incoming Exemplified Advantage, and a topological One-of-a-Kind Score protecting rare prompts. Pre-filtering restricts LLM twin arbitration to 1%-5% of records while achieving 4.5x enrichment in confirmed errors on HelpSteer2 and 2.2x on Anthropic HH-RLHF, catching contradictions that zero-shot likelihood and margin filters miss. During DPO training, confirmed contradictions oppose the update far more often than clean records (56.7% vs. 37.0% in LoRA gradients) and stay below chance in reward accuracy. Finally, auditing the HH-RLHF benchmark exposes label errors where aligned models disagree with human labels 73%-83% of the time, and retraining DPO after repairing or filtering flagged records improves reward accuracy by +0.5%-0.6% overall and +3.1%-3.5% on repaired inversions, even without LLM arbitration.
| Subjects: | Machine Learning (cs.LG); Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2607.22766 [cs.LG] |
| (or arXiv:2607.22766v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2607.22766 arXiv-issued DOI via DataCite |
Submission history
From: Yunting Song [view email]
[v1]
Fri, 24 Jul 2026 03:31:44 UTC (144 KB)
[v2]
Thu, 8 Oct 2026 20:38:14 UTC (256 KB)
来源:arXiv:cs.AI · arxiv.org