European Conference on Computer Vision (ECCV), 2026

Learning Riemannian Geometry on 3D Objects

1 Imperial College London    2 Peking University

Figure 1. The method takes raw point samples without requiring manifold connectivity, encodes them into a continuous Geometric Embedding Field, and conditions lightweight decoders for exponential and logarithmic map queries.

Abstract

Computing exponential and logarithmic maps on raw geometries is a fundamental yet challenging task in geometry processing, often hindered by the lack of explicit connectivity and the presence of noise. We present a learning-based framework that computes these maps efficiently by learning a continuous geometric embedding field directly from discrete point sets. The field encapsulates the intrinsic local geometry governing exponential and logarithmic maps. A sparse octree-based CNN encodes the input into a high-dimensional feature volume; for an arbitrary query point, its embedding modulates two specialized triplane-based neural networks that predict the maps in a single forward pass. This formulation avoids a requirement for mesh connectivity and is evaluated under irregular sampling, noise, non-manifold input geometry, non-rigid shapes, and real scans.

Keywords: Geometry Processing · Geometric Learning · Neural Fields

Contributions

  1. 1

    A continuous geometric embedding field learned from discrete point sets for efficient evaluation of exponential and logarithmic maps.

  2. 2

    A geometry-aware conditioning mechanism and geometric regularization terms that adapt a shared model to diverse local geometries without explicit connectivity.

  3. 3

    Evaluation of accuracy, runtime, and robustness on remeshed and raw ShapeNet data, non-uniform and noisy samples, DFAUST, and Kinect-v2 scans.

Geometric embedding and hyper-modulation

The pipeline separates a one-time shape encoding stage from repeated local map queries. A sparse octree U-Net produces a continuous volumetric feature field. At a source point p, the interpolated embedding zp generates source-specific triplanes for ExpNet and LogNet.

Input
Coarse boundary point samples
Backbone
Sparse octree-based 3D U-Net
Local representation
Embedding-conditioned triplanes
Query coordinates
Relative ambient 3D coordinates
Outputs
Expp(v) and Logp(q)
Connectivity
Not required at inference
Figure 2. The local geometric embedding generates separate triplane feature fields for exponential and logarithmic maps. Relative coordinates query these planes, and small MLPs produce the final outputs.

Local coordinate representation

The networks operate on relative coordinates in the ambient 3D coordinate system rather than on an arbitrarily oriented 2D tangent basis. This avoids basis-rotation ambiguity while retaining tangent-aligned inputs for ExpNet.

Training data filtering

Samples associated with non-injective regions are filtered when multiple distinct tangent vectors reach the same surface location. This excludes ambiguous supervision near or beyond the cut locus.

Query complexity

After the embedding field has been computed, each map evaluation consists of triplane interpolation and a small number of matrix multiplications. Queries can be batched and parallelized.

Quantitative and qualitative evaluation

For logarithmic maps, the paper reports mean distance distortion EMDD and mean angular error Eθ. For exponential maps, it reports average Euclidean endpoint error EExp. Lower is better for all three metrics.

Figure 5. Logarithmic-map comparisons on point-cloud inputs. The rows include thin and highly curved structures; meshes are used only for visualization.

Main benchmark

Runtime is average CPU time per query, measured on an Apple M4 Max without GPU parallelism. Best values within each input group and operator are bold.

Input N Vertices Logarithmic map Exponential map
MethodEMDDEθTime (s) ↓ MethodEExpTime (s) ↓
ShapeNet100016.6kHeat0.25016.70.178Tracing0.0360.396
Proj.0.11514.60.001Proj.0.0490.003
Ours0.11011.10.012Ours0.0260.019
RawShapeNet20010.0kHeat0.31421.30.101Tracing0.0500.374
Proj.0.16617.10.001Proj.0.0480.003
Ours0.16915.40.013Ours0.0360.020
Non-uniform20015.0kHeat0.33922.40.051Tracing0.0430.415
Proj.0.15516.30.0005Proj.0.0470.003
Ours0.15214.00.011Ours0.0300.019
Noisy20016.6kHeat0.32621.10.101Tracing0.0600.374
Proj.0.20319.40.001Proj.0.0550.003
Ours0.20217.80.013Ours0.0380.020

Interpretation. The proposed method has the lowest angular error and exponential-map error in all four input conditions. It has the lowest distance distortion in three conditions; on RawShapeNet, projection is slightly lower (0.166 versus 0.169). Projection is the fastest baseline, whereas the learned method is substantially faster than Heat or tracing while retaining higher accuracy on most reported metrics.

DFAUST and Kinect-v2 generalization

The checkpoint trained for the ShapeNet experiments is used without retraining or fine-tuning.

Logarithmic map
MethodDFAUST EMDDDFAUST EθKinect EMDDKinect Eθ
Heat0.2325.470.128.45
Proj.0.1719.040.119.29
Ours0.1716.510.118.37
Exponential map, EExp
MethodDFAUSTKinect-v2
Tracing0.0330.063
Proj.0.0710.064
Ours0.0430.060

Robustness, extensions, and scope

Cut-locus filtering. Non-injectivity creates ambiguous Log supervision; the training pipeline filters such samples.
Other 3D representations. The supplementary material describes sampling SDFs and occupancy fields into point clouds using projection or bisection before applying the same pipeline.

Limitations stated in the paper

  • Training targets are produced by classical numerical methods, so the quality of the supervision bounds the learned model.
  • Finite octree and triplane resolution can smooth fine-scale geometric detail.
  • Large-scale parameterization currently requires a separate stitching algorithm rather than a single globally consistent end-to-end model.
Representative failure case. Thin articulated regions can be difficult when surfaces are close in Euclidean distance but differ in intrinsic geodesic behavior.

Paper and citation

The PDFs are included locally so the page can be deployed as a self-contained GitHub Pages project.

BibTeX
@inproceedings{pang2026riemannian,
  title     = {Learning Riemannian Geometry on 3D Objects},
  author    = {Pang, Bo and Foti, Simone and Birdal, Tolga},
  booktitle = {European Conference on Computer Vision},
  year      = {2026}
}

Acknowledgements

T. Birdal was supported by a UKRI Future Leaders Fellowship (MR/Y018818/1). S. Foti and T. Birdal were supported by the EPSRC Project GNOMON (EP/X011364/1). S. Foti was also supported by the Turing AI Fellowship MAGAL (EP/Z534699/1).

BibTeX copied