Optimal Transport with Symmetry Groups
Jiechao Zhang, Huichun Zhang, Jian Sun, Wei Zeng
Abstract
We propose a novel algorithm that accelerates optimal transport by exploiting intrinsic symmetries induced by finite group actions. The core of our approach is to recover the orbit decomposition and the associated algebraic structure directly from the cost matrix—without requiring prior knowledge of the group—and to reduce the original transport problem to a substantially smaller problem on the orbit space. This reduction preserves optimality while achieving a significant drop in computational complexity. We develop efficient solvers for two central classes of optimal transport: linear OT and entropy-regularized OT. Experiments on synthetic data, real-world image datasets, and molecular graph data confirm the efficiency and robustness of the method. To our knowledge, this work is the first to systematically incorporate symmetry groups into optimal transport, providing both a theoretical framework and a practical pathway to computational acceleration.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on4
- Keypoint-Guided Optimal Transport with Applications in Heterogeneous Domain AdaptationXiang Gu, Yucheng Yang, Wei Zeng, Jian Sun et al.NeurIPS 2022 · 43 citations
- Semantic Correspondence as an Optimal Transport ProblemYanbin Liu, Linchao Zhu, Makoto Yamada, Yi YangCVPR 2020
- Gromov-Wasserstein Problem with Cyclic SymmetryShoichiro Takeda, Yasunori AkagiCVPR 2025
- Optimal Transport with Cyclic SymmetryShoichiro Takeda, Yasunori Akagi, Naoki Marumo, Kenta NiwaAAAI 2024
Related papers
- Recovery Bounds on Class-Based Optimal Transport: A Sum-of-Norms Regularization FrameworkArman Rahbar, Ashkan Panahi, Morteza Haghir Chehreghani, Devdatt P. Dubhashi et al.ICML 2023
- Meta Optimal TransportBrandon Amos, Giulia Luise, Samuel Cohen, Ievgen RedkoICML 2023 · 32 citations
- A Truncated Newton Method for Optimal TransportMete Kemertas, Amir-massoud Farahmand, Allan Douglas JepsonICLR 2025
- Scalable Optimal Transport in High Dimensions for Graph Distances, Embedding Alignment, and MoreJohannes Klicpera, Marten Lienen, Stephan GünnemannICML 2021 · 14 citations
- A fast and accurate splitting method for optimal transport: analysis and implementationVien V. Mai, Jacob Lindbäck, Mikael JohanssonICLR 2022 · 15 citations
