A convergence result of a continuous model of deep learning via Łojasiewicz--Simon inequality

Noboru Isobe
2024-04-14
Abstract:This study focuses on a Wasserstein-type gradient flow, which represents an optimization process of a continuous model of a Deep Neural Network (DNN). First, we establish the existence of a minimizer for an average loss of the model under $L^2$-regularization. Subsequently, we show the existence of a curve of maximal slope of the loss. Our main result is the convergence of flow to a critical point of the loss as time goes to infinity. An essential aspect of proving this result involves the establishment of the Łojasiewicz--Simon gradient inequality for the loss. We derive this inequality by assuming the analyticity of NNs and loss functions. Our proofs offer a new approach for analyzing the asymptotic behavior of Wasserstein-type gradient flows for nonconvex functionals.
Functional Analysis,Probability,Analysis of PDEs
What problem does this paper attempt to address?
The paper discusses the convergence problem of the gradient flow in continuous models of deep learning. The study establishes a minimization problem of average loss under L2 regularization and proves the existence of a limit point, where the gradient flow converges to the critical point of the loss function as time approaches infinity. The key contribution of the paper lies in the use of the Łojasiewicz-Simon inequality to prove the convergence of the gradient flow for non-convex functions. The authors assume that the neural network (NN) and the loss function are analytic and derive this inequality through three steps to analyze the asymptotic behavior of Wasserstein-type gradient flows. The paper also presents general convergence results for the gradient flow applicable to non-convex functions.