Expert-aware quantisation: near-Q4 quality at near-Q2 size?

· Source: Martin Alderson · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, medium

Summary

Expert-aware quantization for Mixture-of-Experts (MoE) models proposes a method to achieve near-Q4 quality at near-Q2 model sizes by selectively quantizing experts based on task-specific usage. The research profiled Qwen3.6 35B-A3B on C++ programming tasks, observing that during code generation, the top 32 of 256 experts handled approximately 50% of the routing, indicating concentrated expert activity. By applying higher precision (e.g., Q4) to these "hot" experts and aggressive 2-bit quantization to "cold" experts, a Q4-hot / Q2-cold tiered model achieved perplexity scores of 1.635 (reading C++) and 1.477 (writing C++). This 14GB model, only 1GB larger than a uniform Q2 model (13GB, PPL 2.103), recovered about 90% of the quality gap between uniform Q2 and uniform Q4 (20GB, PPL 1.582), demonstrating significant quality improvement for smaller model sizes.

Key takeaway

For AI Engineers optimizing Mixture-of-Experts models for local or edge deployment, this research demonstrates a viable strategy to achieve near-Q4 quality at near-Q2 model sizes. By profiling expert usage for specific tasks and applying tiered quantization (e.g., Q4 for hot experts, Q2 for cold), you can significantly reduce model memory footprint without a proportional loss in performance. Explore integrating domain-specific expert profiling into your quantization workflows to maximize efficiency for targeted applications.

Key insights

Selectively quantizing MoE experts based on task-specific usage can achieve near-Q4 quality at near-Q2 model sizes.

Principles

Method

Profile MoE models on specific tasks to identify "hot" and "cold" experts. Apply higher precision quantization (e.g., Q4) to hot experts and lower precision (e.g., Q2) to cold experts.

In practice

Topics

Code references

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

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Martin Alderson.