A Riemannian Network for SPD Matrix Learning
About
Symmetric Positive Definite (SPD) matrix learning methods have become popular in many image and video processing tasks, thanks to their ability to learn appropriate statistical representations while respecting Riemannian geometry of underlying SPD manifolds. In this paper we build a Riemannian network architecture to open up a new direction of SPD matrix non-linear learning in a deep model. In particular, we devise bilinear mapping layers to transform input SPD matrices to more desirable SPD matrices, exploit eigenvalue rectification layers to apply a non-linear activation function to the new SPD matrices, and design an eigenvalue logarithm layer to perform Riemannian computing on the resulting SPD matrices for regular output layers. For training the proposed deep network, we exploit a new backpropagation with a variant of stochastic gradient descent on Stiefel manifolds to update the structured connection weights and the involved SPD matrix data. We show through experiments that the proposed SPD matrix network can be simply trained and outperform existing SPD matrix learning and state-of-the-art methods in three typical visual classification tasks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| EEG Classification | EEG Classification | Forward Latency (ms)0.03 | 8 | |
| Action Recognition | HDM05 (10-fold cross sample val) | Accuracy61.45 | 7 | |
| Action Recognition | HDM05 (random splits) | Accuracy61.45 | 6 | |
| EEG Classification | MAMEM LOSO (test) | Mean Accuracy19.85 | 5 | |
| EEG Classification | BCI2a LOSO (test) | Mean Test Accuracy27.74 | 5 | |
| EEG Classification | BCIcha LOSO (test) | Mean Test Accuracy50.57 | 5 | |
| EEG Classification | BCI2a (S1) | Accuracy48.09 | 5 | |
| EEG Classification | BCIcha (S2) | Accuracy87 | 5 | |
| EEG Classification | MAMEM (S1) | Accuracy32.69 | 5 | |
| Video Classification | FPHA (val) | Accuracy65.39 | 4 |