Skip to content

Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"

Notifications You must be signed in to change notification settings

Iolo-Jones/ManifoldDiffusionGeometry

Repository files navigation

ManifoldDiffusionGeometry

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:

  1. intro_notebook.ipynb demonstrates all the main methods
  2. ManifoldDiffusionGeometry.py defines the main functions

The figures, tests, and auxiliary functions from the paper are in

  1. tangents_figs_tests.ipynb
  2. dimension_figs_tests.ipynb
  3. curvature_figs_tests.ipynb
  4. Visualise.py
  5. paper_utils.py

The other methods we compare with in the paper are in the 'Other_methods' folder.

About

Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published