Prompt-Level Distillation: A Non-Parametric Alternative to Model Fine-Tuning for Efficient Reasoning
Organizations: Google Mountain View, California, USA
Abstract
Advanced reasoning typically requires Chain-of-Thought prompting, which is accurate but incurs prohibitive latency and substantial test-time inference costs. The standard alternative, fine-tuning smaller models, often sacrifices interpretability while introducing significant resource and operational overhead. To address these limitations, we introduce Prompt-Level Distillation (PLD). We extract explicit reasoning patterns from a Teacher model and organize them into a structured list of expressive instructions for the Student model's System Prompt. Evaluated using Gemma-3 4B, PLD improved Macro F1 scores on StereoSet (57% to 90.0%) and Contract-NLI (67% to 83%), while increasing LogiQA accuracy to 70%. Similar results on Mistral Small 3.1 demonstrate cross-architecture generalizability, enabling these compact models to match frontier performance with negligible latency overhead. These expressive instructions render the decision-making process transparent, allowing for full human verification of logic, making this approach ideal for regulated industries such as law, finance, and content moderation, as well as high-volume use cases and edge devices.
Figures & tables
| Method | Gemma | Gemini 2 Flash | Gemini 3 Flash | Mistral Small 3.1 |
|---|---|---|---|---|
| Stereoset (Macro-F1) | ||||
| Zero-shot | 0.57 | 0.53 | 0.92 | 0.65 |
| Few-shot | 0.75 | 0.83 | 0.90 | 0.77 |
| Fine-tuning | 0.89 | - | - | 0.96 |
| TextGrad | 0.87 | 0.90 | 0.91 | 0.96 |
| Clustered-Inst. (Our) | 0.90 | 0.92 | 0.93 | 0.96 |
| Dataset | Train | Validation | Test | Total |
|---|---|---|---|---|
| Contract NLI | 7,191 | 1,037 | 2,091 | 10,319 |
| Stereoset | 842 | 211 | 1,053 | 2,106 |
| LogiQA | 7376 | 651 | 651 | 8,678 |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Total Clusters | Prompt Length (Tokens) | F1 Macro (%) | ||
| 0.2 | 6 | 27 | 6,449 | 79 |
| 0.3 | 6 | 22 | 5,580 | 82 |
| 0.4 | 6 | 17 | 4,630 | 83 |
| 0.5 | 6 | 14 | 4,068 | 80 |
| 0.4 | 3 | 24 | 5,914 | 81 |
| 0.4 | 9 | 15 | 4,432 | 82 |
| Index | Dataset Size | Total Clusters | Prompt Length (Tokens) | F1 Macro |
|---|---|---|---|---|
| 1 | 1,030 | 16 | 4,062 | 0.77 |
| 2 | 2,060 | 18 | 4,486 | 0.79 |
| 3 | 3,090 | 18 | 4,410 | 0.80 |
| 4 | 4,119 | 18 | 4,580 | 0.80 |
| 5 | 5,148 | 18 | 4,520 | 0.82 |
| 6 | 6,178 | 18 | 4,665 | 0.83 |
Efficient Language Model Reasoning
Efficient language model reasoning studies how to improve problem-solving accuracy while reducing inference, memory, latency, and training costs.
See all 370 papers