Norm-Based Generalization Bounds for Compositionally Sparse Neural Networks

TitleNorm-Based Generalization Bounds for Compositionally Sparse Neural Networks
Publication TypeCBMM Memos
Year of Publication2023
AuthorsGalanti, T, Xu, M, Galanti, L, Poggio, T
Abstract

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.

DSpace@MIT

https://hdl.handle.net/1721.1/145776

CBMM Memo No:  139

Associated Module: 

Research Area: 

CBMM Relationship: 

  • CBMM Funded