arXiv Machine Learning

Correcting test set contamination by spiking the training data

arXiv Computation and Language
Sep 4

Contamination Inflates Scores but Rarely Reorders Large Language Model Leaderboards

The paper investigates how benchmark contamination—leakage of test items into training data—affects large language model (LLM) leaderboards. By comparing original test items with semantically equivalent paraphrases, the authors measure contamination as a violation of anchor-item invariance and find that it inflates absolute scores but rarely changes model rankings. Across 47 public models and 74 finetuned models on four benchmarks, the rank correlation between standard and paraphrase-controlled leaderboards is 0.997, with only a handful of cases showing differential contamination that could alter rankings.

By Xingyao Xiao (Stanford University), Yihong Cheng (City University of Macau)
arXiv AI
6d ago

Counterfactual Model Testing Under Distribution Shift

The paper addresses the problem of evaluating deployed machine learning models when their input data distribution shifts, which can arise from changes in the underlying population or from data integrity issues in the feature pipeline. It proposes an online model‑testing framework that reproduces the detected shift on a controlled baseline with known labels, distinguishing between the two mechanisms of shift to estimate current performance and potential gains from repairing data integrity problems. Experiments on real‑world datasets with simulated faults demonstrate that separating the two sources of shift generally improves performance estimates, and a churn‑prediction case study shows how these tests can guide monitoring and repair prioritization.

By Joran Leest, Ilias Gerostathopoulos, Claudia Raibulet, Patricia Lago
arXiv AI
6d ago

Counterfactual Test-Time Augmentation for Shortcut Removal in Language Models

The paper introduces Counterfactual Test-Time Augmentation (CTTA) to mitigate shortcut exploitation in large language models. CTTA generates counterfactual edits of an input using an auxiliary model, then averages the LLM’s predictions across these edits, with theoretically optimal weights derived from the training distribution. The authors propose Select-CTTA, which selects among CTTA variants based on tests for assumption violations, and demonstrate that it outperforms other test-time methods and rivals models fine‑tuned on shortcut‑balanced data.

By Katie Matton, Rosalind W. Picard, John Guttag, Maggie Makar
arXiv Machine Learning
Sep 10

The BatchNorm Illusion: Diagnosing Normalization Artifacts in Machine Unlearning Evaluation

The paper uncovers a hidden flaw in evaluating machine unlearning on BatchNorm-based models: a single forward pass over retained data can alter the model’s normalization state without changing weights, misleadingly restoring performance. This effect, formalized as a weight‑preserving fixed‑point operator, shows that apparent forgetting can be entirely due to BatchNorm running statistics, not to the unlearning method itself. Experiments demonstrate that the artifact can inflate forget accuracy by up to 78 percentage points across nine methods, and that switching to GroupNorm eliminates the problem.

By Aaryaman Kalani, Murari Mandal, Dhruv Kumar, Mohan Kankanhalli, Yash Sinha