On-Device Distillation: Small Models Learn Differently from Diverse Teachers

Vinay Kumar Chaganti· July 10, 2026 View original

Summary

This research explores distilling an 8B reasoning teacher model into a 0.6B on-device student for structured text enrichment, finding that different teachers impart distinct capabilities. A reasoning teacher improves summary quality, while a managed pipeline enhances label diversity, highlighting the need for a per-field routing map.

Distilling large language models into smaller, on-device models offers significant advantages in terms of latency and cost for high-volume tasks like structured text extraction. This study investigates the effectiveness of such distillation, specifically mapping news articles to JSON objects with summaries and categorical labels. An 8B reasoning teacher model (deepseek-r1:8b) was distilled into a 0.6B student model (Qwen3-0.6B), with additional controls from a same-size non-reasoning teacher and a larger managed pipeline. The results show that the student model, running significantly faster than the teacher, recovered 58% of the summary quality gap, outperforming baselines. Crucially, the nature of the teacher model dictated the capabilities transferred: the reasoning teacher improved writing quality, while the managed pipeline enhanced label diversity. A same-size non-reasoning teacher yielded no improvement over the untuned base model for summary quality, indicating the importance of the teacher's reasoning capabilities. Interestingly, the study also observed that while the reasoning-lineage student excelled in quality, it sometimes fabricated information on short, thin-source articles, whereas a same-size instruction teacher's student remained more grounded. This suggests that no single teacher or distillation strategy is optimal for all tasks, necessitating a per-field routing map for on-device enrichment to leverage the strengths of different teacher lineages.

Why it matters

This research provides critical insights for optimizing on-device AI deployments, demonstrating that strategic teacher selection and multi-teacher distillation can significantly improve performance and efficiency for specific tasks.

How to implement this in your domain

  1. 1Identify specific sub-tasks within your AI workflow that could benefit from on-device model distillation.
  2. 2Experiment with different "teacher" models (e.g., reasoning-focused, instruction-tuned) to find the best fit for each sub-task.
  3. 3Develop a routing mechanism to dynamically select the most appropriate distilled student model for different types of input or desired outputs.
  4. 4Implement rigorous evaluation, including human judgment, to assess the quality and faithfulness of distilled model outputs.

Who benefits

Mobile TechEdge ComputingContent PublishingFinancial ServicesE-commerce

Key takeaways

  • On-device distillation can significantly reduce latency and cost for AI tasks.
  • Different teacher models impart distinct capabilities to student models.
  • Reasoning teachers improve writing quality, while managed pipelines enhance label diversity.
  • A per-field routing map is needed for optimal on-device enrichment.

Original post by Vinay Kumar Chaganti

"arXiv:2607.08268v1 Announce Type: new Abstract: High-volume structured extraction pays a large model's latency on every item, so distilling the task into a small on-device model is attractive: comparable output at a fraction of the time and cost. We measure what that distillation…"

View on X

Originally posted by Vinay Kumar Chaganti on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses