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

Published: 22 Oct 2021, Last Modified: 05 May 2023NeurIPS 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