Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Entering the Era of Discrete Diffusion Models: A Benchmark for Schr\"odinger Bridges and Entropic Optimal Transport

About

The Entropic Optimal Transport (EOT) problem and its dynamic counterpart, the Schr\"odinger bridge (SB) problem, play an important role in modern machine learning, linking generative modeling with optimal transport theory. While recent advances in discrete diffusion and flow models have sparked growing interest in applying SB methods to discrete domains, there remains no reliable way to assess how well these methods actually solve the underlying problem. We address this challenge by introducing a benchmark for SB on discrete spaces. Our construction yields pairs of probability distributions with analytically known SB solutions, enabling rigorous evaluation. As a byproduct of building this benchmark, we obtain two new SB algorithms, DLightSB and DLightSB-M, and additionally extend prior related work to construct the $\alpha$-CSBM algorithm. We demonstrate the utility of our benchmark by evaluating both existing and new solvers in high-dimensional discrete settings. This work provides the first step toward proper evaluation of SB methods on discrete spaces, paving the way for more reproducible future studies. The code for the benchmark and all associated experiments is available at https://github.com/gregkseno/catsbench.

Xavier Aramayo Carrasco, Grigoriy Ksenofontov, Aleksei Leonov, Iaroslav Sergeevich Koshelev, Alexander Korotin• 2025

Related benchmarks

TaskDatasetResultRank
Discrete-space EOT/SB solvingHigh-dimensional Gaussian Mixture
Conditional Shape Score95
21
Discrete-space EOT/SB modelingHigh-Dimensional Gaussian Mixture D=2, gamma=0.02 v1 (test)
Conditional Trend Score91
16
Discrete-space EOT/SB modelingHigh-Dimensional Gaussian Mixture D=2, gaussian, gamma=0.05 v1 (test)
Conditional Trend Score85
16
Discrete-space EOT/SB modelingHigh-Dimensional Gaussian Mixture D=2, uniform, gamma=0.005 v1 (test)
Conditional Trend Score0.87
16
Discrete-space EOT/SB modelingHigh-Dimensional Gaussian Mixture (D=2, uniform, gamma=0.01) v1 (test)
Conditional Trend Score0.87
16
Discrete-space EOT/SB modelingHigh-Dimensional Gaussian Mixture D=16, gaussian, gamma=0.02 v1 (test)
Conditional Trend Score82
16
Discrete-space EOT/SB modelingHigh-Dimensional Gaussian Mixture D=16, gaussian, gamma=0.05 v1 (test)
Conditional Trend Score84
16
Discrete-space EOT/SB modelingHigh-Dimensional Gaussian Mixture (D=16, uniform, gamma=0.005) v1 (test)
Conditional Trend Score84
16
Discrete-space EOT/SB modelingHigh-Dimensional Gaussian Mixture D=16, uniform, gamma=0.01 v1 (test)
Conditional Trend Score84
16
Discrete-space EOT/SB modelingHigh-Dimensional Gaussian Mixture D=64, gaussian, gamma=0.02 v1 (test)
Conditional Trend Score85
16
Showing 10 of 25 rows

Other info

Follow for update