Adaptive Riemannian Graph Neural Networks
ARGNN learns a node specific Riemannian metric instead of imposing one fixed curvature on an entire graph. A diagonal metric parameterization and geometric regularization support efficient training, with convergence analysis and experiments on homophilic and heterophilic graphs.