arXiv:2512.08218cs.LGcs.AI2025-12被引 1

用可自适应曲率的伪黎曼空间提升图神经网络的表征能力。

PR-CapsNet: Pseudo-Riemannian Capsule Network with Adaptive Curvature Routing for Graph Learning

  • 引入可变曲率的伪黎曼流形,改进胶囊网络的动态路由机制。
  • 在多个图分类任务上超越现有最优模型,尤其擅长捕捉复杂结构。
  • 适合研究图表示学习与几何深度学习的学者参考。

胶囊网络(CapsNets)通过动态路由和向量化的层次化表示,在图表示学习中展现强大能力,但受限于固定曲率空间,难以建模真实世界图的复杂几何结构,导致性能不佳。近期研究发现,非欧几里得的伪黎曼流形能为图数据嵌入提供特定归纳偏置,但如何将其融入胶囊网络仍不明确。本文提出伪黎曼胶囊网络(PR-CapsNet),将欧氏胶囊路由扩展至具有测地线不连通性的伪黎曼流形,并构建自适应曲率的空间路由机制。PR-CapsNet通过微分同胚变换将胶囊状态分解为球面-时间与欧氏-空间子空间;设计自适应曲率路由,利用可学习的曲率张量和局部流形几何注意力,融合不同曲率空间的特征;最后构建保持几何性质的伪黎曼胶囊分类器,通过曲率加权softmax进行分类。在节点与图分类基准测试中,PR-CapsNet显著优于当前最先进模型,验证了其对复杂图结构的强大表征能力。

原文摘要 · Abstract (English)

Capsule Networks (CapsNets) show exceptional graph representation capacity via dynamic routing and vectorized hierarchical representations, but they model the complex geometries of real\-world graphs poorly by fixed\-curvature space due to the inherent geodesical disconnectedness issues, leading to suboptimal performance. Recent works find that non\-Euclidean pseudo\-Riemannian manifolds provide specific inductive biases for embedding graph data, but how to leverage them to improve CapsNets is still underexplored. Here, we extend the Euclidean capsule routing into geodesically disconnected pseudo\-Riemannian manifolds and derive a Pseudo\-Riemannian Capsule Network (PR\-CapsNet), which models data in pseudo\-Riemannian manifolds of adaptive curvature, for graph representation learning. Specifically, PR\-CapsNet enhances the CapsNet with Adaptive Pseudo\-Riemannian Tangent Space Routing by utilizing pseudo\-Riemannian geometry. Unlike single\-curvature or subspace\-partitioning methods, PR\-CapsNet concurrently models hierarchical and cluster or cyclic graph structures via its versatile pseudo\-Riemannian metric. It first deploys Pseudo\-Riemannian Tangent Space Routing to decompose capsule states into spherical\-temporal and Euclidean\-spatial subspaces with diffeomorphic transformations. Then, an Adaptive Curvature Routing is developed to adaptively fuse features from different curvature spaces for complex graphs via a learnable curvature tensor with geometric attention from local manifold properties. Finally, a geometric properties\-preserved Pseudo\-Riemannian Capsule Classifier is developed to project capsule embeddings to tangent spaces and use curvature\-weighted softmax for classification. Extensive experiments on node and graph classification benchmarks show PR\-CapsNet outperforms SOTA models, validating PR\-CapsNet's strong representation power for complex graph structures.

图学习胶囊网络几何深度学习

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