Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Efficient Multi-bit Quantization Network Training via Weight Bias Correction and Bit-wise Coreset Sampling

About

Multi-bit quantization networks enable flexible deployment of deep neural networks by supporting multiple precision levels within a single model. However, existing approaches suffer from significant training overhead as full-dataset updates are repeated for each supported bit-width, resulting in a cost that scales linearly with the number of precisions. Additionally, extra fine-tuning stages are often required to support additional or intermediate precision options, further compounding the overall training burden. To address this issue, we propose two techniques that greatly reduce the training overhead without compromising model utility: (i) Weight bias correction enables shared batch normalization and eliminates the need for fine-tuning by neutralizing quantization-induced bias across bit-widths and aligning activation distributions; and (ii) Bit-wise coreset sampling strategy allows each child model to train on a compact, informative subset selected via gradient-based importance scores by exploiting the implicit knowledge transfer phenomenon. Experiments on CIFAR-10/100, TinyImageNet, and ImageNet-1K with both ResNet and ViT architectures demonstrate that our method achieves competitive or superior accuracy while reducing training time up to 7.88x. Our code is released at https://github.com/a2jinhee/EMQNet_jk.

Jinhee Kim, Jae Jun An, Kang Eun Jeon, Jong Hwan Ko• 2025

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 LT
Top-1 Acc57.78
225
Image ClassificationCIFAR-10-LT gamma=50 (test)
Accuracy75.66
68
Image ClassificationDigits
Accuracy (MNIST-M Source)90.17
41
Image ClassificationCIFAR-10-LT gamma=10 (test)
Accuracy85.83
34
Image ClassificationSynDigits-LT gamma=10
Accuracy53.82
32
Image ClassificationCIFAR-10-LT (gamma=200) (test)
Accuracy61.35
31
Image ClassificationSynDigits-LT gamma=50
Accuracy19.61
31
Image ClassificationSynDigits-LT (gamma=200)
Accuracy19.84
28
object recognitionOffice-31
Accuracy (Amazon)17.94
18
Showing 9 of 9 rows

Other info

Follow for update