Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Detecting Distillation Data from Reasoning Models

About

Reasoning distillation has emerged as a prevailing paradigm for transferring reasoning capabilities from large reasoning models to small language models. Yet, reasoning distillation risks data contamination: benchmark data may inadvertently be included in the distillation data, thereby inflating model performance metrics. In this work, we formally define the distillation data detection task, which determines whether a given question is included in the model's distillation data. The unique challenge of this task lies in the partial availability of distillation data. To address this, we propose Token Probability Deviation (TPD), a detection method that leverages the probability patterns of output tokens generated by the model instead of input tokens. Our method is motivated by the observation that seen questions tend to elicit more near-deterministic tokens generated by the models than unseen ones. Our TPD score is thus designed to quantify the token-level deviation of generated tokens from a high-confidence reference probability. Consequently, seen questions can yield substantially lower TPD scores than unseen ones, enabling strong detection performance. Extensive experiments demonstrate the effectiveness of our approach, improving detection AUC by up to 31% on distillation datasets.

Hengxiang Zhang, Hyeong Kyu Choi, Sharon Li, Hongxin Wei• 2025

Related benchmarks

TaskDatasetResultRank
Distillation data detectionS1
AUC95.3
63
Distillation data detectionLIMO
AUC72.8
39
Distillation data detectionS 1.1
AUC64.9
39
Training data detectionS1
TPR@1%FPR47
39
Training data detectionLIMO
TPR@1%FPR33.5
39
Training data detectionS 1.1
TPR@1%FPR11
39
Machine-text detectionS1
AUC91.8
15
Distillation data detectionDistillation Data (test)
Accuracy86
8
Distillation data detectionOpenR1-Math 220k (balanced evaluation set)
AUC0.665
8
Showing 9 of 9 rows

Other info

Follow for update