RMLR: Extending Multinomial Logistic Regression into General Geometries

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Ziheng, Song, Yue, Wang, Rui, Wu, Xiaojun, Sebe, Nicu
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914962261147648
author Chen, Ziheng
Song, Yue
Wang, Rui
Wu, Xiaojun
Sebe, Nicu
author_facet Chen, Ziheng
Song, Yue
Wang, Rui
Wu, Xiaojun
Sebe, Nicu
contents Riemannian neural networks, which extend deep learning techniques to Riemannian spaces, have gained significant attention in machine learning. To better classify the manifold-valued features, researchers have started extending Euclidean multinomial logistic regression (MLR) into Riemannian manifolds. However, existing approaches suffer from limited applicability due to their strong reliance on specific geometric properties. This paper proposes a framework for designing Riemannian MLR over general geometries, referred to as RMLR. Our framework only requires minimal geometric properties, thus exhibiting broad applicability and enabling its use with a wide range of geometries. Specifically, we showcase our framework on the Symmetric Positive Definite (SPD) manifold and special orthogonal group, i.e., the set of rotation matrices. On the SPD manifold, we develop five families of SPD MLRs under five types of power-deformed metrics. On rotation matrices we propose Lie MLR based on the popular bi-invariant metric. Extensive experiments on different Riemannian backbone networks validate the effectiveness of our framework.
format Preprint
id arxiv_https___arxiv_org_abs_2409_19433
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle RMLR: Extending Multinomial Logistic Regression into General Geometries
Chen, Ziheng
Song, Yue
Wang, Rui
Wu, Xiaojun
Sebe, Nicu
Machine Learning
Artificial Intelligence
Riemannian neural networks, which extend deep learning techniques to Riemannian spaces, have gained significant attention in machine learning. To better classify the manifold-valued features, researchers have started extending Euclidean multinomial logistic regression (MLR) into Riemannian manifolds. However, existing approaches suffer from limited applicability due to their strong reliance on specific geometric properties. This paper proposes a framework for designing Riemannian MLR over general geometries, referred to as RMLR. Our framework only requires minimal geometric properties, thus exhibiting broad applicability and enabling its use with a wide range of geometries. Specifically, we showcase our framework on the Symmetric Positive Definite (SPD) manifold and special orthogonal group, i.e., the set of rotation matrices. On the SPD manifold, we develop five families of SPD MLRs under five types of power-deformed metrics. On rotation matrices we propose Lie MLR based on the popular bi-invariant metric. Extensive experiments on different Riemannian backbone networks validate the effectiveness of our framework.
title RMLR: Extending Multinomial Logistic Regression into General Geometries
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2409.19433