International SoC Design Conference
Mixed-Precision Neural Network Quantization with Quantum-Inspired Annealers
Cing-Yuan Lai, Chin-Fu Nien, Lien-Po Yu, Lai Chao-Sung
Abstract
Mixed-precision quantization (MPQ) aims to assign different bit-widths to different neural network layers, balancing accuracy and efficiency. Prior studies have formulated the MPQ search problem as an integer linear programming (ILP) prob-lem, relying on layer-wise importance indicators derived from quantization-aware training. This work proposes an alternative approach that reformulates the MPQ as a Quadratic Uncon-strained Binary Optimization (QUBO) problem, which can be solved effectively using the quantum-inspired annealers. This method leverages learned importance indicators while enabling flexible incorporation of additional constraints via soft penalty terms. Experimental results on the Imagenette dataset demon-strate that our QUBO-based approach achieves quantization policies with comparable performance to the ILP-based method, validating its effectiveness.