Riemannian Multinomial Logistics Regression for SPD Neural Networks

Ziheng Chen,Yue Song,Gaowen Liu,Ramana Rao Kompella,Xiaojun Wu,Nicu Sebe
2024-03-20
Abstract:Deep neural networks for learning Symmetric Positive Definite (SPD) matrices are gaining increasing attention in machine learning. Despite the significant progress, most existing SPD networks use traditional Euclidean classifiers on an approximated space rather than intrinsic classifiers that accurately capture the geometry of SPD manifolds. Inspired by Hyperbolic Neural Networks (HNNs), we propose Riemannian Multinomial Logistics Regression (RMLR) for the classification layers in SPD networks. We introduce a unified framework for building Riemannian classifiers under the metrics pulled back from the Euclidean space, and showcase our framework under the parameterized Log-Euclidean Metric (LEM) and Log-Cholesky Metric (LCM). Besides, our framework offers a novel intrinsic explanation for the most popular LogEig classifier in existing SPD networks. The effectiveness of our method is demonstrated in three applications: radar recognition, human action recognition, and electroencephalography (EEG) classification. The code is available at <a class="link-external link-https" href="https://github.com/GitZH-Chen/SPDMLR.git" rel="external noopener nofollow">this https URL</a>.
Machine Learning
What problem does this paper attempt to address?
The paper attempts to address the issue of traditional Euclidean classifiers being used in approximate spaces rather than intrinsic classifiers that accurately capture the geometry of the Symmetric Positive Definite (SPD) manifold when performing classification on SPD matrices. Specifically, although existing SPD neural networks have made significant progress, most networks still rely on traditional Euclidean classifiers in approximate spaces rather than intrinsic classifiers that can accurately capture the geometric properties of the SPD manifold. To address this limitation, the authors propose a Riemannian Multinomial Logistic Regression (RMLR) for the classification layer of SPD neural networks and introduce a unified framework to construct Riemannian classifiers under metrics pulled back from Euclidean space. This framework demonstrates applications under the parameterized Log-Euclidean Metric (LEM) and Log-Cholesky Metric (LCM) and provides a novel intrinsic interpretation of the widely used LogEig classifier in existing SPD networks. The main contributions of the paper are: 1. A general framework is proposed for constructing SPD Multinomial Logistic Regression (SPD MLR) under metrics pulled back from Euclidean space, and specific SPD MLRs are designed under two families of parameterized metrics. 2. The framework provides an intrinsic interpretation of the widely used LogEig classifier, which consists of consecutive matrix logarithms, fully connected (FC) layers, and softmax layers. 3. Extensive experiments show that the proposed classifiers consistently outperform previous baselines in commonly used SPD benchmarks.