Flexible Learning of Sparse Neural Networks via Constrained $L_0$ RegularizationDownload PDF

Oct 16, 2021 (edited Nov 07, 2021)NeurIPS 2021 Workshop LatinX in AI PosterReaders: Everyone
  • Keywords: sparsity, regularization, constrained optimization
  • TL;DR: Constrained formulations provide greater interpretability and flexibility for learning sparse neural networks
  • Abstract: We propose to approach the problem of learning $L_0$-sparse networks using a constrained formulation of the optimization problem. This is in contrast to commonly used penalized approaches, which combine the regularization terms additively with the (surrogate) empirical risk. Our experiments demonstrate that we can obtain approximate solutions to the constrained optimization problem with comparable performance to state-of-the art methods for $L_0$-sparse training. Finally, we discuss how this constrained approach provides greater (hyper-)parameter interpretability and accountability from a practitioner's point of view.
1 Reply