New CUDA implementation speeds up optimal transport calculations on GPUs

Researchers have developed FastSinkhorn, a new CUDA implementation for the Sinkhorn algorithm used in optimal transport computations. This method operates entirely in the log-domain, ensuring numerical stability even with very small regularization parameters where other methods fail. Benchmarks show FastSinkhorn achieves significant speedups over existing libraries like POT and PyTorch, while using minimal GPU memory. AI

影响 This optimized implementation could accelerate various machine learning tasks that rely on optimal transport, such as image and point cloud processing.

排序理由 The cluster contains a new academic paper detailing a novel algorithm and its implementation for optimal transport. [lever_c_demoted from research: ic=1 ai=1.0]

在 arXiv cs.LG 阅读 →

• CUDA
• FastSinkhorn
• GPU
• optimal transport
• PyTorch
• Sinkhorn algorithm

• 论文
• 基础设施

AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →