Neural Networks on Symmetric Spaces of Noncompact Type
Xuan Son Nguyen, Shuo Yang, Aymeric Histace
TL;DR
This work extends neural networks to symmetric spaces of noncompact type, unifying hyperplane concepts across hyperbolic spaces and SPD manifolds. It introduces a general point-to-hyperplane distance based on Busemann functions and group-theoretic structure, deriving closed-form expressions for higher rank spaces with G-invariant metrics. This distance enables new fully-connected layers and an attention mechanism that operate directly on the manifold, and the authors validate the approach on image classification, EEG classification, image generation, and natural language inference. The results demonstrate the potential of symmetric-space neural networks to capture hierarchical and geometric structure in diverse data while offering competitive performance and compact representations.
Abstract
Recent works have demonstrated promising performances of neural networks on hyperbolic spaces and symmetric positive definite (SPD) manifolds. These spaces belong to a family of Riemannian manifolds referred to as symmetric spaces of noncompact type. In this paper, we propose a novel approach for developing neural networks on such spaces. Our approach relies on a unified formulation of the distance from a point to a hyperplane on the considered spaces. We show that some existing formulations of the point-to-hyperplane distance can be recovered by our approach under specific settings. Furthermore, we derive a closed-form expression for the point-to-hyperplane distance in higher-rank symmetric spaces of noncompact type equipped with G-invariant Riemannian metrics. The derived distance then serves as a tool to design fully-connected (FC) layers and an attention mechanism for neural networks on the considered spaces. Our approach is validated on challenging benchmarks for image classification, electroencephalogram (EEG) signal classification, image generation, and natural language inference.
