Paper · Dec 21, 2021

Dynamically Stable Poincaré Embeddings for Neural Manifolds

Jun Chen, Yuang Liu, Xiangrui Zhao, Mengmeng Wang, Yong Liu

As on arXiv

Abstract

In a Riemannian manifold, the Ricci flow is a partial differential equation for evolving the metric to become more regular. We hope that topological structures from such metrics may be used to assist in the tasks of machine learning. However, this part of the work is still missing. In this paper, we propose Ricci flow assisted Eucl2Hyp2Eucl neural networks that bridge this gap between the Ricci flow and deep neural networks by mapping neural manifolds from the Euclidean space to the dynamically stable Poincaré ball and then back to the Euclidean space. As a result, we prove that, if initial metrics have an -norm perturbation which deviates from the Hyperbolic metric on the Poincaré ball, the scaled Ricci-DeTurck flow of such metrics smoothly and exponentially converges to the Hyperbolic metric. Specifically, the role of the Ricci flow is to serve as naturally evolving to the stable Poincaré ball. For such dynamically stable neural manifolds under the Ricci flow, the convergence of neural networks embedded with such manifolds is not susceptible to perturbations. And we show that Ricci flow assisted Eucl2Hyp2Eucl neural networks outperform with their all Euclidean counterparts on image classification tasks.

Submitted
Dec 21, 2021
Last revised
Feb 16, 2022 (v2)
Subjects
cs.LG, math-ph

Its neighbourhood

Connected papers

The 8 papers on this site closest to this one, judged by the topics they share, the words and phrases in their titles and abstracts, and their authors. They are not its citations, but a map of the work around it. Papers here go back to 2021, plus the canon.

Connected papers This paper in the centre and the 8 papers closest to it around it, in date order clockwise from the top. The closer a paper sits to the centre, the stronger the connection. This paper A physics-informed search for metric solutions to Ricci flow, their embeddings, and visualisation (Aarjav Jain, Challenger Mishra and Pietro Liò, 2022)7Jain 2022 Convergence stability for Ricci flow on manifolds with bounded geometry (Eric Bahuaud, Christine Guenther, James Isenberg et al., 2023)5Bahuaud 2023 Metric Flows with Neural Networks (James Halverson and Fabian Ruehle, 2023)3Halverson 2023 Rigor with Machine Learning from Field Theory to the Poincaré Conjecture (Sergei Gukov, James Halverson and Fabian Ruehle, 2024)2Gukov 2024 Deep Learning as Ricci Flow (Anthony Baptista, Alessandro Barp, Tapabrata Chakraborti et al., 2024)4Baptista 2024 Extensions and Applications of Bredon's Trick in Geometric and Topological Contexts (Mauricio Angel, 2025)6Angel 2025 Neural Feature Geometry Evolves as Discrete Ricci Flow (Moritz Hehl, Max von Renesse and Melanie Weber, 2025)1Hehl 2025 Combinatorial Calabi flows on infinitely triangulated surfaces (Yuerong Bian and Xu Xu, 2026)8Bian 2026
Closer to the centre: more closely connected. Clockwise from the top: oldest to newest. Ricci flowWider flowsalso connected to each other
  1. Neural Feature Geometry Evolves as Discrete Ricci Flow

    Moritz Hehl, Max von Renesse and Melanie Weber

    Topics Discrete & graph Ricci · Shared terms “neural networks”, “classification tasks”, “Deep neural”

  2. Rigor with Machine Learning from Field Theory to the Poincaré Conjecture

    Sergei Gukov, James Halverson and Fabian Ruehle

    Topics Discrete & graph Ricci · Shared terms “machine learning”, “neural networks”, “Poincaré”

  3. Metric Flows with Neural Networks

    James Halverson and Fabian Ruehle

    Topics Discrete & graph Ricci · Shared terms “neural networks”, “learning”, “Poincaré”

  4. Deep Learning as Ricci Flow

    Anthony Baptista, Alessandro Barp, Tapabrata Chakraborti et al.

    Topics Discrete & graph Ricci · Shared terms “neural networks”, “learning”, “classification tasks”

  5. Convergence stability for Ricci flow on manifolds with bounded geometry

    Eric Bahuaud, Christine Guenther, James Isenberg et al.

    Topics Harmonic maps & DeTurck trick · Shared terms “hyperbolic metric”, “initial metrics”

  6. Extensions and Applications of Bredon's Trick in Geometric and Topological Contexts

    Mauricio Angel

    Topics Discrete & graph Ricci · Shared terms “neural networks”

  7. A physics-informed search for metric solutions to Ricci flow, their embeddings, and visualisation

    Aarjav Jain, Challenger Mishra and Pietro Liò

    Topics Discrete & graph Ricci · Shared terms “neural networks”, “embedding”

  8. Combinatorial Calabi flows on infinitely triangulated surfaces

    Yuerong Bian and Xu XuWider flows

    Topics Harmonic maps & DeTurck trick, Discrete & graph Ricci · Shared terms “Euclidean”

Read around it

Background reading

The guides for the topics of this paper: what each idea is, why it matters, and the papers to read first.

In the guidebook

  1. 6D.11
    Computing Geometric FlowsBook 6D · Diffusions, Finance and Computation

    Numerical curve shortening and mean curvature flow, level sets and phase fields, discrete Ricci flow on surfaces and in conformal mapping, rotationally symmetric neckpinches, and what computation has taught about singularities.

  2. 5A.5
    Uniformization and the Two-Dimensional Ricci FlowBook 5A · Complex Analysis and Conformal Geometry

    Constant-curvature metrics on surfaces, and Ricci flow as a heat equation for the conformal factor.

  3. 10A.12
    Curve Shortening and Minimal DiscsBook 10A · Three-Manifolds and Geometrization

    How loops shrink under curvature, how the discs they bound lose area, and Perelman's extinction estimate.