search

JuliaOptimalTransport / OptimalTransport.jl

Optimal transport algorithms for Julia
https://juliaoptimaltransport.github.io/OptimalTransport.jl/dev
MIT License
93 stars 8 forks source link

Greenkhorn and screenkhorn - near-linear time sinkhorn #151

Open flyaflya opened 2 years ago

flyaflya commented 2 years ago

Thx for the great package, I have just been toying with it. I noticed there are some newer near-linear time sinkhorn-based algorithms (Greenkhorn and screenkhorn) that are available in POT. The link to the Greenkhorn paper is below.

Just curious if these are on the roadmap for OptimalTransport.jl?

https://arxiv.org/pdf/1705.09634.pdf

davibarreira commented 2 years ago

Awesome! Just became. I'll take a look. Thanks!

davibarreira commented 2 years ago

Curious that these algorithms are not in the Computational Optimal Transport book,

zsteve commented 2 years ago

Thx for the great package, I have just been toying with it. I noticed there are some newer near-linear time sinkhorn-based algorithms (Greenkhorn and screenkhorn) that are available in POT. The link to the Greenkhorn paper is below.

Just curious if these are on the roadmap for OptimalTransport.jl?

https://arxiv.org/pdf/1705.09634.pdf

@flyaflya glad to hear this package has been helpful! I don't think there's any reason why this should not be a candidate algorithm for inclusion. (Basically it will depend on the amount of free time people have xD). However, if you implement these algorithms as part of your own research, please feel more than welcome to update this issue or make a pull request! If you have existing code, am more than happy to help adapt it to fit into the package.

davibarreira commented 2 years ago

@flyaflya #159