Code for our TMLR Journal: MMD-Regularized Unbalanced Optimal Transport

• You may first install PyTorch and Torchvision via conda install pytorch==1.13.1 torchvision==0.14.1 pytorch-cuda=11.6 -c pytorch -c nvidia.
• The other packages used in this repository can be installed via pip install -r requirements.txt.
• To install our ot_mmd package, please clone this repository and run pip install .

Algorithms

• Code for solving MMD-OT using Accelerated PGD.
  • Code for computing a batch of MMD-OT problems parallelly.
• Code for solving the MMD-OT barycenter problem using Accelerated PGD.

Basic Examples

• OT plan between Gaussians.
• Barycenter between Gaussians.

More Examples

Video Presentation

If you find this useful, consider giving a ⭐ to this repository & citing our work.