DistillDetect-gemma-3-4b-pt-from-gpt-oss-120b-OMI-1K
Unofficial reproduction of a distilled student model from the paper
Reference-Based Distillation Detection in LLMs (Rawat et al.,
arXiv:2607.09692),
retrained with the authors' released code and teacher-generated data
(
github.com/RajatRawat-creator/DistillDetect, MIT).
The original authors did not release student checkpoints; this repo is an independent
reproduction and is
not affiliated with the authors.
- Base (student) model: google/gemma-3-4b-pt
- Teacher: openai/gpt-oss-120b
- Training data: OpenMathInstruct-2 (1K prompts) — 1000 teacher-generated responses, shipped verbatim in the authors' repo (
data/training/Teacher=GPT-OSS-120B_Data=OMI(1K)_Template=Chat.jsonl)
- Prompt template: plain
Problem:\n{question}\n\nSolution:\n
Training
SFT with the authors' released training/ scripts (paper Appendix A recipe):
3 epochs, LR 1e-5, cosine schedule, 5% warmup, per-device batch 4 x grad-accum 4
(effective batch 16), block size 4096, bf16, gradient checkpointing, loss on
response tokens only (prompt masked -100). Teacher responses were pre-truncated
to 2,048 tokens in the released data. Trained on 1x H100
(paper used 2x H200; hyperparameters identical), transformers 4.55.4 / trl 0.19.1,
seed 42 (HF default; paper seed unknown). One compatibility patch: trl renamed
max_seq_length to max_length, so the block size is passed explicitly
(no behavioral change).
Evaluation (ours vs. paper Table 9)
Greedy decoding, template matched to training, scored with math_verify.
GSM8K 4-shot; MATH500 zero-shot. Few-shot counts were calibrated so the
base models reproduce their Table 9 baselines. The paper does not document its
eval protocol, so treat cross-paper comparisons as approximate.
| Benchmark | This reproduction | Paper Table 9 | Gen. budget | Hit budget cap |
|---|
| GSM8K | 51.55 | 51.17 | 4096 tok | 1.1% |
| MATH500 | 26.40 | 27.60 | 16384 tok | 26.8% |
A high "hit budget cap" fraction means the model was still generating when the
token budget ran out, so that accuracy is a lower bound — these students are
SFT'd on teacher traces that were themselves truncated at 2,048 tokens, which
makes some of them generate very long self-checking traces.
Base-model reference (our protocol / paper): GSM8K 27.37 / 38.13,
MATH500 18.80 / 23.60.
License
The base model's license applies: gemma. Teacher-generated training
data redistributed by the paper authors under their repo's MIT license.