A simple approach for quantizing neural networks

J Maly, R Saab - Applied and Computational Harmonic Analysis, 2023 - Elsevier
Applied and Computational Harmonic Analysis, 2023Elsevier
In this short note, we propose a new method for quantizing the weights of a fully trained
neural network. A simple deterministic pre-processing step allows us to quantize network
layers via memoryless scalar quantization while preserving the network performance on
given training data. On one hand, the computational complexity of this pre-processing
slightly exceeds that of state-of-the-art algorithms in the literature. On the other hand, our
approach does not require any hyper-parameter tuning and, in contrast to previous methods …
Abstract
In this short note, we propose a new method for quantizing the weights of a fully trained neural network. A simple deterministic pre-processing step allows us to quantize network layers via memoryless scalar quantization while preserving the network performance on given training data. On one hand, the computational complexity of this pre-processing slightly exceeds that of state-of-the-art algorithms in the literature. On the other hand, our approach does not require any hyper-parameter tuning and, in contrast to previous methods, allows a plain analysis. We provide rigorous theoretical guarantees in the case of quantizing single network layers and show that the relative error decays with the number of parameters in the network if the training data behave well, e.g., if it is sampled from suitable random distributions. The developed method also readily allows the quantization of deep networks by consecutive application to single layers.
Elsevier
以上显示的是最相近的搜索结果。 查看全部搜索结果