LGAICVDec 4, 2023

PLUM: Improving Inference Efficiency By Leveraging Repetition-Sparsity Trade-Off

Georgia TechMicrosoft
arXiv:2312.01581v2h-index: 6Trans. Mach. Learn. Res.
AI Analysis

This work addresses efficient model deployment in resource-limited environments, presenting an incremental improvement over existing quantization and sparsity techniques.

The paper tackles the problem of improving inference efficiency for deep neural networks on edge devices by introducing a unified co-design framework called PLUM that leverages a repetition-sparsity trade-off. It achieves a 26% speedup, doubles energy efficiency, reduces density by 2.8x, and retains top-1 accuracy of 66.2% on ImageNet compared to prior binary methods.

Efficient inference of Deep Neural Networks (DNNs) on resource-constrained edge devices is essential. Quantization and sparsity are key techniques that translate to repetition and sparsity within tensors at the hardware-software interface. This paper introduces the concept of repetition-sparsity trade-off that helps explain computational efficiency during inference. We propose PLUM, a unified co-design framework that integrates DNN inference systems and quantization (forward and backward pass) to leverage the repetition-sparsity trade-off to improve inference efficiency. Our results demonstrate that PLUM's quantization method is more accurate than binary quantization with the same number of non-zero weights. Detailed analysis indicates that signed binarization generates a smaller distribution of effectual (non-zero) parameters nested within a larger distribution of total parameters of latent full-precision weights for a DNN block. Finally, the proposed PLUM framework achieves a 26% speedup on real hardware, doubles energy efficiency, and reduces density by 2.8x compared to binary methods while retaining top-1 accuracy when compared to prior-art methods for ResNets on ImageNet (by achieving 66.2% top-1 accuracy), presenting an alternative solution for deploying efficient models in resource-limited environments.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes