Riemannian Local Mechanism for SPD Neural Networks
Ziheng Chen, Tianyang Xu, Xiao-Jun Wu, Rui Wang, Zhiwu Huang, Josef Kittler
Abstract
The Symmetric Positive Definite (SPD) matrices have received wide attention for data representation in many scientific areas. Although there are many different attempts to develop effective deep architectures for data processing on the Riemannian manifold of SPD matrices, very few solutions explicitly mine the local geometrical information in deep SPD feature representations. Given the great success of local mechanisms in Euclidean methods, we argue that it is of utmost importance to ensure the preservation of local geometric information in the SPD networks. We first analyse the convolution operator commonly used for capturing local information in Euclidean deep networks from the perspective of a higher level of abstraction afforded by category theory. Based on this analysis, we define the local information in the SPD manifold and design a multi-scale submanifold block for mining local geometry. Experiments involving multiple visual tasks validate the effectiveness of our approach. The supplement and source code can be found in https://github.com/GitZH-Chen/MSNet.git .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers12
- SPD-DDPM: Denoising Diffusion Probabilistic Models in the Symmetric Positive Definite SpaceYunchen Li, Zhou Yu, Gaoqi He, Yunhang Shen et al.AAAI 2024 · 17 citations
- RMLR: Extending Multinomial Logistic Regression into General GeometriesZiheng Chen, Yue Song, Rui Wang, Xiaojun Wu et al.NeurIPS 2024 · 16 citations
- Improving Expressivity of GNNs with Subgraph-specific Factor Embedded NormalizationKaixuan Chen, Shunyu Liu, Tongtian Zhu, Ji Qiao et al.KDD 2023 · 13 citations
- A Lie Group Approach to Riemannian Batch NormalizationZiheng Chen, Yue Song, Yunmei Liu, Nicu SebeICLR 2024 · 11 citations
- Towards a General Attention Framework on Gyrovector Spaces for Matrix ManifoldsRui Wang, Chen Hu, Xiaoning Song, Xiaojun Wu et al.NeurIPS 2025 · 5 citations
Builds on2
- Dilated Convolutional Neural Networks for Sequential Manifold-Valued DataRudrasis Chakraborty, Xingjian Zhen, Nicholas Vogt, Barbara B. Bendlin et al.ICCV 2019 · 43 citations
- GeomNet: A Neural Network Based on Riemannian Geometries of SPD Matrix Space and Cholesky Space for 3D Skeleton-Based Interaction RecognitionXuan Son NguyenICCV 2021 · 40 citations
Related papers
- Understanding Matrix Function Normalizations in Covariance Pooling through the Lens of Riemannian GeometryZiheng Chen, Yue Song, Xiaojun Wu, Gaowen Liu et al.ICLR 2025 · 1 citation
- Riemannian Embedding Banks for Common Spatial Patterns with EEG-based SPD Neural NetworksYoon-Je Suh, Byung Hyung KimAAAI 2021 · 41 citations
- Riemannian Networks over Full-Rank Correlation MatricesZiheng Chen, Xiaojun Wu, Bernhard Schölkopf, Nicu SebeICML 2026
- Matrix Manifold Neural Networks++Xuan Son Nguyen, Shuo Yang, Aymeric HistaceICLR 2024 · 11 citations
- Riemannian Multinomial Logistics Regression for SPD Neural NetworksZiheng Chen, Yue Song, Gaowen Liu, Ramana Rao Kompella et al.CVPR 2024
