Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Geometric Mixture-of-Experts with Curvature-Guided Adaptive Routing for Graph Representation Learning

About

Graph-structured data typically exhibits complex topological heterogeneity, making it difficult to model accurately within a single Riemannian manifold. While emerging mixed-curvature methods attempt to capture such diversity, they often rely on implicit, task-driven routing that lacks fundamental geometric grounding. To address this challenge, we propose a Geometric Mixture-of-Experts framework (GeoMoE) that adaptively fuses node representations across diverse Riemannian spaces to better accommodate multi-scale topological structures. At its core, GeoMoE leverages Ollivier-Ricci Curvature (ORC) as an intrinsic geometric prior to orchestrate the collaboration of specialized experts. Specifically, we design a graph-aware gating network that assigns node-specific fusion weights, regularized by a curvature-guided alignment loss to ensure interpretable and geometry-consistent routing. Additionally, we introduce a curvature-aware contrastive objective that promotes geometric discriminability by constructing positive and negative pairs according to curvature consistency. Extensive experiments on six benchmark datasets demonstrate that GeoMoE outperforms state-of-the-art baselines across diverse graph types.

Haifang Cao, Yu Wang, Timing Li, Xinjie Yao, Pengfei Zhu• 2026

Related benchmarks

TaskDatasetResultRank
Node ClassificationPubmed
Accuracy80.13
396
Node ClassificationCiteseer
Accuracy74.72
393
Node ClassificationPhoto
Mean Accuracy95.18
343
Node ClassificationwikiCS
Accuracy79.65
317
Showing 4 of 4 rows

Other info

Follow for update