arXiv paper· machine learning privacy and security (LLM membership inference)· method paper

Black-Box Membership Inference via Word-Level Probability Estimation

Shengjie Niu, Yeheng Ge, Jian Huang

cs.CR · cs.LG · stat.ML

Quick verdict

WPMIA recovers likelihoods from sampled text for black-box MIA; it beats black-box baselines incl. proprietary LLMs, but small-n eval tempers audit claim.

Summary

The paper tackles membership inference against LLMs under strict black-box access, where only textual continuations are observable and likelihood-based attacks requiring per-token logits fail. WPMIA estimates each word's generation probability by sampling M continuations per context and smoothing matches with a semantic kernel over word embeddings; word-level log-probabilities are aggregated into a sequence-level log-likelihood, and a contrastive-prefix variant scores the normalized difference between likelihoods conditioned on member versus non-member prefixes. On WikiMIA with four open-source models (lengths 32/64/128), WPMIA averages 84.8/84.5/81.7 AUC, improving over the best black-box baseline (SimMIA) by 4.6 AUC and 14.1 TPR@5%FPR points on average; it is the strongest black-box method on most MIMIR domain-size pairs. On a 240-example WikiMIA-25 subset it reaches AUC 66.3–91.0 on Claude-4.5-Haiku, Gemini-2.5-Flash, and GPT-5-Chat, averaging TPR@5%FPR of 42.0. Ablations cover components, estimator-by-prefix crosses, embeddings, temperature, prefix corruption, seed variance, and token-budget-matched comparisons. Main limitations: performance depends on verified member prefixes (heavy mislabeling drops AUC to about 48; synthetic prefixes yield 75.3 versus 84.8 clean, untested on proprietary models); proprietary results use 120/120 examples at M=10 without confidence intervals; queries cost roughly 3ML requests (about 237k, $78–267 per model); the estimator's bias is unanalyzed; gray-box ReCaLL still leads where applicable; evaluation is English-only.

Review complete

Assessment at a glance

machine learning privacy and security (LLM membership inference)

5

risk flags identified

● high 0

● medium 3

● low 2

Reproducibility readinesshigh
Code & data availableyes
Review confidencehigh
AI text signal0/20 passages flagged
Deep research viableCode-reproducibleNeeds GPU

5 strengths · 5 weaknesses · 5 author questions

AI text signal = Binoculars zero-shot detector score over 20 sampled passages (Fast-DetectGPT co-reported) — absence does not prove human authorship, and a flag does not imply misconduct.

Levels reflect verification complexity — not a quality score. Evolve never rates your paper.

Key claims under review

  • Word-level generation probabilities estimated by Monte Carlo sampling with a semantic kernel can be aggregated into a sequence-level log-likelihood that serves as a membership score under strict black-box access.
  • Contrastive prefix conditioning — contrasting likelihoods under member versus non-member prefixes with sequence-level normalization — amplifies the membership signal, whereas word-level normalization destabilizes scores (Table 3).
  • WPMIA outperforms existing black-box MIAs (PETAL, SaMIA, SimMIA) across WikiMIA, MIMIR, and WikiMIA-25, e.g., +4.6 AUC and +14.1 TPR@5%FPR averaged over WikiMIA lengths.
  • WPMIA transfers to proprietary LLMs (GPT-5-Chat, Gemini-2.5-Flash, Claude-4.5-Haiku) with AUC 66.3–91.0 and an average TPR@5%FPR of 42.0 on a 240-example WikiMIA-25 subset.
  • The attack relies on correctly labeled member and non-member prefix pools: corruption of the prefix labels collapses performance toward chance (Table 6).

Strengths

  • Clear threat-model framing with a real practical gap

    Fig. 1 and §2.1 precisely separate gray-box (token logits) from strict black-box (text-only continuations); targeting proprietary APIs like GPT-5-Chat and Claude-4.5-Haiku addresses an underexplored setting where existing likelihood-based MIAs simply cannot run.

  • Design choices are validated, not just asserted

    Table 3 shows sequence-level aggregation before contrastive normalization is essential (word-level normalization drops AUC to 46.9–54.7, below the 55.4–62.0 unprefixed baseline); Table 4's crossed ablation isolates estimator vs. prefix effects, with the semantic kernel beating exact matching by 6.3 AUC on average under full contrast.

  • Consistent, budget-fair black-box state of the art

  • Unusually thorough robustness and cost reporting

  • Reproducibility scaffolding is present

