TY - JOUR
T1 - Robust and resource efficient identification of shallow neural networks by fewest samples
AU - Fornasier, Massimo
AU - Vybíral, Jan
AU - Daubechies, Ingrid
N1 - Publisher Copyright:
© 2021 The Author(s). Published by Oxford University Press on behalf of the Institute of Mathematics and its Applications. All rights reserved.
PY - 2021/6/1
Y1 - 2021/6/1
N2 - We address the structure identification and the uniform approximation of sums of ridge functions f(x)=∑ i=1m gi(ai,x) on Rd, representing a general form of a shallow feed-forward neural network, from a small number of query samples. Higher order differentiation, as used in our constructive approximations, of sums of ridge functions or of their compositions, as in deeper neural network, yields a natural connection between neural network weight identification and tensor product decomposition identification. In the case of the shallowest feed-forward neural network, second-order differentiation and tensors of order two (i.e., matrices) suffice as we prove in this paper. We use two sampling schemes to perform approximate differentiation-active sampling, where the sampling points are universal, actively and randomly designed, and passive sampling, where sampling points were preselected at random from a distribution with known density. Based on multiple gathered approximated first-and second-order differentials, our general approximation strategy is developed as a sequence of algorithms to perform individual sub-tasks. We first perform an active subspace search by approximating the span of the weight vectors a_1, a m. Then we use a straightforward substitution, which reduces the dimensionality of the problem from d to m. The core of the construction is then the stable and efficient approximation of weights expressed in terms of rank-1 matrices a_i otimes a_i, realized by formulating their individual identification as a suitable nonlinear program. We prove the successful identification by this program of weight vectors being close to orthonormal and we also show how we can constructively reduce to this case by a whitening procedure, without loss of any generality. We finally discuss the implementation and the performance of the proposed algorithmic pipeline with extensive numerical experiments, which illustrate and confirm the theoretical results.
AB - We address the structure identification and the uniform approximation of sums of ridge functions f(x)=∑ i=1m gi(ai,x) on Rd, representing a general form of a shallow feed-forward neural network, from a small number of query samples. Higher order differentiation, as used in our constructive approximations, of sums of ridge functions or of their compositions, as in deeper neural network, yields a natural connection between neural network weight identification and tensor product decomposition identification. In the case of the shallowest feed-forward neural network, second-order differentiation and tensors of order two (i.e., matrices) suffice as we prove in this paper. We use two sampling schemes to perform approximate differentiation-active sampling, where the sampling points are universal, actively and randomly designed, and passive sampling, where sampling points were preselected at random from a distribution with known density. Based on multiple gathered approximated first-and second-order differentials, our general approximation strategy is developed as a sequence of algorithms to perform individual sub-tasks. We first perform an active subspace search by approximating the span of the weight vectors a_1, a m. Then we use a straightforward substitution, which reduces the dimensionality of the problem from d to m. The core of the construction is then the stable and efficient approximation of weights expressed in terms of rank-1 matrices a_i otimes a_i, realized by formulating their individual identification as a suitable nonlinear program. We prove the successful identification by this program of weight vectors being close to orthonormal and we also show how we can constructively reduce to this case by a whitening procedure, without loss of any generality. We finally discuss the implementation and the performance of the proposed algorithmic pipeline with extensive numerical experiments, which illustrate and confirm the theoretical results.
KW - breaking the curse of dimensionality
KW - nonlinear programming for optimizations in matrix subspaces
KW - randomized algorithms
KW - training shallow neural networks
KW - whitening
UR - http://www.scopus.com/inward/record.url?scp=85108641732&partnerID=8YFLogxK
U2 - 10.1093/imaiai/iaaa036
DO - 10.1093/imaiai/iaaa036
M3 - Article
AN - SCOPUS:85108641732
SN - 2049-8772
VL - 10
SP - 625
EP - 695
JO - Information and Inference
JF - Information and Inference
IS - 2
ER -