Optimal Transport on the Lie Group of Roto-translations

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Bon, Daan, Pai, Gautam, Bellaard, Gijs, Mula, Olga, Duits, Remco
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909387248893952
author Bon, Daan
Pai, Gautam
Bellaard, Gijs
Mula, Olga
Duits, Remco
author_facet Bon, Daan
Pai, Gautam
Bellaard, Gijs
Mula, Olga
Duits, Remco
contents The roto-translation group SE2 has been of active interest in image analysis due to methods that lift the image data to multi-orientation representations defined on this Lie group. This has led to impactful applications of crossing-preserving flows for image de-noising, geodesic tracking, and roto-translation equivariant deep learning. In this paper, we develop a computational framework for optimal transportation over Lie groups, with a special focus on SE2. We make several theoretical contributions (generalizable to matrix Lie groups) such as the non-optimality of group actions as transport maps, invariance and equivariance of optimal transport, and the quality of the entropic-regularized optimal transport plan using geodesic distance approximations. We develop a Sinkhorn like algorithm that can be efficiently implemented using fast and accurate distance approximations of the Lie group and GPU-friendly group convolutions. We report valuable advancements in the experiments on 1) image barycentric interpolation, 2) interpolation of planar orientation fields, and 3) Wasserstein gradient flows on SE2. We observe that our framework of lifting images to SE2 and optimal transport with left-invariant anisotropic metrics leads to equivariant transport along dominant contours and salient line structures in the image. This yields sharper and more meaningful interpolations compared to their counterparts on R^2
format Preprint
id arxiv_https___arxiv_org_abs_2402_15322
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Optimal Transport on the Lie Group of Roto-translations
Bon, Daan
Pai, Gautam
Bellaard, Gijs
Mula, Olga
Duits, Remco
Computer Vision and Pattern Recognition
Differential Geometry
Optimization and Control
62H35, 68U10, 90B06, 68T45, 68U99
The roto-translation group SE2 has been of active interest in image analysis due to methods that lift the image data to multi-orientation representations defined on this Lie group. This has led to impactful applications of crossing-preserving flows for image de-noising, geodesic tracking, and roto-translation equivariant deep learning. In this paper, we develop a computational framework for optimal transportation over Lie groups, with a special focus on SE2. We make several theoretical contributions (generalizable to matrix Lie groups) such as the non-optimality of group actions as transport maps, invariance and equivariance of optimal transport, and the quality of the entropic-regularized optimal transport plan using geodesic distance approximations. We develop a Sinkhorn like algorithm that can be efficiently implemented using fast and accurate distance approximations of the Lie group and GPU-friendly group convolutions. We report valuable advancements in the experiments on 1) image barycentric interpolation, 2) interpolation of planar orientation fields, and 3) Wasserstein gradient flows on SE2. We observe that our framework of lifting images to SE2 and optimal transport with left-invariant anisotropic metrics leads to equivariant transport along dominant contours and salient line structures in the image. This yields sharper and more meaningful interpolations compared to their counterparts on R^2
title Optimal Transport on the Lie Group of Roto-translations
topic Computer Vision and Pattern Recognition
Differential Geometry
Optimization and Control
62H35, 68U10, 90B06, 68T45, 68U99
url https://arxiv.org/abs/2402.15322