Fast and Stable Riemannian Metrics on SPD Manifolds via Cholesky Product Geometry
Ziheng Chen, Yue Song, Xiaojun Wu, Nicu Sebe
摘要
Recent advances in Symmetric Positive Definite (SPD) matrix learning show that Riemannian metrics are fundamental to effective SPD neural networks. Motivated by this, we revisit the geometry of the Cholesky factors and uncover a simple product structure that enables convenient metric design. Building on this insight, we propose two fast and stable SPD metrics, Power--Cholesky Metric (PCM) and Bures--Wasserstein--Cholesky Metric (BWCM), derived via Cholesky decomposition. Compared with existing SPD metrics, the proposed metrics provide closed-form operators, computational efficiency, and improved numerical stability. We further apply our metrics to construct Riemannian Multinomial Logistic Regression (MLR) classifiers and residual blocks for SPD neural networks. Experiments on SPD deep learning, numerical stability analyses, and tensor interpolation demonstrate the effectiveness, efficiency, and robustness of our metrics. The code is available at https://github.com/GitZH-Chen/PCM_BWCM.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper2
- Proper Velocity Neural NetworksZiheng Chen, Zihan Su, Bernhard Schölkopf, Nicu SebeICLR 2026
- Riemannian Networks over Full-Rank Correlation MatricesZiheng Chen, Xiaojun Wu, Bernhard Schölkopf, Nicu SebeICML 2026
它引用的顶会 Paper21
- Hyperbolic Neural Networks++Ryohei Shimizu, Yusuke Mukuta, Tatsuya HaradaICLR 2021 · 被引用 791 次
- Flow Matching on General GeometriesRicky T. Q. Chen, Yaron LipmanICLR 2024 · 被引用 193 次
- Mixed-curvature Variational AutoencodersOndrej Skopek, Octavian-Eugen Ganea, Gary BécigneulICLR 2020 · 被引用 122 次
- MAtt: A Manifold Attention Network for EEG DecodingYue-Ting Pan, Jing-Lun Chou, Chun-Shu WeiNeurIPS 2022 · 被引用 105 次
- SPD domain-specific batch normalization to crack interpretable unsupervised domain adaptation in EEGReinmar J. Kobler, Jun-ichiro Hirayama, Qibin Zhao, Motoaki KawanabeNeurIPS 2022 · 被引用 102 次
相关 Paper
- Riemannian Multinomial Logistics Regression for SPD Neural NetworksZiheng Chen, Yue Song, Gaowen Liu, Ramana Rao Kompella 等CVPR 2024
- Learning to Normalize on the SPD Manifold under Bures-Wasserstein GeometryRui Wang, Shaocheng Jin, Ziheng Chen, Xiaoqing Luo 等CVPR 2025
- RMLR: Extending Multinomial Logistic Regression into General GeometriesZiheng Chen, Yue Song, Rui Wang, Xiaojun Wu 等NeurIPS 2024 · 被引用 16 次
- On Riemannian Optimization over Positive Definite Matrices with the Bures-Wasserstein GeometryAndi Han, Bamdev Mishra, Pratik Kumar Jawanpuria, Junbin GaoNeurIPS 2021 · 被引用 55 次
- Riemannian Embedding Banks for Common Spatial Patterns with EEG-based SPD Neural NetworksYoon-Je Suh, Byung Hyung KimAAAI 2021 · 被引用 41 次
