Pruning vs Quantization: Which is Better?

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Kuzmin, Andrey, Nagel, Markus, van Baalen, Mart, Behboodi, Arash, Blankevoort, Tijmen
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866929244782723072
author Kuzmin, Andrey
Nagel, Markus
van Baalen, Mart
Behboodi, Arash
Blankevoort, Tijmen
author_facet Kuzmin, Andrey
Nagel, Markus
van Baalen, Mart
Behboodi, Arash
Blankevoort, Tijmen
contents Neural network pruning and quantization techniques are almost as old as neural networks themselves. However, to date only ad-hoc comparisons between the two have been published. In this paper, we set out to answer the question on which is better: neural network quantization or pruning? By answering this question, we hope to inform design decisions made on neural network hardware going forward. We provide an extensive comparison between the two techniques for compressing deep neural networks. First, we give an analytical comparison of expected quantization and pruning error for general data distributions. Then, we provide lower bounds for the per-layer pruning and quantization error in trained networks, and compare these to empirical error after optimization. Finally, we provide an extensive experimental comparison for training 8 large-scale models on 3 tasks. Our results show that in most cases quantization outperforms pruning. Only in some scenarios with very high compression ratio, pruning might be beneficial from an accuracy standpoint.
format Preprint
id arxiv_https___arxiv_org_abs_2307_02973
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Pruning vs Quantization: Which is Better?
Kuzmin, Andrey
Nagel, Markus
van Baalen, Mart
Behboodi, Arash
Blankevoort, Tijmen
Machine Learning
Neural network pruning and quantization techniques are almost as old as neural networks themselves. However, to date only ad-hoc comparisons between the two have been published. In this paper, we set out to answer the question on which is better: neural network quantization or pruning? By answering this question, we hope to inform design decisions made on neural network hardware going forward. We provide an extensive comparison between the two techniques for compressing deep neural networks. First, we give an analytical comparison of expected quantization and pruning error for general data distributions. Then, we provide lower bounds for the per-layer pruning and quantization error in trained networks, and compare these to empirical error after optimization. Finally, we provide an extensive experimental comparison for training 8 large-scale models on 3 tasks. Our results show that in most cases quantization outperforms pruning. Only in some scenarios with very high compression ratio, pruning might be beneficial from an accuracy standpoint.
title Pruning vs Quantization: Which is Better?
topic Machine Learning
url https://arxiv.org/abs/2307.02973