Binary domain generalization for sparsifying binary neural networks

Riccardo Schiavone, Francesco Galati, Maria A. Zuluaga
6/23/2023
cs.LG

Official code: robustml-eurecom/SBNN (3 stars, last commit 2023-06-22).

Abstract

Binary neural networks (BNNs) are an attractive solution for developing and deploying deep neural network (DNN)-based applications in resource constrained devices. Despite their success, BNNs still suffer from a fixed and limited compression factor that may be explained by the fact that existing pruning methods for full-precision DNNs cannot be directly applied to BNNs. In fact, weight pruning of BNNs leads to performance degradation, which suggests that the standard binarization domain of BNNs is not well adapted for the task. This work proposes a novel more general binary domain that extends the standard binary one that is more robust to pruning techniques, thus guaranteeing improved compression and avoiding severe performance losses. We demonstrate a closed-form solution for quantizing the weights of a full-precision network into the proposed binary domain. Finally, we show the flexibility of our method, which can be combined with other pruning strategies. Experiments over CIFAR-10 and CIFAR-100 demonstrate that the novel approach is able to generate efficient sparse networks with reduced memory usage and run-time latency, while maintaining performance.

View on arXivView PDF

Code Implementations(1)

This is the pytorch implementation of the paper "A binary domain generalization for sparsifying binary neural networks", published in ECML PKDD 2023. Authors: R. Schiavone, F. Galati and M. A. Zuluaga.

30Shell, PythonJun 22, 20233 years ago

Cite this paper

@article{schiavone2023binary,
  title  = {Binary domain generalization for sparsifying binary neural networks},
  author = {Riccardo Schiavone and Francesco Galati and Maria A. Zuluaga},
  year   = {2023},
  eprint = {2306.13515},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2306.13515v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2306.13515.svg)](https://codeofpaper.com/paper/2306.13515)

Discussion