While reasoning traces improve model capability, they also provide supervision signals for downstream distillation. Initial efforts to mitigate this issue rely on decoding-time interventions, but they incur high inference overhead and can harm the teacher model's natural fluency. We propose Conflict-Aware Logit Adapters (CALA), a lightweight training-time defense. CALA attaches a small residual adapter to a frozen teacher model and trains only the adapter. A proxy student identifies positions where imitation is most sensitive, and the adapter perturbs the output distribution at these positions while remaining within high-probability regions. A conflict-aware optimization strategy balances anti-distillation and utility objectives, while regularization preserves the teacher's performance and fluency. At inference time, CALA incurs modest overhead and requires no auxiliary model. Experiments on reasoning benchmarks show that CALA substantially reduces successful student imitation while maintaining near-original teacher performance and generation quality. These results establish CALA as a practical and efficient alternative to decoding-time approaches.
Figures & tables
Figure 1: Representative GSM8K example: ADS can introduce visible surface artifacts such as mixed-language reasoning and malformed fragments, while CALA produces a cleaner trace.
Figure 2: Overview of Conflict-Aware Logit Adapters (CALA). A trainable residual adapter modifies the frozen teacher’s logits zt , while proxy-guided anti-distillation and teacher-derived utility anchors jointly shape the adapted distribution.
∇θL=∇θLutil+λ∇θLanti,
Algorithm 1 Conflict-Aware Logit Adapter
Figure 3: Utility-distillability trade-offs across three benchmarks. The x-axis reports defended teacher accuracy, while the y-axis reports the accuracy of a downstream student distilled from the generated traces. Lower student accuracy at comparable teacher accuracy indicates stronger protection. For GSM8K, the lines show three-seed means and the shaded bands show ± one standard deviation in student accuracy; MMLU and MATH report the original evaluation runs.
Figure 4: Fluency evaluation using GPT-4o mini as an automatic judge. The judge scores surface quality while ignoring mathematical correctness and the final answer. For GSM8K, the lines show three-seed means and the shaded bands show ± one standard deviation in fluency; MMLU reports the original evaluation run.
Method
Time / batch
Tokens / s
Avg. memory
Peak memory
Base teacher
25.8s
828.5
15.26
20.07
ADS
89.2s
465.1
32.39
39.28
CALA
31.5s
708.3
15.88
20.24
Table 1: Comparison of inference costs and memory footprints. Memory is reported in GB.
Figure 5: GSM8K ablations. Panel (a) reports the utility-distillability trade-off for objective and support variants as well as individual removals of the base anchor, virtual ascent, and conflict-aware update. Panel (b) reports teacher fluency for variants with available fluency measurements. Lower student accuracy at comparable teacher accuracy indicates stronger protection.
Table 7: Exact utility-distillability results. Teacher and student accuracies are reported in percentage points. Point numbers follow Table 5 .
Point 1
Point 2
Point 3
Point 4
Benchmark
Method
Teacher
Fluency
Teacher
Fluency
Teacher
Fluency
Teacher
Fluency
GSM8K
CALA
89.7
97.30
83.5
93.05
79.2
93.68
65.1
85.63
ADS
89.7
96.54
84.0
89.81
78.9
69.68
64.8
47.84
Temp.
90.0
96.96
88.3
96.26
89.4
95.49
52.5
27.62
MATH
CALA
61.6
93.96
57.7
93.35
56.5
86.65
45.6
73.52
ADS
58.2
85.74
57.9
85.70
58.8
86.41
59.9
85.16
Appendix
Table 8: Exact automatic fluency scores and corresponding teacher accuracies. Fluency is scored from 1 to 100 by the automatic judge described in Appendix E . Point numbers follow Table 5 .
Figure 6: MATH fluency evaluation using GPT-4o mini as an automatic judge.
Method
Configuration
Teacher acc.
Student acc.
Fluency
CALA
(0.50,0.25)
89.43±0.46
56.07±0.81
97.51±0.30
CALA
(0.75,0.75)
82.97±0.68
52.23±1.29
94.06±1.42
CALA
(0.50,0.80)
79.93±0.67
47.70±1.73
87.54±8.69
CALA
(0.50,0.85)
68.23±3.43
39.20±5.51
79.32±8.92
ADS
(0.90,1.58×10−2)
89.20±1.90
60.53±1.55
96.38±0.23
ADS
(0.90,2.37×10−2)
84.50±0.56
56.33±0.60
88.47±1.90
Appendix
Table 9: Three-seed GSM8K results. Accuracy and fluency values are mean ± standard deviation.
Variant
Config. (τ,δ)
Teacher Fluency
Teacher Acc.
Student Acc.
Δ
Full objective
(0.50,0.25)
97.30
89.7
55.5
34.2
Full objective
(0.75,0.75)
93.05
83.5
50.8
32.7
Full objective
(0.50,0.80)
93.68
79.2
47.3
31.9
Full objective
(0.50,0.85)
85.63
65.1
34.2
30.9
w/o Lmid
(0.50,0.25)
97.19
89.8
57.1
32.7
w/o Lmid
(0.90,0.25)
90.78
88.4
54.1
34.3
Appendix
Table 10: Exact GSM8K ablation results. Teacher fluency, teacher accuracy, and student accuracy are reported in percentage points. Δ=Accteacher−Accstudent ; dashes denote unavailable fluency measurements.
Method
Step-correct
Consistent
Complete
Strictly valid
Correct but invalid
CALA
36/50
45/50
47/50
36/50 (72%)
6/42
ADS
31/50
42/50
45/50
31/50 (62%)
9/40
Appendix
Table 11: Blinded manual audit of GSM8K reasoning traces. “Correct but invalid” denotes a correct final answer with a rationale that fails at least one strict-validity criterion.
Method
Configuration
Teacher/student acc.
CALA
(τ,δ)=(0.50,0.25)
39.8/26.2
CALA
(τ,δ)=(0.75,0.75)
37.9/27.2
CALA
(τ,δ)=(0.50,0.80)
36.9/23.3
CALA
(τ,δ)=(0.50,0.85)
23.3/19.4
ADS
(τ,λADS)=(0.90,1.58×10−2)
36.9/23.1
ADS
(τ,λADS)=(0.90,2.37×10−2)
35.0/27.5
Appendix
Table 12: GPQA results under the MetaMathQA → MMLU-traces → GPQA distribution shift.
Figure 7: Utility-distillability operating points under the MetaMathQA → MMLU-traces → GPQA distribution shift. Point numbers follow the configuration order in Table 12 ; lower student accuracy at comparable teacher accuracy indicates stronger protection.
Teacher acc.
Gemma-2-2B-IT
Qwen2.5-3B
89.7/89.7
35.5/39.1
78.6/77.7
83.5/84.0
32.5/38.1
73.9/75.1
79.2/78.9
28.6/35.1
71.8/68.2
65.1/64.8
21.5/29.7
67.7/54.3
Appendix
Table 13: Cross-student transfer on GSM8K. Each pair is CALA/ADS accuracy in percentage points. Lower student accuracy is better for the defense.
Figure 8: Proxy-size sensitivity on GSM8K. We compare the default 1.5B training-time proxy with a 3B proxy.
Language model distillation transfers a larger model’s capabilities to smaller or specialized models using outputs, probabilities, or internal representations.
Distillation via reasoning traces exposes frontier models to adversarial third parties, who can bypass their guardrails and misappropriate their capabilities. Existing antidistillation defenses perturb the teacher's sampling distribution to hinder downstream student learning, but these perturbations also degrade the trace quality even when task accuracy is preserved, often leaving visible artifacts such as syntactic incoherencies and repetition chains. Beyond eroding user trust, such traces are statistically distinguishable, allowing adversaries to filter or resample them and weaken the defense. An effective defense must therefore limit how visibly it degrades the teacher. To address this gap, we formulate antidistillation as a max-min game whose constraint set explicitly encodes defense detectability. Satisfying both axes--- degrading the student while remaining undetectable---may require more clever perturbations, akin to sparse but targeted adversarial attacks in robustness literature. We apply the same analogy to reasoning steps, targeting thought anchors: sentences identified in prior interpretability work as strongly influencing a model's output. We propose Thought Anchor RemovalS (TARS), which identifies and removes thought anchors, whose content is semantically redundant, so that the reasoning trace remains legible to the user. Modeling these removals as burst-deletions, we then provide formal guarantees on its detectability. Finally, we empirically demonstrate the efficacy of TARS in consistently hindering downstream distillation across teacher model scales (DeepSeek R1 Distill Qwen 7B, Gemma 3 12B), reasoning benchmarks (MMLU Pro, MATH, DeepMath), and student architectures (Llama 3.2 1B, Llama 3.2 3B, Gemma 3 1B, Qwen 3 1.7B) while preserving teacher performance.
Max Hartman, Vidhata Jayaraman, Moulik Choraria +1
Reasoning distillation from powerful teacher models to smaller students faces the Gap Curse: as teachers grow more sophisticated, their complex distributions increasingly diverge from what students can approximate, causing performance degradation. Existing mitigation strategies either filter out challenging examples through data selection or introduce weaker intermediate assistant models, inherently compromising supervision coverage or quality. We propose Teacher Alignment, which directly adapts the teacher toward the student's distribution without discarding data or degrading reasoning quality. However, naive alignment through standard knowledge distillation triggers catastrophic collapse of the teacher's reasoning capabilities. To address this, we reformulate teacher alignment as reinforcement learning and introduce TeacherGRPO, built on Group Relative Policy Optimization with two key innovations: (i) Curriculum Selective Alignment applies dual token- and distribution-level curricula to focus rewards on high-signal reasoning gaps while filtering noise from trivial tokens and uncertain tail distributions, and (ii) Importance-Adaptive Length Regularization selectively penalizes verbose redundancy while preserving pedagogically critical reasoning steps. The aligned teacher then distills knowledge to students via standard pipelines. Extensive experiments show TeacherGRPO significantly outperforms baselines across diverse reasoning benchmarks and distillation methods. Our code is available at https://github.com/LzyFischer/TeacherGRPO.
Zhenyu Lei, Zihan Chen, Yaochen Zhu +5
University of Virginia · Netflix · University of Washington +2
Distillation attacks copy the reasoning capabilities of closed-source large language models, allowing bad actors to replicate state-of-the-art performance at low cost. Attackers systematically collect a large volume of frontier model reasoning traces and then train (i.e., "distill") their own models on these traces. Existing defenses against distillation attacks are typically evaluated immediately after distillation, implicitly assuming attackers do not train their models any further. In this paper, we argue that a more realistic threat model includes further training with reinforcement learning after distillation. A misspecified threat model can give a false sense of security -- some defenses that seem effective after distillation can be broken after subsequent reinforcement learning. Practically, reinforcement learning lowers the bar for a distillation attack to be effective. We show that simple attacks can steal reasoning capabilities from existing closed-source language models using data easily obtainable from current APIs, yielding reasoning improvements equivalent to more sophisticated attacks that extract the full hidden traces. Results indicate that any distillation defense that leaks sufficient information to reconstruct approximate reasoning traces is likely ineffective. We conclude by discussing broader implications and batch-level distillation defenses which could be more effective.
Shidan Javaheri, Alexander Panfilov, Oliver Britton +2
University of Oxford · ELLIS Institute Tübingen & MPI for Intelligent Systems