Statistics > Machine Learning
arXiv:2008.03920 (stat)
[Submitted on 10 Aug 2020 (v1), last revised 27 Oct 2022 (this version, v3)]
Abstract:We introduce a GP generalization of ResNets (including ResNets as a particular case). We show that ResNets (and their GP generalization) converge, in the infinite depth limit, to a generalization of image registration variational algorithms. Whereas computational anatomy aligns images via warping of the material space, this generalization aligns ideas (or abstract shapes as in Plato's theory of forms) via the warping of the RKHS of functions mapping the input space to the output space. While the Hamiltonian interpretation of ResNets is not new, it was based on an Ansatz. We do not rely on this Ansatz and present the first rigorous proof of convergence of ResNets with trained weights and biases towards a Hamiltonian dynamics driven flow. Our constructive proof reveals several remarkable properties of ResNets and their GP generalization. ResNets regressors are kernel regressors with data-dependent warping kernels. Minimizers of $L_2$ regularized ResNets satisfy a discrete least action principle implying the near preservation of the norm of weights and biases across layers. The trained weights of ResNets with $L^2$ regularization can be identified by solving an autonomous Hamiltonian system. The trained ResNet parameters are unique up to the initial momentum whose representation is generally sparse. The kernel regularization strategy provides a provably robust alternative to Dropout for ANNs. We introduce a functional generalization of GPs leading to error estimates for ResNets. We identify the (EPDiff) mean fields limit of trained ResNet parameters. We show that the composition of warping regression blocks with reduced equivariant multichannel kernels (introduced here) recovers and generalizes CNNs to arbitrary spaces and groups of transformations.
| Comments: | 65 pages. To appear in Physica D (special issue on Machine Learning and Dynamical Systems) |
| Subjects: | Machine Learning (stat.ML); Artificial Intelligence (cs.AI); Machine Learning (cs.LG) |
| MSC classes: | 62J02, 68T01, 62M45 |
| Cite as: | arXiv:2008.03920 [stat.ML] |
| (or arXiv:2008.03920v3 [stat.ML] for this version) | |
| https://doi.org/10.48550/arXiv.2008.03920 arXiv-issued DOI via DataCite |
Submission history
From: Houman Owhadi [view email]
[v1]
Mon, 10 Aug 2020 06:46:43 UTC (1,912 KB)
[v2]
Mon, 2 Nov 2020 15:55:24 UTC (1,885 KB)
[v3]
Thu, 27 Oct 2022 18:01:18 UTC (1,626 KB)
Bibliographic and Citation Tools
Code, Data and Media Associated with this Article
Demos
arXivLabs: experimental projects with community collaborators
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.
Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.