Affinity Workshop: LatinX in AI (LXAI) Research @ NeurIPS 2021

Flexible Learning of Sparse Neural Networks via Constrained $L_0$ Regularization

Jose Gallego-Posada · Juan Ramirez · Akram Erraqabi

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.