Beyond Shapley: An Influence-Based Data Auditing Pipeline for LLM Alignment and Evaluation
The paper introduces a scalable, inference-only data valuation pipeline for LLM alignment that approximates Shapley values without iterative model retraining. It maps semantic k-NN neighborhoods into a directed graph to evaluate data utility using zero-shot and one-shot conditional log-likelihood shifts from a reference LLM. The pipeline isolates gradient-conflicting records by translating predictive influence scores into localized advantage metrics, enabling efficient auditing of large-scale al
Analysis
TL;DR
- The paper introduces a scalable, inference-only data valuation pipeline for LLM alignment that approximates Shapley values without iterative model retraining.
- It maps semantic k-NN neighborhoods into a directed graph to evaluate data utility using zero-shot and one-shot conditional log-likelihood shifts from a reference LLM.
- The pipeline isolates gradient-conflicting records by translating predictive influence scores into localized advantage metrics, enabling efficient auditing of large-scale alignment datasets.
- Applied to HelpSteer2, it reduced manual audit search space by 99.1% and uncovered falsely-labeled records across diverse failure modes.
- When applied to Anthropic’s HH-RLHF training and evaluation splits, it revealed thousands of hidden safety and factual preference inversions, exposing vulnerabilities in benchmark integrity due to flawed human ground-truth labels.
Why It Matters
This work addresses a critical bottleneck in LLM alignment: the degradation of dataset quality at scale, which undermines model performance and safety guarantees. By providing an efficient, mathematically grounded method to detect subtle but harmful data inconsistencies—especially those invisible to standard deduplication or judge-based audits—it offers practitioners a powerful tool to sanitize both training and evaluation datasets, ensuring more reliable and trustworthy alignment processes.
Technical Details
- The framework uses a reference LLM to compute conditional log-likelihood shifts under zero-shot and one-shot settings, measuring how each data point influences the model’s output distribution.
- Semantic k-NN neighborhoods are constructed based on embedding similarities and mapped into a directed graph where edges represent directional influence between samples.
- Shapley value approximation is achieved via Monte Carlo sampling over subsets of the neighborhood graph, avoiding costly retraining while preserving theoretical fairness properties.
- Localized advantage metrics are derived from these influence scores to identify records that contradict dominant patterns or introduce conflicting gradients during fine-tuning.
- The method was validated on two major alignment benchmarks: HelpSteer2 (instruction tuning) and HH-RLHF (preference learning), demonstrating high precision in detecting label errors and preference inversions.
Industry Insight
AI teams should integrate automated influence-based auditing pipelines like this one into their standard data curation workflows, especially when scaling up instruction-tuning or RLHF datasets. Relying solely on human annotation or surface-level deduplication risks propagating latent contradictions that can degrade model behavior post-deployment. Furthermore, evaluation benchmarks must be regularly audited using such methods to ensure they reflect true capability rather than artifacts of biased or erroneous labeling—otherwise, reported improvements may be illusory.
Disclaimer: The above content is generated by AI and is for reference only.