OP-Print: Tracing Membership Imprints in On-Policy Self-Distillation

cs.LG cs.CR Qi Li, Xinchao Wang · Oct 22, 2026
Local to this browser
What it does
On-policy self-distillation trains language models by distilling privileged self-supervision over the model's own rollouts, yet its privacy risks have remained almost entirely unexplored. This paper presents the first study of membership...
Why it matters
The authors propose OP-Print, a unified gray-box and black-box attack framework that either aggregates privileged teacher-student residuals directly or infers them from restart-based output-basin stability. Their empirical results suggest...
Main concern
The paper makes a timely and novel contribution by identifying a new membership leakage channel in on-policy self-distillation and providing both theoretical intuition and empirical validation for a structured correction imprint. The...
Community signal
0
0 up · 0 down
Sign in to vote with arrows
AI Review AI reviewed
Plain-language introduction

On-policy self-distillation trains language models by distilling privileged self-supervision over the model's own rollouts, yet its privacy risks have remained almost entirely unexplored. This paper presents the first study of membership inference attacks against such systems, showing that the training imprint is sparse and concentrates on uncertainty-sensitive reasoning tokens rather than spreading uniformly across outputs. The authors propose OP-Print, a unified gray-box and black-box attack framework that either aggregates privileged teacher-student residuals directly or infers them from restart-based output-basin stability. Their empirical results suggest that on-policy self-distillation introduces a distinct, previously overlooked privacy leakage surface that conventional MIA heuristics fail to capture, raising important concerns for privacy auditing of modern post-training pipelines.

Critical review
Verdict
Bottom line

The paper makes a timely and novel contribution by identifying a new membership leakage channel in on-policy self-distillation and providing both theoretical intuition and empirical validation for a structured correction imprint. The gray-box attack is conceptually clean and the black-box surrogate—measuring output-basin stability via prefix restarts—is creative. However, the evaluation is limited to a narrow setting: only 500 member examples from a single math-reasoning dataset, models up to 8B parameters, and a fixed-teacher distillation protocol that may not reflect standard practice. While the evidence clearly supports the claim that generic baselines fail and OP-Print outperforms them, the practical impact of the attack is tempered by high query costs and restricted scope.

“the supervision imprint in on-policy self-distillation is sparse and structured, concentrating on uncertainty-sensitive correction sites rather than across the entire rollout”
OP-Print paper · Abstract
“We use a fixed-teacher setup in all reported target-model training runs.”
OP-Print paper · Appendix A
What holds up

The strongest part of the paper is its careful characterization of where the membership signal lives. Figure 2 demonstrates that teacher-student divergence is organized by token difficulty rather than absolute position: divergence rises substantially in higher-uncertainty regions and occupies only a restricted region of the reliability map. This observation is well exploited methodologically—the gray-box score $S_{\mathrm{gray}}(x)$ directly targets the KL gap between privileged and standard conditioning, and the black-box score $S_{\mathrm{black}}(x)$ uses restart-text stability as a principled surrogate. Table 1 shows consistent gains over likelihood-based baselines, with OP-Print-G-Max reaching AUCs up to 0.611 on Qwen3-4B where the best baseline stagnates near 0.52.

“Divergence increases with student uncertainty, showing that correction concentrates on difficult reasoning tokens.”
OP-Print paper · Figure 2 caption
“members should therefore exhibit smaller privileged correction on average, because self-distillation has already adapted the model on those examples.”
OP-Print paper · Section 3
Main concerns

The experimental scope is worryingly narrow for general claims about on-policy self-distillation privacy. All results derive from just 500 member and 500 non-member math problems drawn from OpenThoughts, trained for only three epochs on models no larger than 8B parameters. This small scale raises doubts about whether the leakage pattern persists in larger, more realistic post-training regimes. The authors also adopt a fixed-teacher setup—which diverges from many on-policy self-distillation methods that use an online or evolving teacher—meaning the leakage surface may differ in practice. A second major concern is query cost: the black-box attack requires sampling $M$ full generations, truncating at $|P|$ prefix ratios, and restarting $R$ times each, making it far more expensive than simple loss-based MIAs. The ablation in Figure 5(c) confirms diminishing returns as restarts increase, suggesting the attack is inherently costly.

