Abstract:Many measurements in computer vision and machine learning manifest as non-Euclidean data samples. Several researchers recently extended a number of deep neural network architectures for manifold valued data samples. Researchers have proposed models for manifold valued spatial data which are common in medical image processing including processing of diffusion tensor imaging (DTI) where images are fields of $3\times 3$ symmetric positive definite matrices or representation in terms of orientation distribution field (ODF) where the identification is in terms of field on hypersphere. There are other sequential models for manifold valued data that recently researchers have shown to be effective for group difference analysis in study for neuro-degenerative diseases. Although, several of these methods are effective to deal with manifold valued data, the bottleneck includes the instability in optimization for deeper networks. In order to deal with these instabilities, researchers have proposed residual connections for manifold valued data. One of the other remedies to deal with the instabilities including gradient explosion is to use normalization techniques including {\it batch norm} and {\it group norm} etc.. But, so far there is no normalization techniques applicable for manifold valued data. In this work, we propose a general normalization techniques for manifold valued data. We show that our proposed manifold normalization technique have special cases including popular batch norm and group norm techniques. On the experimental side, we focus on two types of manifold valued data including manifold of symmetric positive definite matrices and hypersphere. We show the performance gain in one synthetic experiment for moving MNIST dataset and one real brain image dataset where the representation is in terms of orientation distribution field (ODF).

Stability to Deformations of Manifold Filters and Manifold Neural Networks

Stability to Deformations of Manifold Filters and Manifold Neural Networks

Deep Manifold Computing and Visualization

Geometric Graph Filters and Neural Networks: Limit Properties and Discriminability Trade-offs

Deep Manifold Computing and Visualization Using Elastic Locally Isometric Smoothness

Dynamically Stable Poincaré Embeddings for Neural Manifolds

ManifoldNet: A Deep Network Framework for Manifold-valued Data

A Convergence Rate for Manifold Neural Networks

Markov-Lipschitz Deep Learning

Convergence of Manifold Filter-Combine Networks

SAGMAN: Stability Analysis of Graph Neural Networks on the Manifolds

Manifold Filter-Combine Networks

Beyond smoothness: A general optimization framework for graph neural networks with negative Laplacian regularization

Algebraic Neural Networks: Stability to Deformations

Deep Manifold Part 1: Anatomy of Neural Network Manifold

Learning nonlinear manifolds based on mixtures of localized linear manifolds under a self-organizing framework

Exploring the Manifold of Neural Networks Using Diffusion Geometry

A Manifold Perspective on the Statistical Generalization of Graph Neural Networks

ManifoldNorm: Extending normalizations on Riemannian Manifolds

Improved Image Classification with Manifold Neural Networks

Matrix Manifold Neural Networks++