Aligning individual brains with fused unbalanced Gromov Wasserstein
Alexis Thual, Quang Huy Tran, Tatiana Zemskova, Nicolas Courty, Rémi Flamary, Stanislas Dehaene, Bertrand Thirion
Abstract
Individual brains vary in both anatomy and functional organization, even within a given species. Inter-individual variability is a major impediment when trying to draw generalizable conclusions from neuroimaging data collected on groups of subjects. Current co-registration procedures rely on limited data, and thus lead to very coarse inter-subject alignments. In this work, we present a novel method for inter-subject alignment based on Optimal Transport, denoted as Fused Unbalanced Gromov Wasserstein (FUGW). The method aligns cortical surfaces based on the similarity of their functional signatures in response to a variety of stimulation settings, while penalizing large deformations of individual topographic organization. We demonstrate that FUGW is well-suited for whole-brain landmark-free alignment. The unbalanced feature allows to deal with the fact that functional areas vary in size across subjects. Our results show that FUGW alignment significantly increases between-subject correlation of activity for independent functional data, and leads to more precise mapping at the group level.
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.
Cited by top-tier papers10
- MindEye2: Shared-Subject Models Enable fMRI-To-Image With 1 Hour of DataPaul S. Scotti, Mihir Tripathy, Cesar Torrico, Reese Kneeland et al.ICML 2024 · 117 citations
- GENOT: Entropic (Gromov) Wasserstein Flow Matching with Applications to Single-Cell GenomicsDominik Klein, Théo Uscidda, Fabian J. Theis, Marco CuturiNeurIPS 2024 · 34 citations
- Temporally Consistent Unbalanced Optimal Transport for Unsupervised Action SegmentationMing Xu, Stephen GouldCVPR 2024 · 15 citations
- Any2Graph: Deep End-To-End Supervised Graph Prediction With An Optimal Transport LossPaul Krzakala, Junjie Yang, Rémi Flamary, Florence d'Alché-Buc et al.NeurIPS 2024 · 7 citations
- Unsupervised Learning for Optimal Transport plan prediction between unbalanced graphsSonia Mazelet, Rémi Flamary, Bertrand ThirionNeurIPS 2025 · 5 citations
Builds on3
- The Unbalanced Gromov Wasserstein Distance: Conic Formulation and RelaxationThibault Séjourné, François-Xavier Vialard, Gabriel PeyréNeurIPS 2021 · 106 citations
- Unbalanced Optimal Transport through Non-negative Penalized Linear RegressionLaetitia Chapel, Rémi Flamary, Haoran Wu, Cédric Févotte et al.NeurIPS 2021 · 67 citations
- Modeling Shared responses in Neuroimaging Studies through MultiView ICAHugo Richard, Luigi Gresele, Aapo Hyvärinen, Bertrand Thirion et al.NeurIPS 2020 · 29 citations
Related papers
- Partial Soft-Matching Distance For Neural Representational Comparison With Partial Unit CorrespondenceChaitanya Kapoor, Alex H Williams, Meenakshi KhoslaICLR 2026 · 2 citations
- Representational Alignment Across Model Layers and Brain Regions with Multi-Level Optimal TransportShaan Shah, Meenakshi KhoslaICLR 2026 · 4 citations
- Graph-Based Decoding Model for Functional Alignment of Unaligned fMRI DataWeida Li, Mingxia Liu, Fang Chen, Daoqiang ZhangAAAI 2020 · 6 citations
- Joint Metric Space Embedding by Unbalanced Optimal Transport with Gromov-Wasserstein Marginal PenalizationFlorian Beier, Moritz Piening, Robert Beinert, Gabriele SteidlICML 2025
- Hierarchical Multi-Marginal Optimal Transport for Network AlignmentZhichen Zeng, Boxin Du, Si Zhang, Yinglong Xia et al.AAAI 2024 · 39 citations
