arXiv · 1906.12183
Neural ODEs as the Deep Limit of ResNets with constant weights
Abstract
In this paper we prove that, in the deep limit, the stochastic gradient descent on a ResNet type deep neural network, where each layer shares the same weight matrix, converges to the stochastic gradient descent for a Neural ODE and that the corresponding value/loss functions converge. Our result gives, in the context of minimization by stochastic gradient descent, a theoretical foundation for considering Neural ODEs as the deep limit of ResNets. Our proof is based on certain decay estimates for associated Fokker-Planck equations.
Explore related subjects
Keep this discovery
Benny Avelin, Kaj Nyström. 2019-06-28. Neural ODEs as the Deep Limit of ResNets with constant weights. https://arxiv.org/abs/1906.12183
Cite the original work for its findings. Save a collection to share your selection of sources.