“We use a fixed-teacher setup in all reported target-model training runs.”
OP-Print paper · Appendix A
“The black-box attack requires repeated restart generation, which introduces additional inference cost (but the adversaries can select the attack configuration themselves).”
OP-Print paper · Limitations
Evidence and comparison

The paper's central empirical claim—that generic MIAs remain near random guessing while OP-Print succeeds—is supported by the data in Table 1, where Min-K% Prob and R\'enyi variants hover around AUC 0.49-0.52. This aligns with broader findings in the literature: Duan et al. showed that for LLM pre-training, "the performance across most MIAs and target domains is near-random." However, the baseline selection is weak. The authors pit OP-Print against pre-training detection heuristics like Min-K% Prob rather than stronger reference-based or calibrated attacks that exploit the exact reference solutions available in their threat model. A fairer comparison would include a reference-model-calibrated loss or a direct likelihood ratio using the initial teacher checkpoint, which is already available in their gray-box setting. Without this, the gap may be inflated by baseline mismatch rather than a fundamentally new attack surface.

“Standard likelihood- and entropy-based MIA baselines, including Min-K% Prob and Rényi scores, remain near random guessing, with AUCs around 0.49–0.52.”
OP-Print paper · Section 4.1
“For the most part, we find that the performance across most MIAs and target domains is near-random.”
Reproducibility

Training hyperparameters are documented in Appendix A (Table 3), including learning rate $5 \times 10^{-6}$, bfloat16 precision, and vLLM-based generation settings, which will help reproduction. The dataset construction protocol is also clearly described: the authors split the data with a fixed seed and sample 500 examples per split. Yet no code or checkpoint release is mentioned, and the black-box attack's query budget is substantial, requiring many generations and restarts per query. Independent reproduction is therefore feasible in principle but blocked by the absence of published code and by the high inference cost of the black-box variant, which the authors acknowledge is left to the adversary to configure.

“we first randomly partition the original training set into two disjoint halves using a fixed random seed (i.e., 42), denoted as the train split and the held-out split. We then uniformly sample 500 examples from each split to construct the final member and non-member pools used throughout the paper.”
OP-Print paper · Appendix A
“The black-box attack requires repeated restart generation, which introduces additional inference cost (but the adversaries can select the attack configuration themselves).”
OP-Print paper · Limitations
Abstract

On-policy distillation has been widely adopted for alleviating the distribution mismatch of off-policy distillation. Self-distillation further removes the need for an external teacher by using the student's own outputs, enhanced with extra context or demonstrations, as supervision. This teacher-free design makes on-policy self-distillation practical, yet its privacy risks remain largely unexplored. In particular, using two coupled views of the same training example may strengthen its memorization. Motivated by this, we present the first study of membership inference attacks against on-policy self-distillation. We investigate both gray-box and black-box threat models and propose a unified attack framework, named OP-Print. OP-Print builds on our finding that the supervision imprint in on-policy self-distillation is sparse and structured, concentrating on uncertainty-sensitive correction sites rather than across the entire rollout. Exploiting this property, we construct attack signals tailored to the two access levels: in the gray-box setting, OP-Print directly aggregates the privileged teacher-student residual correction along the same rollout; in the black-box setting, it measures restart-based output-basin stability as an observable surrogate for the hidden correction process. Across diverse models and distillation schemes, all baseline attacks remain essentially at random guessing even under gray-box access, whereas OP-Print is broadly validated to be effective in both gray-box and black-box settings.

Challenge the Review

Pick a starting point or write your own. Challenges run in the background, so you can keep reading while the AI investigates.

No challenges yet. Disagree with the review? Ask the AI to revisit a specific claim.