arXiv:2409.14760cs.LG2024-09被引 1

用黎曼几何构建无畸变数据表示,提升三维几何学习效果

Isometric Immersion Learning with Riemannian Geometry

  • 基于黎曼几何设计神经网络,同时学习流形与度量结构
  • 在多个3D几何数据集上显著优于现有方法,提升明显
  • 可直接用于真实场景预测,平均提升8.8%准确率

流形学习被证明能有效捕捉非欧几里得数据的内在结构,但如何保持表示的无畸变(等距)仍是主要挑战。现有方法缺乏等距的理论保障。受纳什等距嵌入定理启发,本文提出基于黎曼几何的等距嵌入学习新范式。据此,设计了一种集成黎曼几何先验的无监督神经网络模型,实现度量与流形学习的联合优化。进一步提出基于最大似然估计的训练方法并完成算法实现。在多种3D几何数据集上的仿真实验表明,该模型在多个评估指标上均显著优于当前最优基线。此外,将模型学习到的黎曼度量应用于真实场景的下游预测任务,平均准确率提升8.8%。

原文摘要 · Abstract (English)

Manifold learning has been proven to be an effective method for capturing the implicitly intrinsic structure of non-Euclidean data, in which one of the primary challenges is how to maintain the distortion-free (isometry) of the data representations. Actually, there is still no manifold learning method that provides a theoretical guarantee of isometry. Inspired by Nash's isometric theorem, we introduce a new concept called isometric immersion learning based on Riemannian geometry principles. Following this concept, an unsupervised neural network-based model that simultaneously achieves metric and manifold learning is proposed by integrating Riemannian geometry priors. What's more, we theoretically derive and algorithmically implement a maximum likelihood estimation-based training method for the new model. In the simulation experiments, we compared the new model with the state-of-the-art baselines on various 3-D geometry datasets, demonstrating that the new model exhibited significantly superior performance in multiple evaluation metrics. Moreover, we applied the Riemannian metric learned from the new model to downstream prediction tasks in real-world scenarios, and the accuracy was improved by an average of 8.8%.

流形学习黎曼几何等距嵌入三维数据

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。