A Geometric Approach of Gradient Descent Algorithms in Neural Networks

  • 2018-11-08 17:45:19
  • Yacine Chitour, Zhenyu Liao, Romain Couillet
  • 3

Abstract

In this article we present a geometric framework to analyze convergence ofgradient descent trajectories in the context of neural networks. In the case oflinear networks of an arbitrary number of hidden layers, we characterizeappropriate quantities which are conserved along the gradient descent system(GDS). We use them to prove boundedness of every trajectory of the GDS, whichimplies convergence to a critical point. We further focus on the local behaviorin the neighborhood of each critical points and perform a study on theassociated basin of attractions so as to measure the "possibility" ofconverging to saddle points and local minima.

 

Introduction (beta)

None

 

Conclusion (beta)

None