Controlling The Fréchet Variance Improves Batch Normalization on the Symmetric Positive Definite Manifold

Reinmar J. Kobler,Jun-ichiro Hirayama,Motoaki Kawanabe
DOI: https://doi.org/10.1109/icassp43922.2022.9746629
2022-05-23
Abstract:Symmetric positive definite (SPD) matrices, and in particular co-variance matrices as data descriptors find widespread application in various fields but also pure machine learning. SPD matrices form a Riemannian manifold, demanding machine learning methods that take this structure into account. In this work, we extend upon previous works and propose a batch normalization algorithm for the SPD manifold that can be readily combined with SPD neural networks and unlike previous works controls both the Fréchet mean and variance on the SPD manifold. The proposed method is validated in simulations and datasets with small sample sizes from three application domains: action recognition from human motion trajectories, image classification and mental imagery detection of electroencephalographic (EEG) signals. The combined results show a systematic performance increase upon previous works and tangent space approximations, as well as improved robustness to low signal-to-noise ratios and lack of data as often encountered in neuroimaging and brain-computer interfacing (BCI).
What problem does this paper attempt to address?