atlas

Research map / Large language models / Language, safety and interpretability

Unlocking Post-hoc Dataset Inference with Synthetic Data

ICML 2025 · Large language models › Language, safety and interpretability · approach: LLM unlearning and benchmarks · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes generating synthetic held-out data via a LoRA-finetuned Llama 3 8B suffix-completion generator plus a dual-classifier post-hoc calibration with a t-test, so dataset inference works without in-distribution held-out data.

Problem

Dataset inference for LLMs requires an in-distribution held-out (validation) set known to be absent from training, which is rarely available; slight distribution shifts inflate false positives.

Approach

Generate the held-out set with a generator trained on suffix completion over the suspect data (short 32-64 token snippets, shared prefix), then calibrate by comparing a text-only real-vs-synthetic classifier against a membership-aware classifier (adding target-model MIA scores), using a difference-comparison t-test so only membership-driven shift counts; also constrains MIA weights to be positive

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use the combined classifier instead of the text-only classifierDataset inference on single-author blog members · Single-author blogs, memberAUC (%): 53.8 -> 55.6Helped
Use synthetic suffix-completion data with calibration and constrained weightsDataset inference on Pile subsets · Pile subsets, member datap-valueHelped
Replace suffix completion with ICL paraphrasing (ablation)Post-hoc dataset inference · Single-author data, memberp-value: p-value 1.0Hurt
Replace suffix completion with ICL text completion (ablation)Post-hoc dataset inference · Single-author data, memberp-value: p-value 1.0Hurt
Replace suffix completion with preference optimization (ablation)Post-hoc dataset inference · Single-author data, memberp-value: p-value 1.0Hurt
Remove segment and shuffle from generator data preparation (ablation)Post-hoc dataset inference · Single-author data, memberp-value: p-value 1.0Hurt
Remove suffix comparison (ablation)Post-hoc dataset inference · Single-author data, memberp-value: p-value 1.0Hurt
Remove post-hoc calibration and use original DI t-test (ablation)Dataset inference with synthetic held-out data · Single-author data, member and non-memberp-value: p-value <0.001 for both setsMixed
Remove the positive weight constraint (ablation)Post-hoc dataset inference · Single-author data, member and non-memberp-value: member 0.004; non-member 0.43Mixed
Increase t-test sample size to 2k queriesDataset inference with generated validation data · Evaluated datasetsp-value: p < 0.01Helped
Use a two-layer GPT2 classifier with initialized weightsText classifier · Classifier architecture comparisonAUC: 53.3Helped

Surprising findings

Limitations

Experimental setup

Datasets: single-author blog posts (1400) The Pile subsets (Pile-CC, Wikipedia, ArXiv, NIH ExPorter, FreeLaw, Ubuntu IRC, PubMed Central, Github, EuroParl, PhilPa

Benchmarks: single-author blog posts The Pile subsets

Metrics: AUC (text and combined classifiers) p-value of difference comparison t-test BoW AUC GPT2 AUC

Baselines: ICL paraphrasing ICL text completion preference optimization (SimPO) original DI without calibration

Builds on: dataset inference LLM dataset inference Min-K%++ SimPO Blind Baselines

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy