Papers
arxiv:2608.20953

Quantization-Aware Healing: A Practical Recipe for Recovering Compressed, 4-Bit LLMs

Published on Aug 21
· Submitted by
Antonio Tiene
on Aug 25
Authors:
,
,
,

Abstract

Quantization-aware healing recovers compressed 4-bit language models faster and more stably than quantization-aware training by distilling directly from the original uncompressed model.

Serving large language models cheaply increasingly means shipping models that are both structurally compressed to a fraction of their parameters and quantized to 4 bits. Together these steps degrade reasoning, mathematics, coding, and long-context behavior enough to require a recovery, or healing, stage before deployment. The default recipe, quantization-aware training (QAT), re-fits the compressed, quantized model to hard labels; in our pipeline it converged slowly and collapsed past its peak. We adopted Quantization-Aware Healing (QAH) instead. Because a structurally compressed model is never independently trained at full precision, its bfloat16 checkpoint is a distillation-recovered approximation of the original; QAH distills the 4-bit student directly from the original, uncompressed model. On a GPT-OSS 120B to 60B to MXFP4 pipeline, the QAH student matches or beats its bfloat16 source on 7 of 9 benchmarks at roughly 4 times less weight memory and half the teacher's parameter count, and is released open-weight as Hypernova-60B. Against a matched QAT baseline it reaches a comparable peak about 7 times faster and stays stable under continued training, without hand-tuned early stopping. We also report deployment lessons, including a large, reproducible quality gap between distributed-training backends. Our aim is a recipe deployable without a multi-week hyper-parameter search.

Community

Paper author Paper submitter

Shrinking a large language model usually means two things: cutting its parameter count, then squeezing what's left down to 4 bits. Both save real money on serving cost. Both also quietly cost you accuracy, so every efficient deployment needs a recovery step afterward, usually called healing.

In our latest paper we show that the standard way of healing, retraining against the model's own already-degraded recovered version, caps how good the result can be. Our method, Quantization-Aware Healing, heals directly from the original uncompressed model instead. Applied to a GPT-OSS 120B model compressed to 60B parameters and quantized to 4 bits, the healed model beats its own full-precision version on 7 of 9 benchmarks. Smaller, cheaper to run, and more accurate, all at once.

QAT collapsing past its peak is the failure mode nobody benchmarks for — everyone reports the best checkpoint, not the one that survives deployment. I've hit this on a 7B after 4-bit compression, so the healing stage is the part I actually care about. But the recipe's usefulness hinges on one thing: does healing need the original training corpus, or will a smaller task-specific set do? That's the difference between a recipe I can run on a Friday and one that means re-engaging the whole data pipeline. If it needs the full corpus, it's a research result; if it works on a few thousand task examples, it's a tool.

·

While having the original corpus may help, in Multiverse Computing most of the time we don’t have access to the data used by the original model providers. We have our internal selection of high quality public data, plus task specific datasets. For us this has been enough to successfully apply QAH to a variety of open source models

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2608.20953
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2608.20953 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2608.20953 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2608.20953 in a Space README.md to link it from this page.

Collections including this paper 1