Work

Bispectral OT: Dataset Comparison using Symmetry-Aware Optimal Transport

Year: 2025
Type: preprint
Abstract: Optimal transport (OT) is a widely used technique in machine learning, graphics, and vision that aligns two distributions or datasets using their relative geometry. In symmetry-rich settings, however, OT alignments based solely on pairwise geometric distances between raw features can ignore the intrinsic coherence structure of the data. We introduce Bispectral Optimal Transport, a symmetry-aware extension of discrete OT that compares elements using their representation using the bispectrum, a group Fourier invariant that preserves all signal structure while removing only the variation due to group actions. Empirically, we demonstrate that the transport plans computed with Bispectral OT achieve greater class preservation accuracy than naive feature OT on benchmark datasets transformed with visual symmetries, improving the quality of meaningful correspondences that capture the underlying semantic label structure in the dataset while removing nuisance variation not affecting class or content.
Authors: Ma, Annabel, Hou, Kaiying, Alvarez-Melis, David, Weber, Melanie
Language: English
Open Access status: green
OpenAlex ID: W4414788692
Cites:
Cited by:
Related to: