Robust and resource-efficient identification of two hidden layer neural networks

From MaRDI portal
Publication:2117339

DOI10.1007/S00365-021-09550-5zbMATH Open1504.65042arXiv1907.00485OpenAlexW3176108623MaRDI QIDQ2117339FDOQ2117339


Authors: Massimo Fornasier, Timo Klock, Michael Rauchensteiner Edit this on Wikidata


Publication date: 21 March 2022

Published in: Constructive Approximation (Search for Journal in Brave)

Abstract: We address the structure identification and the uniform approximation of two fully nonlinear layer neural networks of the type f(x)=1Th(BTg(ATx)) on mathbbRd from a small number of query samples. We approach the problem by sampling actively finite difference approximations to Hessians of the network. Gathering several approximate Hessians allows reliably to approximate the matrix subspace mathcalW spanned by symmetric tensors a1otimesa1,dots,am0otimesam0 formed by weights of the first layer together with the entangled symmetric tensors v1otimesv1,dots,vm1otimesvm1, formed by suitable combinations of the weights of the first and second layer as vell=AG0bell/|AG0bell|2, ellin[m1], for a diagonal matrix G0 depending on the activation functions of the first layer. The identification of the 1-rank symmetric tensors within mathcalW is then performed by the solution of a robust nonlinear program. We provide guarantees of stable recovery under a posteriori verifiable conditions. We further address the correct attribution of approximate weights to the first or second layer. By using a suitably adapted gradient descent iteration, it is possible then to estimate, up to intrinsic symmetries, the shifts of the activations functions of the first layer and compute exactly the matrix G0. Our method of identification of the weights of the network is fully constructive, with quantifiable sample complexity, and therefore contributes to dwindle the black-box nature of the network training phase. We corroborate our theoretical results by extensive numerical experiments.


Full work available at URL: https://arxiv.org/abs/1907.00485




Recommendations




Cites Work


Cited In (6)

Uses Software





This page was built for publication: Robust and resource-efficient identification of two hidden layer neural networks

Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q2117339)