References & Citations
Computer Science > Machine Learning
Title: Thoughts on the Consistency between Ricci Flow and Neural Network Behavior
(Submitted on 16 Nov 2021 (v1), last revised 16 Feb 2022 (this version, v3))
Abstract: The Ricci flow is a partial differential equation for evolving the metric in a Riemannian manifold to make it more regular. On the other hand, neural networks seem to have similar geometric behavior for specific tasks. In this paper, we construct the linearly nearly Euclidean manifold as a background to observe the evolution of Ricci flow and the training of neural networks. Under the Ricci-DeTurck flow, we prove the dynamical stability and convergence of the linearly nearly Euclidean metric for an $L^2$-Norm perturbation. In practice, from the information geometry and mirror descent points of view, we give the steepest descent gradient flow for neural networks on the linearly nearly Euclidean manifold. During the training process of the neural network, we observe that its metric will also regularly converge to the linearly nearly Euclidean metric, which is consistent with the convergent behavior of linearly nearly Euclidean metrics under the Ricci-DeTurck flow.
Submission history
From: Jun Chen [view email][v1] Tue, 16 Nov 2021 12:23:09 GMT (477kb,D)
[v2] Tue, 8 Feb 2022 03:46:55 GMT (470kb,D)
[v3] Wed, 16 Feb 2022 09:27:19 GMT (279kb,D)
Link back to: arXiv, form interface, contact.