atlas

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

Membership Inference Attacks against Fine-tuned Large Language Models via Self-prompt Calibration

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: Uncertainty quantification for LLMs · 28 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 SPV-MIA, a membership inference attack on fine-tuned LLMs that builds a self-prompt reference model from the target's own generations and uses a paraphrase-based probabilistic variation signal tied to memorization rather than overfitting.

Problem

Membership inference against fine-tuned LLMs is unreliable in practice: reference-based attacks need a reference dataset closely matching training data, and probability-based signals depend on overfitting, leading to high false-positive rates.

Approach

Fine-tune a reference model on texts the target LLM generates when prompted with domain text chunks (self-prompt calibration), and score membership by probabilistic variation: the average probability of symmetric paraphrases minus the original probability, approximating a second-order derivative to capture memorization-induced local maxima.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace LiRA-Candidate with SPV-MIAMembership inference attack · Wiki, four-model averageAUC: 0.744 -> 0.938Helped
replace LiRA-Candidate with SPV-MIAMembership inference attack · AG News, four-model averageAUC: 0.707 -> 0.909Helped
replace LiRA-Candidate with SPV-MIAMembership inference attack · XSum, four-model averageAUC: 0.797 -> 0.924Helped
add practical difficulty calibration (PDC) (ablation)SPV-MIA · Wiki, GPT-J and LLaMAAUC: 0.648 -> 0.929; 0.653 -> 0.951Helped
add practical difficulty calibration (PDC) (ablation)SPV-MIA · AG News, GPT-J and LLaMAAUC: 0.632 -> 0.885; 0.641 -> 0.903Helped
add practical difficulty calibration (PDC) (ablation)SPV-MIA · XSum, GPT-J and LLaMAAUC: 0.653 -> 0.897; 0.661 -> 0.937Helped
add probabilistic variation assessment (PVA) (ablation)SPV-MIA · Wiki, GPT-J and LLaMAAUC: 0.901 -> 0.929; 0.913 -> 0.951Helped
add probabilistic variation assessment (PVA) (ablation)SPV-MIA · AG News, GPT-J and LLaMAAUC: 0.864 -> 0.885; 0.885 -> 0.903Helped
add probabilistic variation assessment (PVA) (ablation)SPV-MIA · XSum, GPT-J and LLaMAAUC: 0.873 -> 0.897; 0.919 -> 0.937Helped
use embedding-domain instead of semantic-domain paraphrasing (ablation)SPV-MIA · WikiAUC: 0.951 -> 0.965Helped
use embedding-domain instead of semantic-domain paraphrasing (ablation)SPV-MIA · AG NewsAUC: 0.903 -> 0.926Helped
use embedding-domain instead of semantic-domain paraphrasing (ablation)SPV-MIA · XSumAUC: 0.937 -> 0.949Helped

Surprising findings

Limitations

Experimental setup

Datasets: Wikitext-103 AG News XSum Wikicorpus TLDR News CNNDM

Benchmarks: Wikitext-103 AG News XSum

Metrics: AUC TPR@1%FPR TPR@0.1%FPR

Baselines: Loss Attack Neighbour Attack DetectGPT Min-K% Min-K%++ LiRA-Base LiRA-Candidate

Builds on: LiRA neighbour attack Min-K% DetectGPT LoRA

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