Different Teachers, Different Capabilities: Sub-1B On-Device Distillation for Structured Text Enrichment

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Internet of Things (IoT) & Connected Devices · Depth: Expert, quick

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

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

Topics

Best for: AI Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.