about
Fast as Chita: Neural Network Pruning with Combinatorial Optimization (arxiv.org)
2 points by kpe on Aug 17, 2023 | hide | past | pdf | discuss on HN

In plain words: It picks weights to cut and how to nudge the rest by weighing many together, using a rough picture of how the loss changes nearby. With just 2% of a network's weights kept, it improved accuracy by 63% over the best earlier pruning trick.

Abstract · Fast as CHITA: Neural Network Pruning with Combinatorial Optimization

The sheer size of modern neural networks makes model serving a serious computational challenge. A popular class of compression techniques overcomes this challenge by pruning or sparsifying the weights of pretrained networks. While useful, these techniques often face serious tradeoffs between computational requirements and compression quality. In this work, we propose a novel optimization-based pruning framework that considers the combined effect of pruning (and updating) multiple weights subject to a sparsity constraint. Our approach, CHITA, extends the classical Optimal Brain Surgeon framework and results in significant improvements in speed, memory, and performance over existing optimization-based approaches for network pruning. CHITA's main workhorse performs combinatorial optimization updates on a memory-friendly representation of local quadratic approximation(s) of the loss function. On a standard benchmark of pretrained models and datasets, CHITA leads to significantly better sparsity-accuracy tradeoffs than competing methods. For example, for MLPNet with only 2% of the weights retained, our approach improves the accuracy by 63% relative to the state of the art. Furthermore, when used in conjunction with fine-tuning SGD steps, our method achieves significant accuracy gains over the state-of-the-art approaches.

Riade Benbaki, Wenyu Chen, Xiang Meng, Hussein Hazimeh, Natalia Ponomareva, Zhe Zhao, Rahul Mazumder
arXiv:2302.14623 · cs.LG, cs.CV, math.OC · submitted Feb 28, 2023
abstract · pdf · html

add comment on HN