mlsanityData to 14 Oct 2026

idSCD: Identifying Training Datasets through Semantic Correlation Descriptors

arXiv 2605.30462 ↗First posted 28 May 2026cs.LG cs.AIcited by 0

On the map: LLM safety and security → Machine unlearning

Abstract (arXiv)

Can a dataset be recognized from the spurious correlations it induces during training? We argue that datasets leave dataset-specific traces in a model's learned semantic correlation structure: incidental regularities that are predictive within a dataset, but not causal for the underlying task, can be internalized during training. We use this insight to study dataset-level membership inference, moving beyond existing methods that rely on behavioral or distributional evidence such as confidence scores, losses, margins, generated samples, or query responses. We introduce a white-box semantic fingerprinting approach based on semantic correlation descriptors (SCDs), which capture the semantic correlation structure learned by a model and make it comparable across dataset mixtures. In a controlled leave-one-dataset-out diagnostic, SCDs recover dataset-specific changes and perfectly separate matching from non-matching dataset pairs. We then propose a practical SCD-based membership score that tests whether a target dataset is part of a model's training mixture using only the model's SCD and the target dataset's standalone SCD, without requiring leave-one-dataset-out models. Across three diverse experimental settings, with dataset groups for natural language inference, emotion classification, and medical text classification, we test both the advantages and limitations of SCD-based membership inference with different degrees of semantic separation and keyword support between dataset splits. On average, the classifier based on this score achieves the highest performance and the lowest std, outperforming black-box baselines RMIA, Attack-P, and LiRA, as well as the white-box SIF baseline. These results show that dataset membership can be traced through internal semantic correlations, with the largest relative gain exceeding 60% in ROC-AUC when dataset groups expose distinct semantic particularities.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Machine unlearning, one of 12 directions in LLM safety and security. Builds on 37 of its 51 references that are in the map. First posted on arXiv 28 May 2026.

Builds on

37 of 51 references.

References · 37

  1. 2016Membership Inference Attacks against Machine Learning Modelscited by 2,281
  2. 2020Shortcut Learning in Deep Neural Networkscited by 1,969
  3. 2019Invariant Risk Minimizationcited by 1,854
  4. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
  5. 2017Understanding Black-box Predictions via Influence Functionscited by 1,741
  6. 2021Membership Inference Attacks From First Principlescited by 878
  7. 2019Right for the Wrong Reasons: Diagnosing Syntactic Heuristics in Natural Language Inferencecited by 876
  8. 2018Annotation Artifacts in Natural Language Inference Datacited by 671
  9. 2018ML-Leaks: Model and Data Independent Membership Inference Attacks and Defenses on Machine Learning Modelscited by 498
  10. 2020GoEmotions: A Dataset of Fine-Grained Emotionscited by 486
  11. 2020Label-Only Membership Inference Attackscited by 388
  12. 2020Systematic Evaluation of Privacy Risks of Machine Learning Modelscited by 292
25 more
  1. 2019White-box vs Black-box: Bayes Optimal Strategies for Membership Inferencecited by 271
  2. 2021Enhanced Membership Inference Attacks against Machine Learning Modelscited by 248
  3. 2019Probing Neural Network Comprehension of Natural Language Argumentscited by 214
  4. 2018Machine Learning with Membership Privacy using Adversarial Regularizationcited by 186
  5. 2019Stolen Memories: Leveraging Model Memorization for Calibrated White-Box Membership Inferencecited by 175
  6. 2020Membership Leakage in Label-Only Exposurescited by 168
  7. 2020A Survey of Privacy Attacks in Machine Learningcited by 146
  8. 2022On Feature Learning in the Presence of Spurious Correlationscited by 145
  9. 2017PubMed 200k RCT: a Dataset for Sequential Sentence Classification in Medical Abstractscited by 124
  10. 2023Low-Cost High-Power Membership Inference Attackscited by 107
  11. 2022Membership Inference Attacks by Exploiting Loss Trajectorycited by 101
  12. 2022Shortcut Learning of Large Language Models in Natural Language Understandingcited by 85
  13. 2021Causal Attention for Unbiased Visual Recognitioncited by 80
  14. 2023Scalable Membership Inference Attacks via Quantile Regressioncited by 65
  15. 2024Scalable Influence and Fact Tracing for Large Language Model Pretrainingcited by 42
  16. 2024The Clever Hans Mirage: A Comprehensive Survey on Spurious Correlations in Machine Learningcited by 41
  17. 2020XED: A Multilingual Dataset for Sentiment Analysis and Emotion Detectioncited by 33
  18. 2022Understanding Rare Spurious Correlations in Neural Networkscited by 27
  19. 2022Membership Inference Attack Using Self Influence Functionscited by 20
  20. 2024Shortcut Learning in Medical Image Segmentationcited by 16
  21. 2024OSLO: One-Shot Label-Only Membership Inference Attackscited by 13
  22. 2018Native Language Cognate Effects on Second Language Lexical Choicecited by 11
  23. 2025Enhancing Training Data Attribution with Representational Optimizationcited by 8
  24. 2024Bridging Explainability and Embeddings: BEE Aware of Spuriousnesscited by 2
  25. 2025Not All Splits Are Equal: Rethinking Attribute Generalization Across Unrelated Categoriescited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Detecting Non-Membership in LLM Training Data via Rank Correlations
  2. 2026LUMOS: Tracing Parametric Knowledge from Training Data to Behavioral Outputs in LLMs
  3. 2026WARP: Weight-Space Analysis for Recovering Training Data Portfolios
  4. 2026Concept Influence: Leveraging Interpretability to Improve Performance and Efficiency in Training Data Attribution
  5. 2026Discovering and Preserving Category Correlation Knowledge via Adaptive Reciprocal Knowledge Distillation
  6. 2026PLDiv: Measuring Dataset Diversity and Semantic Collapse via Persistence Landscapes
  7. 2024CAKD: A Correlation-Aware Knowledge Distillation Framework Based on Decoupling Kullback-Leibler Divergence
  8. 2026STAMP: Predicting Out-of-Distribution Generalization without Target Data