Different Teachers, Different Capabilities: Sub-1B On-Device Distillation for Structured Text Enrichment
Summary
A study investigated distilling an 8B reasoning teacher model, deepseek-r1:8b, into a 0.6B student model, Qwen3-0.6B, for on-device structured text enrichment. The process involved mapping news articles to JSON objects containing a short summary and five categorical labels. The student model achieved a processing time of 0.8 seconds per article, significantly faster than the teacher's 39 seconds. It recovered 58% of the base-to-teacher gap in summary quality, surpassing constrained decoding by +16.8 points and few-shot prompting by +4.9 points. This summary gain was attributed to the teacher's reasoning capabilities rather than its scale. The research found that different teachers impart distinct capabilities: the reasoning teacher transferred writing quality, while a managed pipeline transferred label diversity. A same-size instruction teacher's student demonstrated better grounding on short articles, fabricating less (74 versus 55 faithful) compared to the reasoning-lineage student. The findings suggest a per-field routing map is necessary for optimal on-device enrichment.
Key takeaway
For Machine Learning Engineers deploying sub-1B models for on-device structured text enrichment, your teacher model selection is paramount. If your primary goal is high-quality summarization, prioritize distilling from a reasoning-capable teacher. Conversely, for tasks requiring diverse categorical labeling, consider teachers like managed pipelines. You should implement a per-field routing strategy to optimize output quality across different enrichment tasks, acknowledging that no single distilled model will excel universally.
Key insights
Distilling large reasoning models into small on-device models significantly reduces latency while retaining substantial quality, but teacher choice dictates specific capability transfer.
Principles
- Reasoning teachers transfer writing quality effectively.
- Teacher scale alone does not guarantee summary quality gain.
- Different teachers excel at transferring distinct capabilities.
Method
An 8B reasoning teacher (deepseek-r1:8b) was distilled into a 0.6B student (Qwen3-0.6B) using QLoRA across three seeds, evaluated by a three-judge panel against baselines.
In practice
- Use reasoning teachers for summary quality.
- Consider managed pipelines for label diversity.
- Route tasks to specific models based on required output quality.
Topics
- On-device AI
- Model Distillation
- Structured Text Enrichment
- Reasoning Models
- QLoRA
- Large Language Models
Best for: AI Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.