In-Context Probing for Membership Inference in Fine-Tuned Language Models

Zhexi Lu, Hongliang Chi, Nathalie Baracaldo, Swanand Ravindra Kadhe, Yuseok Jeon +1 more
12/18/2025
cs.CRcs.LG

No official code found. Best implementation: RPI-DSPlab/ICP-MIA (12 stars, MIT, last commit 2025-12-30).

Abstract

Membership inference attacks (MIAs) pose a critical privacy threat to fine-tuned large language models (LLMs), especially when models are adapted to domain-specific tasks using sensitive data. While prior black-box MIA techniques rely on confidence scores or token likelihoods, these signals are often entangled with a sample's intrinsic properties - such as content difficulty or rarity - leading to poor generalization and low signal-to-noise ratios. In this paper, we propose ICP-MIA, a novel MIA framework grounded in the theory of training dynamics, particularly the phenomenon of diminishing returns during optimization. We introduce the Optimization Gap as a fundamental signal of membership: at convergence, member samples exhibit minimal remaining loss-reduction potential, while non-members retain significant potential for further optimization. To estimate this gap in a black-box setting, we propose In-Context Probing (ICP), a training-free method that simulates fine-tuning-like behavior via strategically constructed input contexts. We propose two probing strategies: reference-data-based (using semantically similar public samples) and self-perturbation (via masking or generation). Experiments on three tasks and multiple LLMs show that ICP-MIA significantly outperforms prior black-box MIAs, particularly at low false positive rates. We further analyze how reference data alignment, model type, PEFT configurations, and training schedules affect attack effectiveness. Our findings establish ICP-MIA as a practical and theoretically grounded framework for auditing privacy risks in deployed LLMs.

View on arXivView PDF

Code Implementations(1)

RPI-DSPlab/ICP-MIACommunity66%
MIT

The repository for the paper "In-Context Probing for Membership Inference in Fine-Tuned Language Models", accepted by NDSS 2026.

122PythonDec 5, 20259 months agoMIT

Cite this paper

@article{lu2025incontext,
  title  = {In-Context Probing for Membership Inference in Fine-Tuned Language Models},
  author = {Zhexi Lu and Hongliang Chi and Nathalie Baracaldo and Swanand Ravindra Kadhe and Yuseok Jeon and Lei Yu},
  year   = {2025},
  eprint = {2512.16292},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2512.16292v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2512.16292.svg)](https://codeofpaper.com/paper/2512.16292)

Discussion