Fused Unbalanced Gromov-Wasserstein for Python

This package implements multiple GPU-compatible PyTorch solvers to the Fused Unbalanced Gromov-Wasserstein optimal transport problem.

This package is under active development. There is no guarantee that the API and solvers won’t change in the near future.

Installation

To install this package, make sure you have an up-to-date version of pip.

From PyPI

In a dedicated Python env, run:

pip install fugw

From source

git clone https://github.com/alexisthual/fugw.git
cd fugw

In a dedicated Python env, run:

pip install -e .

Contributors should also install the development dependencies in order to test and automatically format their contributions.

pip install -e ".[dev]"
pre-commit install

Tests run on CPU and GPU, depending on the configuration of your machine. You can run them with:

pytest

Citing this work

If this package was useful to you, please cite it in your work:

@article{Thual-2022-fugw,
  title={Aligning individual brains with Fused Unbalanced Gromov-Wasserstein},
  author={Thual, Alexis and Tran, Huy and Zemskova, Tatiana and Courty, Nicolas and Flamary, Rémi and Dehaene, Stanislas and Thirion, Bertrand},
  publisher={arXiv},
  doi={10.48550/ARXIV.2206.09398},
  url={https://arxiv.org/abs/2206.09398},
  year={2022},
  copyright={Creative Commons Attribution 4.0 International}
}