Adaptive-Policy Distillation: Knowledge Injection under Information Asymmetry
Organizations: KAIST · DGIST
Abstract
On-policy distillation (OPD), which uses reverse-KL supervision on student rollouts to transfer a context-conditioned model's behavior into student model parameters for closed-book inference, has shown strong results. However, we find that this success does not translate to knowledge injection: across two document-QA benchmarks, OPD injects document knowledge far less efficiently than simple off-policy distillation. We trace this reversal to two failures that are structural, yet sparse across token positions. Reverse KL weights its gradient by the student distribution and therefore barely lifts document-specific tokens the student considers unlikely; after forward KL restores this signal, student rollouts expose later positions on the target trajectory only after generating the preceding tokens correctly, so coverage deteriorates with target length. Motivated by the concentration of teacher-student divergence at these hard positions, we propose Adaptive-Policy Distillation (APD), whose per-token gate jointly shifts the rollout toward the teacher and the objective toward forward KL where needed, withdrawing both interventions as the student learns. Without labels or a verifier, APD improves over OPD by up to 54.6%, achieving the best scores of 0.558 F1 on QASPER and 0.486 MRR on BioASQ.
Figures & tables
| Method | Loss | Llama-3.1-8B-Instruct | Qwen3-8B | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| QASPER | BioASQ | QASPER | BioASQ | ||||||||||
| P | R | F1 | strict | len. | MRR | P | R | F1 | strict | len. | MRR | ||
| no training (closed-book) | .158 | .164 | .148 | .121 | .199 | .151 | .140 | .250 | .157 | .115 | .209 | .151 | |
| base document (open-book) | .756 | .860 | .764 | .733 | .769 | .748 | .699 | .805 | .701 | .678 | .762 | .713 | |
| No teacher resampling during training | |||||||||||||
| SFT | CE | .475 | .622 | .485 | .412 | .488 | .446 | .352 | .506 | .373 | .310 | .463 | .373 |
| loss | |||
|---|---|---|---|
| sampling | RKL | gated | FKL |
| student | .324 | .472 | .450 |
| gated | .309 | .486 | .479 |
| teacher | .310 | .481 | .464 |
| Setting | gate | BioASQ | ||||
|---|---|---|---|---|---|---|
| start | mean | end | strict | lenient | MRR | |
| (OPD) | .00 | .00 | .00 | .317 | .333 | .324 |
| .03 | .02 | .02 | .424 | .472 | .444 | |
| .09 | .06 | .05 | .453 | .489 | .469 | |
| .22 | .14 | .09 | .465 | .510 | .485 | |
| (APD) | .44 | .26 | .19 | .469 | .508 | .486 |
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| Question | Answer | |
|---|---|---|
| QASPER, one unit (paper 1806.09103) | ||
| training | Why does the performance of deep neural networks drop significantly when the ground-truth answer contains out-of-vocabulary words in cloze-style reading comprehension? | The performance of deep neural networks drops significantly when the ground-truth answer contains out-of-vocabulary (OOV) words in cloze-style […] [397 words] |
| training | Why might the element-wise product (mul) be superior in combining embeddings? Answer in 15 words or fewer. | Element-wise product captures interactions and reduces distribution differences between embeddings. |
| generated eval | what mechanism is introduced by Cui2016Attention? | attended attention (AoA) |
| human-written | how are rare words defined? | low-frequency words |
| BioASQ, one unit (article PMC3472786) | ||
| Time per step (s) | ||||||
| Rollout | Update | Total | Length | Memory | ||
| Llama-3.1-8B-Instruct | ||||||
| QASPER | OPD | 2.6 | 10.7 | 13.1 | 176 | 167 |
| APD | 5.5 | 7.1 | 11.5 | 118 | 172 | |
| BioASQ | OPD | 2.5 | 16.0 | 18.4 | 139 | 171 |
| APD | 5.7 | 7.6 | 12.7 | 98 | 176 | |
| Llama-3.1-8B-Instruct | Qwen3-8B | |||||
|---|---|---|---|---|---|---|
| strict | len. | MRR | strict | len. | MRR | |
| .437 | .472 | .453 | .351 | .466 | .399 | |
| .449 | .482 | .465 | .353 | .477 | .406 | |
| .471 | .496 | .483 | .382 | .520 | .440 | |
| .512 | .550 | .530 | .386 | .506 | .436 | |
| .468 | .491 | .478 | .394 | .538 | .454 | |
| SWITCH, decay factor | ||||||
|---|---|---|---|---|---|---|
| original paper | 78 | 53 | 7 | – | – | – |
| Llama, QASPER | 86.0 | 81.0 | 72.8 | 31.2 | 10.7 | 2.2 |
| Llama, BioASQ | 84.2 | 74.4 | 67.5 | 31.3 | 10.9 | 2.4 |
| Qwen, QASPER | 90.6 | 84.4 | 78.7 | 49.6 | 28.5 | 11.5 |
| Qwen, BioASQ | 89.1 | 82.9 | 74.8 | 45.6 | 27.0 | 12.3 |
| SKD, acceptance rank | ||||
|---|---|---|---|---|
| 25 | 5 | 2 | 1 | |
| Llama, QASPER | 0.7 | 4.3 | 12.5 | 31.7 |
| Llama, BioASQ | 1.0 | 4.2 | 12.1 | 24.0 |
| Qwen, QASPER | 0.8 | 4.6 | 12.8 | 25.8 |
| Qwen, BioASQ | 0.8 | 3.8 | 11.6 | 22.2 |
| QASPER | BioASQ | |||||
| Method | P | R | F1 | strict | lenient | MRR |
| Llama-3.1-8B-Instruct | ||||||
| no training | .158 .046 | .164 .044 | .148 .040 | .121 .074 | .199 .097 | .151 .082 |
| base document | .756 .052 | .860 .048 | .764 .042 | .733 .050 | .769 .044 | .748 .046 |
| No teacher resampling during training | ||||||
| SFT | .475 - .085 | .622 .074 | .485 - .070 | .412 - .063 | .488 .057 | .446 - .057 |
| Method | 5 ep | 10 ep | 15 ep |
|---|---|---|---|
| No teacher resampling during training | |||
| SFT | .466 | .446 | .444 |
| PD | .391 | .399 | .402 |
| ToDi | .430 | .435 | .431 |
| GKD | .327 | .400 | .425 |
| OPD | .256 | .324 | .375 |
| Loss | fixed corpus | resampled | ( ) |
|---|---|---|---|
| SFT (CE) | .446 | .441 | ( ) |
| PD (FKL) | .399 | .464 | ( ) |
| GKD (JSD) | .400 | .399 | ( ) |
| Llama-3.1-8B-Instruct | Qwen3-8B | |||||||||||||||
| QASPER | BioASQ | QASPER | BioASQ | |||||||||||||
| gate | gate | gate | gate | |||||||||||||
| Setting | start | mean | end | F1 | start | mean | end | MRR | start | mean | end | F1 | start | mean | end | MRR |
| OPD | .00 | .00 | .00 | .390 | .00 | .00 | .00 | .324 | .00 | .00 | .00 | .302 | .00 | .00 | .00 | .276 |
| .03 | .03 | .02 | .478 | .03 | .02 | .02 | .444 | .05 | .03 | .02 | .394 | .05 | .03 | .02 | .383 | |
| .10 | .07 | .05 | .543 | .09 | .06 | .05 | .469 | .14 | .08 | .05 | .410 | .12 | .06 | .05 | .423 | |
| Llama | Qwen | |||
| Method | QASPER | BioASQ | QASPER | BioASQ |
| SFT | 0 | 0 | 0 | 0 |
| PD | 2 | 0 | 0 | 0 |
| ToDi | 3 | 0 | 0 | 0 |
| GKD | 2 | 0 | 0 | 0 |
| OPD | 3 | 0 | 0 | 1 |
| Llama | Qwen | |||
| Method | QASPER | BioASQ | QASPER | BioASQ |
| SFT start | .485 | .446 | .373 | .373 |
| APD, from base | .558 | .486 | .442 | .426 |
| GKD | .439 - | .421 - | .433 - | .411 - |
| OPD | .494 - | .409 - | .416 - | .407 - |
| MiniLLM | .385 - | .340 - | .401 - | .370 - |
| Method | IFEval | MATH | HumanEval | MMLU-Pro | BBH | GPQA | Avg. |
|---|---|---|---|---|---|---|---|
| Llama-3.1-8B-Instruct , trained on QASPER | |||||||
| no training | .827 | .204 | .689 | .458 | .668 | .255 | .517 |
| SFT | .745 | .133 | .580 | .393 | .555 | .261 | .444 |
| PD | .560 | .129 | .559 | .360 | .539 | .254 | .400 |
| ToDi | .610 | .141 | .577 | .389 | .540 | .255 | .419 |
| GKD | .601 | .138 | .573 | .365 | .536 | .250 | .411 |
| Llama | Qwen | |||
| Method | QASPER | BioASQ | QASPER | BioASQ |
| no training | .240 | .392 | .204 | .500 |
| base document | .494 | .487 | .469 | .588 |
| SFT | .471 | .500 | .382 | .625 |
| PD | .384 | .500 | .375 | .633 |
| ToDi | .376 | .417 - | .375 | .700 + |
Language Model Distillation
Language model distillation transfers a larger model’s capabilities to smaller or specialized models using outputs, probabilities, or internal representations.
See all 551 papersDoes On-Policy Distillation Really Distill? From Noisy Teacher to Self-Improvement
Qwen3-1.7B, OPSA improves Avg@32 by 35.41 points on AIME24, corresponding to a 263% relative gain, and more than doubles Pass@32 across all three benchmarks. It also outperforms OPD by 16.77 points in Avg@32 on AIME24. Extensive experiments and analyses across model families and tasks further demonstrate its effectiveness and generalizability.