Norm-based Generalization Bounds for Sparse Neural Networks

Tomer Galanti, Mengjia Xu, Liane Galanti, Tomaso Poggio

Keywords: generalization bounds, convolution, rademacher, generalization, sparsity

Abstract: 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.

Submission Number: 8309