Weaknesses

  • Verified-member-prefix dependence undercuts the external-audit story

  • Small-sample proprietary headline without uncertainty quantification

  • Information asymmetry versus key baselines

  • No formal analysis of the 'principled' estimator

  • Query cost and prompt compliance unquantified

Full Review Report

$1.99 · one-time

Unlock the complete preliminary analysis Evolve produced for this paper.

Report preview

Scope & Claims

The paper targets pretraining-data detection (membership inference, MIA) on LLMs under a **strict black-box threat model**: the adversary observes only textual continuations, with no access to weights, tokenizer, or per-token logits (Fig. 1, §2.1). Three central claims are made: (1) word-level generation probabilities recovered from sampled continuations aggregate into a sequence-level log-likelihood usable as an MIA score; (2) contrasting likelihoods conditioned on member vs. non-member prefixes (Eq. 9) amplifies the membership signal; (3) WPMIA beats existing black-box MIAs on WikiMIA, MIMIR, and WikiMIA-25 and transfers to proprietary models (GPT-5-Chat, Gemini-2.5-Flash, Claude-4.5-Haiku) with an average TPR@5%FPR of 42.0 (Tables 1, 2, 16, 17).

The remaining sections cover Methodology Assessment, Statistical Evidence, Results & Reproducibility, Limitations & Risks, and Verdict & Recommendations.

  • All 5 strengths & 5 weaknesses, analyzed in detail
  • 5 risk flags with severity, section & reviewer questions
  • Methods & statistical evidence assessment
  • 5 questions a peer reviewer would ask
  • Full six-dimension evaluation report & recommendations

One-time purchase · this paper only

One-time purchase, this paper only. AI-generated preliminary analysis — not peer review.

Frequently asked questions

What is WPMIA in one sentence?

WPMIA is a membership inference attack that estimates how likely a language model is to generate each word of a text by sampling many continuations and semantically smoothing matches in embedding space, aggregates these into a sequence-level log-likelihood, and contrasts likelihoods under member versus non-member prefixes to decide whether the text was in the model's training data — using only text outputs, no logits.

Does WPMIA actually work on GPT-5, Gemini, and Claude?

In the paper's evaluation on a 240-example WikiMIA-25 subset, WPMIA reached AUC 66.3 on Claude-4.5-Haiku, 78.2 on Gemini-2.5-Flash, and 91.0 on GPT-5-Chat, with TPR@5%FPR of 22.1, 36.5, and 67.3 respectively (average 42.0), querying official APIs about 237k–241k times per model at a cost of $78–267. These are point estimates on a small benchmark subset with no reported confidence intervals.

What are WPMIA's main limitations?

It needs many API queries (about 3×M×L per text; ~237k requests and $78–267 per proprietary model in the paper), it depends on correctly labeled member and non-member prefix examples — mislabeled prefixes collapse AUC to about chance (~48) and a synthetic-prefix fallback scored 75.3 vs. 84.8 clean, untested on proprietary models — it was evaluated only on English, and its probability estimator has no formal bias or error analysis.

How is WPMIA different from SimMIA?

Both methods sample model continuations, but SimMIA averages raw word-level score ratios (frequency or embedding similarity), while WPMIA first smooths matches with a semantic kernel, aggregates word log-probabilities into a sequence-level log-likelihood, and only then applies the prefix contrast. The paper's ablation (Table 3) shows sequence-level normalization scores ~86.5 AUC versus below-baseline ~48.8–54.7 for word-level normalization, and WPMIA beats SimMIA by 4.6 AUC on average on WikiMIA.

About this review

This is an AI-generated preliminary review by Evolve.Pub, produced from the openly available text of the paper. It is not peer review and does not represent the authors' views. The canonical record of the paper is at the publisher link above.

  • Figure panels (Figs. 2–5) are image content; only their captions were extracted, so curve values inside figures (continuation-ratio, sample-size, prefix-shot, and temperature sweeps) could not be independently verified.
  • The extraction runs continuously from the abstract through references and Appendices A–E, including Tables 7–17 and Algorithm 1; no truncation was observed at the far end of the provided text.

Go deeper on this paper

Wrote a paper? Stress-test it before reviewers do.

Upload your manuscript for the same structured scrutiny — reasoning, statistics, evidence, figures, and reproducibility.

Scan your paper
Black-Box Membership Inference via Word-Level Probability Estimation | Evolve.Pub