Gyrogroup Batch Normalization
Abstract
Several Riemannian manifolds in machine learning, such as Symmetric Positive Definite (SPD), Grassmann, spherical, and hyperbolic manifolds, have been proven to admit gyro structures, thus enabling a principled and effective extension of Euclidean Deep Neural Networks (DNNs) to manifolds. Inspired by this, this study introduces a general Riemannian Batch Normalization (RBN) framework on gyrogroups, termed GyroBN. We identify the least requirements to guarantee GyroBN with theoretical control over sample statistics, referred to as \textit{pseudo-reduction} and \textit{gyroisometric gyrations}, which are satisfied by all the existing gyrogroups in machine learning. Besides, our GyroBN incorporates several existing normalization methods, including the one on general Lie groups and different types of RBN on the non-group SPD geometry. Lastly, we instantiate our GyroBN on the Grassmannian and hyperbolic spaces. Experiments on the Grassmannian and hyperbolic networks demonstrate the effectiveness of our GyroBN. The code is available at https://github.com/GitZH-Chen/GyroBN.git.
Cite
Text
Chen et al. "Gyrogroup Batch Normalization." International Conference on Learning Representations, 2025.Markdown
[Chen et al. "Gyrogroup Batch Normalization." International Conference on Learning Representations, 2025.](https://mlanthology.org/iclr/2025/chen2025iclr-gyrogroup/)BibTeX
@inproceedings{chen2025iclr-gyrogroup,
title = {{Gyrogroup Batch Normalization}},
author = {Chen, Ziheng and Song, Yue and Wu, Xiaojun and Sebe, Nicu},
booktitle = {International Conference on Learning Representations},
year = {2025},
url = {https://mlanthology.org/iclr/2025/chen2025iclr-gyrogroup/}
}