跳到正文
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 倍。

正文

View PDF HTML (experimental)

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