提出可并行的可微测地线算法,提升网格上机器学习的几何精度。
Parallelised Differentiable Straightest Geodesics for 3D Meshes
- 基于直测地线构造可微指数映射,支持向量平行传输
- 在网格上实现高效并行计算,准确追踪测地线路径
- 适用于几何深度学习、流匹配和优化,适合研究网格建模者
机器学习正逐步拓展至非欧几里得域,但表面学习中的几何精确方法仍滞后。缺乏闭式黎曼算子、离散版本不可微、并行效率低是主要障碍。本文提出直测地线框架,在离散网格上实现黎曼曲面的指数映射,并可同时追踪测地线与向量平行传输。我们提供基于GPU的并行实现,推导出两种可微化方法:一种利用外嵌代理函数,另一种基于测地线有限差分。经验证,该方法兼具高精度与强并行性能。实验表明,其可显著提升一般几何上的学习与优化效果。具体应用包括:新型测地卷积层、面向网格的流匹配方法,以及用于重心沃罗诺伊分割的二阶优化器。代码、模型及可安装库(digeo)已公开于 circle-group.github.io/research/DSG。
原文摘要 · Abstract (English)
Machine learning has been progressively generalised to operate within non-Euclidean domains, but geometrically accurate methods for learning on surfaces are still falling behind. The lack of closed-form Riemannian operators, the non-differentiability of their discrete counterparts, and poor parallelisation capabilities have been the main obstacles to the development of the field on meshes. A principled framework to compute the exponential map on Riemannian surfaces discretised as meshes is straightest geodesics, which also allows to trace geodesics and parallel-transport vectors as a by-product. We provide a parallel GPU implementation and derive two different methods for differentiating through the straightest geodesics, one leveraging an extrinsic proxy function and one based upon a geodesic finite differences scheme. After proving our parallelisation performance and accuracy, we demonstrate how our differentiable exponential map can improve learning and optimisation pipelines on general geometries. In particular, to showcase the versatility of our method, we propose a new geodesic convolutional layer, a new flow matching method for learning on meshes, and a second-order optimiser that we apply to centroidal Voronoi tessellation. Our code, models, and pip-installable library (digeo) are available at: circle-group.github.io/research/DSG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。