Norm-based Generalization Bounds for Sparse Neural Networks

TitleNorm-based Generalization Bounds for Sparse Neural Networks
Publication TypeConference Paper
Year of Publication2023
AuthorsGalanti, T, Xu, M, Galanti, L, Poggio, T
Conference NameNeurIPS 2023
Date Published12/2023
Conference LocationNew Orleans

In this paper, we derive norm-based generalization bounds for sparse ReLU neural networks, including convolutional neural networks. These bounds differ from previous ones because they consider the sparse structure of the neural network architecture and the norms of the convolutional filters, rather than the norms of the (Toeplitz) matrices associated with the convolutional layers. Theoretically, we demonstrate that these bounds are significantly tighter than standard norm-based generalization bounds. Empirically, they offer relatively tight estimations of generalization for various simple classification problems. Collectively, these findings suggest that the sparsity of the underlying target function and the model’s architecture plays a crucial role in the success of deep learning.


Associated Module: 

CBMM Relationship: 

  • CBMM Funded