This is the official PyTorch implementation of the pre-print Light and Optimal Schrödinger Bridge Matching by Nikita Gushchin, Sergei Kholkin, Evgeny Burnaev, Alexander Korotin.
An example: Unpaired Adult -> Child translation by out LightSB-M solver applied in the latent space of ALAE for 1024x1024 FFHQ images.
python=3.10
Install project requirements
pip install -r requirements.txt
For Entropic Optimal Transport Benchmark (EOTBench) experiments install it from link (see their instructions)
For ALAE experiments install ALAE requirements
pip install -r ALAE/requirements.txt
ALAE
- Code for the ALAE model.
src
- LightSBM implementation with discrete optimal transport.
notebooks
- Jupyter notebooks with experiments for LightSBM.
notebooks/LightSBM_EOT.ipynb
- code for EOT Benchmark problems.
notebooks/LightSBM_MSCI.ipynb
- code for single cell data analysis problems.
notebooks/LightSBM_swiss_roll.ipynb
- code for Swiss Roll experiments.
notebooks/LightSBM_ALAE.ipynb
- Code for image experiments with ALAE.
@misc{gushchin2024light,
title={Light and Optimal Schr\"odinger Bridge Matching},
author={Nikita Gushchin and Sergei Kholkin and Evgeny Burnaev and Alexander Korotin},
year={2024},
eprint={2402.03207},
archivePrefix={arXiv},
primaryClass={cs.LG}
}