Proceedings

Where and How to Enhance: Discovering Bit-Width Contribution for Mixed Precision Quantization

Download BibTeX Cite

Proceedings of the Thirty-Fourth International Joint Conference on Artificial Intelligence. Main Track. Pages 5517-5526.

https://doi.org/10.24963/ijcai.2025/614

Abstract

Mixed precision quantization (MPQ) is an effective quantization approach to achieve accuracy-complexity trade-off of neural network, through assigning different bit-widths to network activations and weights in each layer. The typical way of existing MPQ methods is to optimize quantization policies (i.e., bit-width allocation) in a gradient descent manner, termed as Differentiable MPQ (DMPQ). At the end of the search, the bit-width associated to the quantization parameters which has the largest value will be selected to form the final mixed precision quantization policy, with the implicit assumption that the values of quantization parameters reflect the operation contribution to the accuracy improvement. While much has been discussed about the MPQ’s improvement, the bit-width selection process has received little attention. We study this problem and argue that the magnitude of quantization parameters does not necessarily reflect the actual contribution of the bit-width to the task performance. Then, we propose a Shapley-based MPQ (SMPQ) method, which measures the bit-width operation’s direct contribution on the MPQ task. To reduce computation cost, a Monte Carlo sampling-based approximation strategy is proposed for Shapley computation. Extensive experiments on mainstream benchmarks demonstrate that our SMPQ consistently achieves state-of-the-art performance than gradient-based competitors.

  1. [Abdolrashidi et al., 2021] AmirAli Abdolrashidi, Lisa Wang, Shivani Agrawal, Jonathan Malmaud, Oleg Rybakov, Chas Leichner, and Lukasz Lew. Pareto-optimal quantized resnet is mostly 4-bit. ArXiv, abs/2105.03536, 2021.
  2. [Aggarwal et al., 2024] Shivam Aggarwal, Kuluhan Binici, and Tulika Mitra. Chameleon: Dual memory replay for online continual learning on edge devices. IEEE Transac- tions on Computer-Aided Design of Integrated Circuits and Systems, 43(6):1663-1676, 2024.
  3. [Ancona et al., 2019] Marco Ancona, Cengiz Oztireli, and Markus Gross. Explaining deep neural networks with a polynomial time algorithm for shapley value approximation. In International Conference on Machine Learning, pages 272-281. PMLR, 2019.
  4. [Arora et al., 2018] Sanjeev Arora, Nadav Cohen, Noah Golowich, and Wei Hu. A convergence analysis of gra- dient descent for deep linear neural networks. ArXiv, abs/1810.02281, 2018.
  5. [Bai et al., 2024] Jinyu Bai, Sifan Sun, Weisheng Zhao, and Wang Kang. Cimq: A hardware-efficient quantization framework for computing-in-memory-based neural network accelerators. IEEE Transactions on Computer-Aided De- sign of Integrated Circuits and Systems, 43(1):189-202, 2024.
  6. [Banner et al., 2019] Ron Banner, Yury Nahshan, and Daniel Soudry. Post training 4-bit quantization of convolutional networks for rapid-deployment. Conference on Neural Information Processing Systems, 32, 2019.
  7. [Bhalgat et al., 2020] Yash Bhalgat, Jinwon Lee, Markus Nagel, Tijmen Blankevoort, and Nojun Kwak. Lsq+: Im- proving low-bit quantization through learnable offsets and better initialization. In The IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 696-697, 2020.
  8. [Cai and Vasconcelos, 2020] Zhaowei Cai and Nuno Vascon- celos. Rethinking differentiable search for mixed-precision neural networks. In The IEEE/CVF Conference on Com- puter Vision and Pattern Recognition, pages 2349-2358, 2020.
  9. [Cai et al., 2017] Zhaowei Cai, Xiaodong He, Jian Sun, and Nuno Vasconcelos. Deep learning with low precision by half-wave gaussian quantization. In The IEEE/CVF Confer- ence on Computer Vision and Pattern Recognition, pages 5918-5926, 2017.
  10. [Castro et al., 2009] Javier Castro, Daniel Gómez, and Juan Tejada. Polynomial calculation of the shapley value based on sampling. COMPUT OPER RES, 36(5):1726-1730, 2009.
  11. [Chakraborty et al., 2020] Indranil Chakraborty, Deboleena Roy, Isha Garg, Aayush Ankit, and Kaushik Roy. Con- structing energy-efficient mixed-precision neural networks through principal component analysis for edge intelligence. Nature Machine Intelligence, 2(1):43-55, 2020.
  12. [Chang et al., 2021] Sung-En Chang, Yanyu Li, Mengshu Sun, Weiwen Jiang, Sijia Liu, Yanzhi Wang, and Xue Lin. Rmsmp: A novel deep neural network quantization frame- work with row-wise mixed schemes and multiple precisions. In International Conference on Computer Vision, pages 5251-5260, 2021.
  13. [Choi et al., 2018] Jungwook Choi, Zhuo Wang, Swagath Venkataramani, Pierce I-Jen Chuang, Vijayalakshmi Srini- vasan, and Kailash Gopalakrishnan. Pact: Parameterized clipping activation for quantized neural networks. ArXiv, abs/1805.06085, 2018.
  14. [Chu et al., 2019] Tianshu Chu, Qin Luo, Jie Yang, and Xi- aolin Huang. Mixed-precision quantized neural network with progressively decreasing bitwidth for image classifica- tion and object detection. ArXiv, abs/1912.12656, 2019.
  15. [Chu et al., 2021] Tianshu Chu, Qin Luo, Jie Yang, and Xi- aolin Huang. Mixed-precision quantized neural networks with progressively decreasing bitwidth. Pattern Recogni- tion, 111:107647, 2021.
  16. [Dong et al., 2019] Zhen Dong, Zhewei Yao, Amir Gholami, Michael W Mahoney, and Kurt Keutzer. Hawq: Hes- sian aware quantization of neural networks with mixed- precision. In International Conference on Computer Vision, pages 293-302, 2019.
  17. [Dong et al., 2023] Peijie Dong, Lujun Li, Zimian Wei, Xin Niu, Zhiliang Tian, and Hengyue Pan. Emq: Evolving training-free proxies for automated mixed precision quanti- zation. In The IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 17076-17086, 2023.
  18. [Habi et al., 2020] Hai Victor Habi, Roy H Jennings, and Arnon Netzer. Hmq: Hardware friendly mixed precision quantization block for cnns. In European Conference on Computer Vision, pages 448-463. Springer, 2020.
  19. [Huang et al., 2022] Xijie Huang, Zhiqiang Shen, Shichao Li, Zechun Liu, Xianghong Hu, Jeffry Wicaksana, Eric Xing, and Kwang-Ting Cheng. Sdq: Stochastic differentiable quantization with mixed precision. ArXiv, abs/2206.04459, 2022.
  20. [Jacob et al., 2018] Benoit Jacob, Skirmantas Kligys, Bo Chen, Menglong Zhu, Matthew Tang, Andrew Howard, Hartwig Adam, and Dmitry Kalenichenko. Quantization and training of neural networks for efficient integer-arithmetic-only inference. In The IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 2704-2713, 2018.
  21. [Kim et al., 2023] Han-Byul Kim, JooHyung Lee, Sungjoo Yoo, and Hong-Seok Kim. Metamix: Meta-state precision searcher for mixed-precision activation quantization. In Association for the Advancement of Artificial Intelligence, Vancouver, Canada, Nov. 2023.
  22. [Koryakovskiy et al., 2023] Ivan Koryakovskiy, Alexandra Yakovleva, and Valentin Buchnev. One-shot model for mixed-precision quantization. In The IEEE/CVF Confer- ence on Computer Vision and Pattern Recognition, Vancou- ver, Canada, jun. 2023.
  23. [Kwon et al., 2024] Eunji Kwon, Jongho Yoon, and Seokhyeong Kang. Mobile transformer accelerator exploiting various line sparsity and tile-based dynamic quantization. IEEE Transactions on Computer-Aided De- sign of Integrated Circuits and Systems, 43(6):1808-1821, 2024.
  24. [Liu et al., 2024] Lian Liu, Ying Wang, Xiandong Zhao, Wei- wei Chen, Huawei Li, Xiaowei Li, and Yinhe Han. An au- tomatic neural network architecture-and-quantization joint optimization framework for efficient model inference. IEEE Transactions on Computer-Aided Design of Integrated Cir- cuits and Systems, 43(5):1497-1510, 2024.
  25. [Lu et al., 2023] Zhichao Lu, Chuntao Ding, Felix Juefei-Xu, Vishnu Naresh Boddeti, Shangguang Wang, and Yun Yang. Tformer: A transmission-friendly vit model for iot devices. IEEE Transactions on Parallel and Distributed Systems, 34(2):598-610, 2023.
  26. [Ma et al., 2023] Yuexiao Ma, Taisong Jin, Xiawu Zheng, Yan Wang, Huixia Li, Yongjian Wu, Guannan Jiang, Wei Zhang, and Rongrong Ji. Ompq: Orthogonal mixed preci- sion quantization. In Association for the Advancement of Artificial Intelligence, volume 37, pages 9029-9037, 2023.
  27. [Sun et al., 2022] Zhenhong Sun, Ce Ge, Junyan Wang, Ming Lin, Hesen Chen, Hao Li, and Xiuyu Sun. Entropy- driven mixed-precision quantization for deep network de- sign. Conference on Neural Information Processing Sys- tems, 35:21508-21520, 2022.
  28. [Sun et al., 2024a] Qiheng Sun, Jiayao Zhang, Jinfei Liu, Li Xiong, Jian Pei, and Kui Ren. Shapley Value Approximation Based on Complementary Contribution . IEEE Transactions on Knowledge & Data Engineering, 36(12):9263-9281, 2024.
  29. [Sun et al., 2024b] Sifan Sun, Jinyu Bai, Zhaoyu Shi, Weisheng Zhao, and Wang Kang. Cim²pq: An array- wise and hardware-friendly mixed precision quantization method for analog computing-in-memory. IEEE Transac- tions on Computer-Aided Design of Integrated Circuits and Systems, 43(7):2084-2097, 2024.
  30. [Wang et al., 2019] Kuan Wang, Zhijian Liu, Yujun Lin, Ji Lin, and Song Han. Haq: Hardware-aware automated quantization with mixed precision. In The IEEE/CVF Con- ference on Computer Vision and Pattern Recognition, pages 8612-8620, 2019.
  31. [Wu et al., 2018] Bichen Wu, Yanghan Wang, Peizhao Zhang, Yuandong Tian, Peter Vajda, and Kurt Keutzer. Mixed precision quantization of convnets via differentiable neu- ral architecture search. arXiv preprint arXiv:1812.00090, 2018.
  32. [Yang and Jin, 2021] Linjie Yang and Qing Jin. Fracbits: Mixed precision quantization via fractional bit-widths. In Association for the Advancement of Artificial Intelligence, volume 35, pages 10612-10620, 2021.
  33. [Yao et al., 2021] Zhewei Yao, Zhen Dong, Zhangcheng Zheng, Amir Gholami, Jiali Yu, Eric Tan, Leyuan Wang, Qi- jing Huang, Yida Wang, Michael Mahoney, et al. Hawq-v3: Dyadic neural network quantization. In International Con- ference on Machine Learning, pages 11875-11886. PMLR, 2021.
  34. [Yu et al., 2020] Haibao Yu, Qi Han, Jianbo Li, Jianping Shi, Guangliang Cheng, and Bin Fan. Search what you want: Barrier panelty nas for mixed precision quantization. In European Conference on Computer Vision, pages 1-16. Springer, 2020.
  35. [Zhang et al., 2018] Dongqing Zhang, Jiaolong Yang, Dongqiangzi Ye, and Gang Hua. Lq-nets: Learned quantization for highly accurate and compact deep neural networks. In European Conference on Computer Vision, pages 365-382, 2018.
  36. [Zhang et al., 2021] Zhaoyang Zhang, Wenqi Shao, Jinwei Gu, Xiaogang Wang, and Ping Luo. Differentiable dynamic quantization with mixed precision and adaptive resolution. In International Conference on Machine Learning, pages 12546-12556. PMLR, 2021.
  37. [Zheng et al., 2022] Yang-Lin Zheng, Wei-Yi Yang, Ya-Shu Chen, and Ding-Hung Han. An energy-efficient inference engine for a configurable reram-based neural network ac- celerator. IEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems, 42(3):740-753, 2022.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards