ИССЛЕДОВАНИЕ · RESEARCH · #1512
FlashSinkhorn 2: блочно-разрежённый GPU-решатель для больших задач энтропийного оптимального транспорта
FlashSinkhorn 2 (FS2) — двухэтапный GPU‑решатель для энтропийного оптимального транспорта с квадратовой евклидовой метрикой на низкоразмерных точечных облаках: грубая стадия по центроидам сочетается с блочно‑разрежённой тонкой стадией (Morton‑порядок, скрининг и объединённое исполнение на тензор‑ядрах). Авторы утверждают, что FS2 достигает целевых маргинальных невязок на бенчмарках и на одном NVIDIA A100 решил задачу EOT между двумя распределениями по 1.34×10^8 частиц из космологической N‑body симуляции до маргинальной невязки <0.01 менее чем за 2.5 часа; исходный код выложен в репозитории GitHub.
КЛЮЧЕВЫЕ ТЕЗИСЫ
- FlashSinkhorn 2 (FS2) — двухэтапный GPU‑решатель для энтропийного оптимального транспорта с квадратовой евклидовой метрикой на низкоразмерных точечных облаках: грубая стадия по центроидам сочетается с блочно‑разрежённой тонкой стадией (Morton‑порядок, скрининг и объединённое исполнение на тензор‑ядрах).
- Авторы утверждают, что FS2 достигает целевых маргинальных невязок на бенчмарках и на одном NVIDIA A100 решил задачу EOT между двумя распределениями по 1.34×10^8 частиц из космологической N‑body симуляции до маргинальной невязки <0.01 менее чем за 2.5 часа; исходный код выложен в репозитории GitHub.
- FS2 существенно увеличивает масштаб дискретного энтропийного ОТ, решаемого на одном GPU, обеспечивая быструю и память‑эффективную обработку больших научных и ML данных.
ПОЧЕМУ ЭТО ВАЖНО
FS2 существенно увеличивает масштаб дискретного энтропийного ОТ, решаемого на одном GPU, обеспечивая быструю и память‑эффективную обработку больших научных и ML данных.