Technology/Quantization-aware distillation
Technology assessment · Ongoing evaluation

Quantization-aware distillation

A compression technique that teaches a low-precision model to remain faithful to its full-precision parent — how large reasoning models fit the power envelope of the edge without losing themselves.

01 — What it is

Fidelity to a known parent

Quantization reduces the numeric precision of a model’s weights and activations — from 16-bit floating point down to 8-bit or 4-bit — shrinking memory and multiplying throughput on hardware built for low-precision math, at the cost of rounding error that degrades accuracy. Knowledge distillation trains a “student” model to reproduce the full output behaviour of a “teacher” — not just its answers, but its distribution over answers. Quantization-aware distillation fuses the two: the student trains with quantization simulated in the loop, so it learns weights that survive rounding, while its training signal comes from a frozen full-precision teacher — typically the original model itself. In production form it is a two-stage recipe: post-training quantization produces a low-precision starting checkpoint, then distillation against the frozen teacher recovers the accuracy the rounding took away. NVIDIA describes it precisely as a pure accuracy-recovery technique — its goal is fidelity to a known parent, not new behaviour. That framing is why it belongs in a trust-focused stack.

Stage 1 — Post-training quantization
Full-precision parent → low-precision starting checkpoint
Stage 2 — Distillation
Student trains against the frozen parent, quantization simulated in the loop

Source: NVIDIA Model Optimizer documentation (S1) · NVIDIA Technical Blog (S2).

02 — Why distillation wins

Where the simpler methods stop

The alternative approaches are simpler and cheaper, and the page owes them a fair account. Post-training quantization alone covers most needs; NVIDIA’s own guidance says so. Quantization-aware training — fine-tuning the quantized model on task data without a teacher — helps at moderate settings. The differentiator appears where modern reasoning models actually live: multi-stage training pipelines involving supervised fine-tuning, reinforcement learning, and model merging. Replaying those stages with quantized passes is impractical, and NVIDIA’s published results show that at aggressive 4-bit settings, quantization-aware training can break an RL-trained model’s capabilities outright — while a single distillation stage against the frozen parent recovers near-parent accuracy regardless of how complex the original pipeline was.

99.4%

of full-precision (BF16) accuracy recovered by a 4-bit QAD checkpoint across reasoning and coding benchmarks — where plain quantization shows noticeable loss. Source: NVIDIA research (S3).

In one published case, the recipe compressed a 66 GB checkpoint to 22 GB with up to 4× higher throughput while preserving near-baseline accuracy (S2).

Recovery costs what plain quantization does not: training compute, training data, and an engineering loop. Where moderate precision suffices, the simpler method is the right one — and NVIDIA notes the optimal balance between the approaches remains an active area of research (S1).

03 — The format and the silicon

Precision is a per-layer budget

The reason this is an edge story has a name: NVFP4, a 4-bit floating-point format for recent NVIDIA GPUs that delivers two to three times the arithmetic throughput of FP8 and roughly halves memory for weights and activations, using small quantization blocks with two-level scaling to hold dynamic range (S3). Production checkpoints are selective rather than uniform — sensitive layers stay in higher precision while the bulk quantizes to 4-bit — a reminder that precision is a per-layer budget, not a switch.

The silicon sets the incentive. The Blackwell-generation edge platforms Enkidu is evaluating for Baru’s fabric publish their headline compute in FP4: up to 2070 FP4 TFLOPS on Jetson Thor T5000-class modules (S4). Those numbers are reachable only by models quantized to 4-bit — and quantization-aware distillation is currently the method that gets reasoning models there without breaking them. On the integer-era silicon of smaller edge nodes, the same methodology applies to 8-bit and 4-bit integer formats: one technique, a different target format per tier.

2070 FP4 TFLOPS · Jetson Thor T5000-class edge modules (S4)

04 — Where it fits in Baru

A derived model has a lineage

Baru, the trusted intelligence-fusion platform, fuses every source into one trusted picture and leaves the decision to a human. Within Enlil — the Defence implementation of Baru — Enkidu is evaluating quantization-aware distillation and speculative decoding together as the path to efficient reasoning at the edge: distilled low-precision reasoning models at the hub tier, smaller integer-format variants at the node tier, and distilled draft models making speculative decoding cheap while the recovered model keeps verification faithful to its parent.

The technique carries an assurance property Enkidu considers as important as its speed: a distilled quantized model is a derived model — with a named parent, a frozen teacher, a training objective of agreement with that teacher, and a measurable divergence from it. That is a chain-of-custody structure, and Enkidu is prototyping its use as Kittu acceptance evidence: teacher–student agreement measured on Enkidu’s own task suites, with model lineage recorded as provenance. The caveat is stated as plainly as the promise: benchmark parity is not behavioural identity, and evaluation of a compressed model is Enkidu’s own and task-specific.

The division of labour with NVIDIA Halos is one sentence: quantization-aware distillation shapes the model that proposes; Halos governs whether anything it proposes ever becomes action.

Evaluation and prototyping activity — no availability, customer, or performance claims.

Maturity

The recipes assessed here are current-generation and moving: the published results are NVIDIA’s, on NVIDIA models, and are cited as such. Enkidu adopts nothing on benchmark parity alone — compressed models pass Enkidu’s own task-specific evaluation gates or they do not ship. Evidence before confidence.

Primary sources

NVIDIA Model Optimizer, QAT/QAD documentation (S1)

Developing Nemotron 3.5 Lightning NVFP4 with QAD Using NVIDIA Model Optimizer (S2)

Quantization-Aware Distillation for NVFP4 Inference Accuracy Recovery (S3)

NVIDIA Jetson Thor specifications (S4)

Related technology

NVIDIA, Nemotron, TensorRT, Jetson, and Blackwell are trademarks of NVIDIA Corporation. Referenced for identification; no endorsement implied.