/Sinkhorn-Embeddings-MNIST

Implementation of the paper "Learning Embeddings into Entropic Wasserstein Spaces" (Sinkhorn embeddings) and application to visual embeddings with MNIST.

Primary LanguageJupyter NotebookMIT LicenseMIT

Sinkhorn embeddings on MNIST

Implementation of the paper "Learning Embeddings into Entropic Wasserstein Spaces" (Sinkhorn embeddings) and application to visual embeddings with MNIST.

Report: https://github.com/ridouaneg/Sinkhorn-Embeddings-MNIST/blob/main/OT_report.pdf

Notebook: https://nbviewer.jupyter.org/github/ridouaneg/Sinkhorn-Embeddings-MNIST/blob/main/sinkhorn_embeddings.ipynb

Credit to https://github.com/gabsens/Learning-Embeddings-into-Entropic-Wasserstein-Spaces-ENSAE for the original implementation of the Sinkhorn embeddings for word embeddings.