FP=xINT:Representing Neural Networks via Low-Bit Series Basis Functions

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Zhang, Boyang, Cheng, Daning, Zhang, Yunquan, Tian, Jiake, Li, Jing, Liu, Fangming
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866915691869765632
author Zhang, Boyang
Cheng, Daning
Zhang, Yunquan
Tian, Jiake
Li, Jing
Liu, Fangming
author_facet Zhang, Boyang
Cheng, Daning
Zhang, Yunquan
Tian, Jiake
Li, Jing
Liu, Fangming
contents Post-Training Quantization (PTQ) converts pre-trained Full-Precision (FP) models into quantized versions without training. While existing methods reduce size and computational costs, they also significantly degrade performance and quantization efficiency at extremely low settings due to quantization noise. We introduce a deep model series expansion framework to address this issue, enabling rapid and accurate approximation of unquantized models without calibration sets or fine-tuning. This is the first use of series expansion for neural network quantization. Specifically, our method expands the FP model into multiple low-bit basis models. To ensure accurate quantization, we develop low-bit basis model expansions at different granularities (tensor, layer, model), and theoretically confirm their convergence to the dense model, thus restoring FP model accuracy. Additionally, we design AbelianAdd/Mul operations between isomorphic models in the low-bit expansion, forming an Abelian group to ensure operation parallelism and commutativity. The experiments show that our algorithm achieves state-of-the-art performance in low-bit settings; for example, 4-bit quantization of ResNet-50 surpasses the original accuracy, reaching 77.03%. The code will be made public.
format Preprint
id arxiv_https___arxiv_org_abs_2412_06865
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FP=xINT:Representing Neural Networks via Low-Bit Series Basis Functions
Zhang, Boyang
Cheng, Daning
Zhang, Yunquan
Tian, Jiake
Li, Jing
Liu, Fangming
Machine Learning
Artificial Intelligence
Post-Training Quantization (PTQ) converts pre-trained Full-Precision (FP) models into quantized versions without training. While existing methods reduce size and computational costs, they also significantly degrade performance and quantization efficiency at extremely low settings due to quantization noise. We introduce a deep model series expansion framework to address this issue, enabling rapid and accurate approximation of unquantized models without calibration sets or fine-tuning. This is the first use of series expansion for neural network quantization. Specifically, our method expands the FP model into multiple low-bit basis models. To ensure accurate quantization, we develop low-bit basis model expansions at different granularities (tensor, layer, model), and theoretically confirm their convergence to the dense model, thus restoring FP model accuracy. Additionally, we design AbelianAdd/Mul operations between isomorphic models in the low-bit expansion, forming an Abelian group to ensure operation parallelism and commutativity. The experiments show that our algorithm achieves state-of-the-art performance in low-bit settings; for example, 4-bit quantization of ResNet-50 surpasses the original accuracy, reaching 77.03%. The code will be made public.
title FP=xINT:Representing Neural Networks via Low-Bit Series Basis Functions
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2412.06865