Rigid Invariant Sliced Wasserstein via Independent Embeddings

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Heile, Zakk, He, Peilin, Tran, Jayson, Wang, Alice, Chand, Shrikant
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914463699959808
author Heile, Zakk
He, Peilin
Tran, Jayson
Wang, Alice
Chand, Shrikant
author_facet Heile, Zakk
He, Peilin
Tran, Jayson
Wang, Alice
Chand, Shrikant
contents Comparing probability measures modulo unknown rigid transformations is a central challenge in geometric data analysis. Classical optimal transport (OT) distances, including Wasserstein and sliced Wasserstein, are sensitive to rotations and reflections, whereas Gromov-Wasserstein (GW) and Procrustes-Wasserstein (PW) distances are invariant to isometries but computationally prohibitive for large datasets. We introduce Rigid-Invariant Sliced Wasserstein via Independent Embeddings (RISWIE), a scalable distance that combines the invariance of NP-hard approaches with the efficiency of projection-based OT. RISWIE utilizes data-adaptive bases and matches optimal signed permutations along axes according to distributional similarity to achieve rigid invariance with nearly linear complexity in the sample size. We prove bounds relating RISWIE to GW in special cases and demonstrate dimension-independent statistical stability. Our experiments on cellular imaging and 3D human meshes demonstrate that RISWIE outperforms GW and PW in clustering tasks and discriminative capability while significantly reducing runtime.
format Preprint
id arxiv_https___arxiv_org_abs_2510_10233
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Rigid Invariant Sliced Wasserstein via Independent Embeddings
Heile, Zakk
He, Peilin
Tran, Jayson
Wang, Alice
Chand, Shrikant
Computational Geometry
Comparing probability measures modulo unknown rigid transformations is a central challenge in geometric data analysis. Classical optimal transport (OT) distances, including Wasserstein and sliced Wasserstein, are sensitive to rotations and reflections, whereas Gromov-Wasserstein (GW) and Procrustes-Wasserstein (PW) distances are invariant to isometries but computationally prohibitive for large datasets. We introduce Rigid-Invariant Sliced Wasserstein via Independent Embeddings (RISWIE), a scalable distance that combines the invariance of NP-hard approaches with the efficiency of projection-based OT. RISWIE utilizes data-adaptive bases and matches optimal signed permutations along axes according to distributional similarity to achieve rigid invariance with nearly linear complexity in the sample size. We prove bounds relating RISWIE to GW in special cases and demonstrate dimension-independent statistical stability. Our experiments on cellular imaging and 3D human meshes demonstrate that RISWIE outperforms GW and PW in clustering tasks and discriminative capability while significantly reducing runtime.
title Rigid Invariant Sliced Wasserstein via Independent Embeddings
topic Computational Geometry
url https://arxiv.org/abs/2510.10233