LieBN Introduces Batch Normalization for Lie Group Manifolds.

Ziheng Chen, Yue Song, Rui Wang, Xiao-Jun Wu, Nicu Sebe· July 13, 2026 View original

Key takeaways

  • LieBN provides a unified framework for Batch Normalization over Lie groups.
  • It leverages invariant metrics to effectively normalize manifold-valued data.
  • The method offers theoretical guarantees for controlling Riemannian mean and variance.
  • LieBN improves training stability and performance for DNNs on complex geometries.

Who benefits

RoboticsComputer VisionMedical ImagingAI/ML DevelopmentAutonomous Systems

Summary

LieBN is a new framework for Riemannian Batch Normalization (RBN) over Lie groups, addressing limitations of existing methods for manifold-valued measurements in machine learning. It leverages left- and right-invariant metrics to effectively normalize sample distributions on various geometries.

Researchers have introduced LieBN, a novel framework for Riemannian Batch Normalization (RBN) specifically designed for operations over Lie groups. This innovation addresses a gap in machine learning where manifold-valued measurements are common, but existing normalization techniques are often limited to specific geometries or fail to effectively normalize sample distributions on these complex manifolds. LieBN capitalizes on the inherent left- and right-invariant metrics present in every Lie group, providing theoretical guarantees for controlling Riemannian mean and variance. This principled approach allows for robust normalization across diverse geometric structures. The framework has been instantiated and validated across nine distinct geometries, including four on the Symmetric Positive Definite (SPD) manifold, one on the group of rotation matrices, and four on full-rank correlation matrices. This includes the introduction of a new right-invariant metric for SPD manifolds and extensions of existing Lie group structures. Extensive experiments confirm its effectiveness.

Why it matters

This advancement enables more stable and effective training of deep neural networks on complex, non-Euclidean data, opening new possibilities for applications in areas like computer vision, medical imaging, and robotics.

How to implement this in your domain

  1. 1Identify machine learning tasks involving manifold-valued data (e.g., rotation matrices, SPD matrices) where current normalization methods are suboptimal.
  2. 2Explore integrating LieBN into existing deep learning architectures that operate on Lie groups or Riemannian manifolds.
  3. 3Benchmark the training stability, convergence speed, and final performance of models using LieBN against traditional or specialized normalization techniques.
  4. 4Consult the provided code repository to understand implementation details and adapt it to specific research or product needs.

Original post by Ziheng Chen, Yue Song, Rui Wang, Xiao-Jun Wu, Nicu Sebe

"arXiv:2607.08783v1 Announce Type: cross Abstract: Manifold-valued measurements are prevalent in various machine learning tasks. Recent advances have extended Deep Neural Networks (DNNs) to operate on manifolds, accompanied by normalization techniques tailored to different geometr…"

View on X

Originally posted by Ziheng Chen, Yue Song, Rui Wang, Xiao-Jun Wu, Nicu Sebe on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses

More in AI Research

AI Engineering & DevToolsAI Research

Resilient Decentralized Federated Learning for Wireless IoT Networks

This paper introduces QEF-GT-AdamW, a communication-efficient and outage-resilient algorithm for decentralized federated learning over wireless IoT networks. It combines gradient tracking, AdamW optimization, and dual-stream biased quantization with error feedback to improve robustness and convergence under heterogeneous data and unreliable communication.

Nguyen Van Thieu, Ti Ti Nguyen, Ons Aouedi, Vu Nguyen Ha, Symeon ChatzinotasAug 27, 2026
AI Engineering & DevToolsAI Research

FedQoS Predicts QoS Risk for Wireless Access Selection

This paper proposes FedQoS, a federated QoS-risk learning framework that predicts future QoS degradation for reliable access selection in heterogeneous indoor-outdoor wireless environments. It enables access nodes to locally learn from network logs and collaboratively train a global predictor without centralizing user data, significantly reducing QoS failure rates.

Nguyen Van Thieu, Ti Ti Nguyen, Ons Aouedi, Zerihun Huruy, Vu Nguyen Ha, Symeon ChatzinotasAug 27, 2026
AI ResearchAI Engineering & DevTools

Parametric Knowledge Graphs Show Storage-Retrieval Gap

This paper explores compiling knowledge graphs into LoRA adapters for parametric memory, finding that while adapters effectively store factual knowledge, retrieving it via semantic similarity or weight-space geometry is ineffective. This highlights a "storage-retrieval gap" and the need for new query-conditioned composition mechanisms.

Martino M. L. Pulici, Cuong Xuan Chu, Evgeny Kharlamov, Volker TrespAug 27, 2026