LLMR: Knowledge Distillation with a Large Language Model-Induced Reward
Organizations: Dept. Computing Science & Alberta Machine Intelligence Institute (Amii), University of Alberta · Canada CIFAR AI Chair, Amii
Abstract
Large language models have become increasingly popular and demonstrated remarkable performance in various natural language processing (NLP) tasks. However, these models are typically computationally expensive and difficult to be deployed in resource-constrained environments. In this paper, we propose LLMR, a novel knowledge distillation (KD) method based on a reward function induced from large language models. We conducted experiments on multiple datasets in the dialogue generation and summarization tasks. Empirical results demonstrate that our LLMR approach consistently outperforms traditional KD methods in different tasks and datasets.
Figures & tables
| DailyDialog | OpenSubtitles | CNN/DailyMail | |||||||
| Model | BLEU2 | BLEU4 | BLEU2 | BLEU4 | ROUGE-1 | ROUGE-2 | ROUGE-L | ||
| 1 | Prompting Teacher | 5.57 | 1.49 | 4.67 | 1.51 | 36.16 | 14.99 | 24.05 | |
| 2 | Prompting Student | 1.35 | 0.31 | 1.21 | 0.25 | 21.23 | 6.73 | 17.88 | |
| 3 | Distilled Students | SeqKD | 6.19 | 1.71 | 3.87 | 1.35 | 35.46 | 14.52 | 23.68 |
| 4 | KL | 5.03 | 1.40 | 3.84 | 1.33 | 34.11 | 14.21 | 22.83 | |
| 5 | RKL | 5.02 | 1.29 | 4.12 | 1.36 | 32.07 | 13.77 | 22.87 | |
| DailyDialog | OpenSubtitles | CNN/DailyMail | ||||
|---|---|---|---|---|---|---|
| Model | Dist1 | Dist2 | Dist1 | Dist2 | Dist1 | Dist2 |
| SeqKD | 4.93 | 27.37 | 4.78 | 23.15 | 3.86 | 33.59 |
| KL | 4.76 | 26.77 | 4.99 | 24.00 | 3.76 | 33.59 |
| RKL | 5.76 | 29.01 | 5.38 | 23.72 | 4.07 | 32.27 |
| JS | 5.84 | 32.25 | 4.44 | 19.21 | 3.83 | 31.47 |
| Our LLMR | 6.02 | 34.83 | 5.82 | 27.21 | 4.20 | 35.38 |