English

ELSA: Partial Weight Freezing for Overhead-Free Sparse Network Deployment

Machine Learning 2023-12-19 v2

Abstract

We present ELSA, a practical solution for creating deep networks that can easily be deployed at different levels of sparsity. The core idea is to embed one or more sparse networks within a single dense network as a proper subset of the weights. At prediction time, any sparse model can be extracted effortlessly simply be zeroing out weights according to a predefined mask. ELSA is simple, powerful and highly flexible. It can use essentially any existing technique for network sparsification and network training. In particular, it does not restrict the loss function, architecture or the optimization technique. Our experiments show that ELSA's advantages of flexible deployment comes with no or just a negligible reduction in prediction quality compared to the standard way of using multiple sparse networks that are trained and stored independently.

Keywords

Cite

@article{arxiv.2312.06872,
  title  = {ELSA: Partial Weight Freezing for Overhead-Free Sparse Network Deployment},
  author = {Paniz Halvachi and Alexandra Peste and Dan Alistarh and Christoph H. Lampert},
  journal= {arXiv preprint arXiv:2312.06872},
  year   = {2023}
}

Comments

updated to reflect PackNet prior work

R2 v1 2026-06-28T13:47:49.281Z