arXiv · 2405.13088
Combining Relevance and Magnitude for Resource-Aware DNN Pruning
Abstract
Pruning neural networks, i.e., removing some of their parameters whilst retaining their accuracy, is one of the main ways to reduce the latency of a machine learning pipeline, especially in resource- and/or bandwidth-constrained scenarios. In this context, the pruning technique, i.e., how to choose the parameters to remove, is critical to the system performance. In this paper, we propose a novel pruning approach, called FlexRel and predicated upon combining training-time and inference-time information, namely, parameter magnitude and relevance, in order to improve the resulting accuracy whilst saving both computational resources and bandwidth. Our performance evaluation shows that FlexRel is able to achieve higher pruning factors, saving over 35% bandwidth for typical accuracy targets.
Explore related subjects
Keep this discovery
Carla Fabiana Chiasserini, Francesco Malandrino, Nuria Molner, Zhiqiang Zhao. 2024-05-21. Combining Relevance and Magnitude for Resource-Aware DNN Pruning. https://arxiv.org/abs/2405.13088
Cite the original work for its findings. Save a collection to share your selection of sources.