Communication Dans Un Congrès Année : 2025

Riemannian flow matching for brain connectivity matrices via pullback geometry

Résumé

Generating realistic brain connectivity matrices is key to analyzing population heterogeneity in brain organization, understanding disease, and augmenting data in challenging classification problems. Functional connectivity matrices lie in constrained spaces, such as the set of symmetric positive definite or correlation matrices, that can be modeled as Riemannian manifolds. However, using Riemannian tools typically requires redefining core operations (geodesics, norms, integration), making generative modeling computationally inefficient. In this work, we propose DiffeoCFM, an approach that enables conditional flow matching (CFM) on matrix manifolds by exploiting pullback metrics induced by global diffeomorphisms on Euclidean spaces. We show that Riemannian CFM with such metrics is equivalent to applying standard CFM after data transformation. This equivalence allows efficient vector field learning, and fast sampling with standard ODE solvers. We instantiate DiffeoCFM with two different settings: the matrix logarithm for covariance matrices and the normalized Cholesky decomposition for correlation matrices. We evaluate DiffeoCFM on three large-scale fMRI datasets with more than 4600 scans from 2800 subjects (ADNI, ABIDE, OASIS-3) and two EEG motor imagery datasets with over 30000 trials from 26 subjects (BNCI2014-002 and BNCI2015-001). It enables fast training and achieves state-of-the-art performance, all while preserving manifold constraints.

Fichier principal
Vignette du fichier
Neurips__Flow_matching_for_brain_connectivity.pdf (10.1 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)
Licence

Dates et versions

hal-05369430 , version 1 (17-11-2025)

Licence

Identifiants

Citer

Antoine Collas, Ce Ju, Nicolas Salvy, Bertrand Thirion. Riemannian flow matching for brain connectivity matrices via pullback geometry. NeurIPS 2025 - Thirty-Ninth Annual Conference on Neural Information Processing Systems, Dec 2025, San Diego, United States. ⟨hal-05369430⟩
59 Consultations
223 Téléchargements

Altmetric

Partager

  • More