Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension" https://arxiv.org/abs/2411.04100
This work builds on the programme of "diffusion geometry" introduced in https://arxiv.org/abs/2405.10858 and applies those methods to the special case of manifolds.
This repository contains the following:
- intro_notebook.ipynb demonstrates all the main methods
- ManifoldDiffusionGeometry.py defines the main functions
The figures, tests, and auxiliary functions from the paper are in
- tangents_figs_tests.ipynb
- dimension_figs_tests.ipynb
- curvature_figs_tests.ipynb
- Visualise.py
- paper_utils.py
The other methods we compare with in the paper are in the 'Other_methods' folder.