Norm-Based Generalization Bounds for Compositionally Sparse Neural Network

In this paper, we investigate the Rademacher complexity of deep sparse neural networks, where each neuron receives a small number of inputs. We prove generalization bounds for multilayered sparse ReLU neural networks, including convolutional neural networks. These bounds differ from previous ones, a...

Full description

Bibliographic Details
Main Authors: Galanti, Tomer, Xu, Mengjia, Galanti, Liane, Poggio, Tomaso
Format: Article
Published: Center for Brains, Minds and Machines (CBMM) 2023
Online Access:https://hdl.handle.net/1721.1/148230
Description
Summary:In this paper, we investigate the Rademacher complexity of deep sparse neural networks, where each neuron receives a small number of inputs. We prove generalization bounds for multilayered sparse ReLU neural networks, including convolutional neural networks. These bounds differ from previous ones, as they consider the norms of the convolutional filters instead of the norms of the associated Toeplitz matrices, independently of weight sharing between neurons. As we show theoretically, these bounds may be orders of magnitude better than standard norm- based generalization bounds and empirically, they are almost non-vacuous in estimating generalization in various simple classification problems. Taken together, these results suggest that compositional sparsity of the underlying target function is critical to the success of deep neural networks.