Understanding Matrix Function Normalizations in Covariance Pooling through the Lens of Riemannian Geometry
Ziheng Chen, Yue Song, Xiaojun Wu, Gaowen Liu, Nicu Sebe
Abstract
Global Covariance Pooling (GCP) has been demonstrated to improve the performance of Deep Neural Networks (DNNs) by exploiting second-order statistics of high-level representations. GCP typically performs classification of the covariance matrices by applying matrix function normalization, such as matrix logarithm or power, followed by a Euclidean classifier. However, covariance matrices inherently lie in a Riemannian manifold, known as the Symmetric Positive Definite (SPD) manifold. The current literature does not provide a satisfactory explanation of why Euclidean classifiers can be applied directly to Riemannian features after the normalization of the matrix power. To mitigate this gap, this paper provides a comprehensive and unified understanding of the matrix logarithm and power from a Riemannian geometry perspective. The underlying mechanism of matrix functions in GCP is interpreted from two perspectives: one based on tangent classifiers (Euclidean classifiers on the tangent space) and the other based on Riemannian classifiers. Via theoretical analysis and empirical validation through extensive experiments on fine-grained and large-scale visual classification datasets, we conclude that the working mechanism of the matrix functions should be attributed to the Riemannian classifiers they implicitly respect. The code is available at https://github.com/GitZH-Chen/RiemGCP.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 papers7
- RMLR: Extending Multinomial Logistic Regression into General GeometriesZiheng Chen, Yue Song, Rui Wang, Xiaojun Wu et al.NeurIPS 2024 · 16 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
- Fast and Stable Riemannian Metrics on SPD Manifolds via Cholesky Product GeometryZiheng Chen, Yue Song, Xiaojun Wu, Nicu SebeICLR 2026 · 4 citations
- Sheaf Neural Networks on SPD Manifolds: Second-Order Geometric Representation LearningYuhan Peng, Junwen Dong, Yuzhi Zeng, Hao Li et al.ICML 2026 · 1 citation
- Riemannian High-Order Pooling for Brain Foundation ModelsChen Hu, Ziheng Chen, Rui Wang, Yefeng Zheng et al.ICLR 2026
Builds on17
- Swin Transformer: Hierarchical Vision Transformer using Shifted WindowsZe Liu, Yutong Lin, Yue Cao, Han Hu et al.ICCV 2021 · 31,683 citations
- Training data-efficient image transformers & distillation through attentionHugo Touvron, Matthieu Cord, Matthijs Douze, Francisco Massa et al.ICML 2021 · 8,974 citations
- Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNetLi Yuan, Yunpeng Chen, Tao Wang, Weihao Yu et al.ICCV 2021 · 2,462 citations
- Hyperbolic Neural Networks++Ryohei Shimizu, Yusuke Mukuta, Tatsuya HaradaICLR 2021 · 791 citations
- On Riemannian Optimization over Positive Definite Matrices with the Bures-Wasserstein GeometryAndi Han, Bamdev Mishra, Pratik Kumar Jawanpuria, Junbin GaoNeurIPS 2021 · 55 citations
Related papers
- Riemannian Local Mechanism for SPD Neural NetworksZiheng Chen, Tianyang Xu, Xiao-Jun Wu, Rui Wang et al.AAAI 2023 · 34 citations
- What Deep CNNs Benefit From Global Covariance Pooling: An Optimization PerspectiveQilong Wang, Li Zhang, Banggu Wu, Dongwei Ren et al.CVPR 2020
- Riemannian Multinomial Logistics Regression for SPD Neural NetworksZiheng Chen, Yue Song, Gaowen Liu, Ramana Rao Kompella et al.CVPR 2024
- Riemannian Embedding Banks for Common Spatial Patterns with EEG-based SPD Neural NetworksYoon-Je Suh, Byung Hyung KimAAAI 2021 · 41 citations
- Matrix Manifold Neural Networks++Xuan Son Nguyen, Shuo Yang, Aymeric HistaceICLR 2024 · 11 citations
