Function-Space Optimality of Neural Architectures with Multivariate Nonlinearities

Abstract

We investigate the function-space optimality (specifically, the Banach-space optimality) of a large class of shallow neural architectures with multivariate nonlinearities/activation functions. To that end, we construct a new family of Banach spaces defined via a regularization operator, the $k$-plane transform, and a sparsity-promoting norm. We prove a representer theorem that states that the solution sets to learning problems posed over these Banach spaces are completely characterized by neural architectures with multivariate nonlinearities. These optimal architectures have skip connections and are tightly connected to orthogonal weight normalization and multi-index models, both of which have received recent interest in the neural network community. Our framework is compatible with a number of classical nonlinearities including the rectified linear unit (ReLU) activation function, the norm activation function, and the radial basis functions found in the theory of thin-plate/polyharmonic splines. We also show that the underlying spaces are special instances of reproducing kernel Banach spaces and variation spaces. Our results shed light on the regularity of functions learned by neural networks trained on data, particularly with multivariate nonlinearities, and provide new theoretical motivation for several architectural choices found in practice.

References in corpus (10)

• The Role of Neural Network Activation Functions
• Splines are Universal Solutions of Linear Inverse Problems with Generalized-TV regularization
• Sharp Bounds on the Approximation Rates, Metric Entropy, and $n$-widths of Shallow Neural Networks
• Sparsity of solutions for variational inverse problems with finite-dimensional data
• What Kinds of Functions do Deep Neural Networks Learn? Insights from Variational Spline Theory
• Near-Minimax Optimal Estimation With Shallow ReLU Neural Networks
• Deep Learning Meets Sparse Regularization: A Signal Processing Perspective
• A unified Fourier slice method to derive ridgelet transform for a variety of depth-2 neural networks
• Distributional Extension and Invertibility of the $k$-Plane Transform and Its Dual
• Weighted variation spaces and approximation by shallow ReLU networks