Quantization without Tears
About
Deep neural networks, while achieving remarkable success across diverse tasks, demand significant resources, including computation, GPU memory, bandwidth, storage, and energy. Network quantization, as a standard compression and acceleration technique, reduces storage costs and enables potential inference acceleration by discretizing network weights and activations into a finite set of integer values. However, current quantization methods are often complex and sensitive, requiring extensive task-specific hyperparameters, where even a single misconfiguration can impair model performance, limiting generality across different models and tasks. In this paper, we propose Quantization without Tears (QwT), a method that simultaneously achieves quantization speed, accuracy, simplicity, and generality. The key insight of QwT is to incorporate a lightweight additional structure into the quantized network to mitigate information loss during quantization. This structure consists solely of a small set of linear layers, keeping the method simple and efficient. More importantly, it provides a closed-form solution, allowing us to improve accuracy effortlessly under 2 minutes. Extensive experiments across various vision, language, and multimodal tasks demonstrate that QwT is both highly effective and versatile. In fact, our approach offers a robust solution for network quantization that combines simplicity, accuracy, and adaptability, which provides new insights for the design of novel quantization paradigms. The code is publicly available at https://github.com/wujx2001/QwT
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language Modeling | WikiText2 | Perplexity6.63 | 4085 | |
| Language Modeling | C4 | Perplexity9.38 | 1565 | |
| Image Classification | ImageNet-1k (val) | Top-1 Accuracy83.8 | 1498 | |
| Image Classification | ImageNet (val) | Top-1 Acc84 | 1206 | |
| Image Generation | ImageNet 256x256 (val) | FID5.35 | 461 | |
| Instance Segmentation | COCO | APmask45 | 301 | |
| Image Classification | CIFAR-100 LT | Top-1 Acc49.06 | 225 | |
| Object Detection | COCO | AP (Box)51.8 | 186 | |
| Image Classification | CIFAR-10-LT gamma=50 (test) | Accuracy71.97 | 68 | |
| Image Classification | Digits | Accuracy (MNIST-M Source)93.75 | 41 |