idSCD: Identifying Training Datasets through Semantic Correlation Descriptors
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
- 2016Membership Inference Attacks against Machine Learning Modelscited by 2,281
- 2020Shortcut Learning in Deep Neural Networkscited by 1,969
- 2019Invariant Risk Minimizationcited by 1,854
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,741
- 2021Membership Inference Attacks From First Principlescited by 878
- 2019Right for the Wrong Reasons: Diagnosing Syntactic Heuristics in Natural Language Inferencecited by 876
- 2018Annotation Artifacts in Natural Language Inference Datacited by 671
- 2018ML-Leaks: Model and Data Independent Membership Inference Attacks and Defenses on Machine Learning Modelscited by 498
- 2020GoEmotions: A Dataset of Fine-Grained Emotionscited by 486
- 2020Label-Only Membership Inference Attackscited by 388
- 2020Systematic Evaluation of Privacy Risks of Machine Learning Modelscited by 292
25 more
- 2019White-box vs Black-box: Bayes Optimal Strategies for Membership Inferencecited by 271
- 2021Enhanced Membership Inference Attacks against Machine Learning Modelscited by 248
- 2019Probing Neural Network Comprehension of Natural Language Argumentscited by 214
- 2018Machine Learning with Membership Privacy using Adversarial Regularizationcited by 186
- 2019Stolen Memories: Leveraging Model Memorization for Calibrated White-Box Membership Inferencecited by 175
- 2020Membership Leakage in Label-Only Exposurescited by 168
- 2020A Survey of Privacy Attacks in Machine Learningcited by 146
- 2022On Feature Learning in the Presence of Spurious Correlationscited by 145
- 2017PubMed 200k RCT: a Dataset for Sequential Sentence Classification in Medical Abstractscited by 124
- 2023Low-Cost High-Power Membership Inference Attackscited by 107
- 2022Membership Inference Attacks by Exploiting Loss Trajectorycited by 101
- 2022Shortcut Learning of Large Language Models in Natural Language Understandingcited by 85
- 2021Causal Attention for Unbiased Visual Recognitioncited by 80
- 2023Scalable Membership Inference Attacks via Quantile Regressioncited by 65
- 2024Scalable Influence and Fact Tracing for Large Language Model Pretrainingcited by 42
- 2024The Clever Hans Mirage: A Comprehensive Survey on Spurious Correlations in Machine Learningcited by 41
- 2020XED: A Multilingual Dataset for Sentiment Analysis and Emotion Detectioncited by 33
- 2022Understanding Rare Spurious Correlations in Neural Networkscited by 27
- 2022Membership Inference Attack Using Self Influence Functionscited by 20
- 2024Shortcut Learning in Medical Image Segmentationcited by 16
- 2024OSLO: One-Shot Label-Only Membership Inference Attackscited by 13
- 2018Native Language Cognate Effects on Second Language Lexical Choicecited by 11
- 2025Enhancing Training Data Attribution with Representational Optimizationcited by 8
- 2024Bridging Explainability and Embeddings: BEE Aware of Spuriousnesscited by 2
- 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
- 2026Detecting Non-Membership in LLM Training Data via Rank Correlations
- 2026LUMOS: Tracing Parametric Knowledge from Training Data to Behavioral Outputs in LLMs
- 2026WARP: Weight-Space Analysis for Recovering Training Data Portfolios
- 2026Concept Influence: Leveraging Interpretability to Improve Performance and Efficiency in Training Data Attribution
- 2026Discovering and Preserving Category Correlation Knowledge via Adaptive Reciprocal Knowledge Distillation
- 2026PLDiv: Measuring Dataset Diversity and Semantic Collapse via Persistence Landscapes
- 2024CAKD: A Correlation-Aware Knowledge Distillation Framework Based on Decoupling Kullback-Leibler Divergence
- 2026STAMP: Predicting Out-of-Distribution Generalization without Target Data