論文の概要: FlashSinkhorn 2: Block-Sparse Entropic Optimal Transport
- arxiv url: http://arxiv.org/abs/2610.02395v1
- Date: Thu, 01 Oct 2026 19:22:49 GMT
- ステータス: 翻訳完了
- システム内更新日: 2026-10-06 00:14:30.056566
- Title: FlashSinkhorn 2: Block-Sparse Entropic Optimal Transport
- Title(参考訳): FlashSinkhorn 2: ブロック・スパース・エントロピー・オプティカルトランスポート
- Abstract要約: FlashSinkhornは、低次元の点雲上の2乗ユークリッドコストの解法である。
2つのステージを結合することで1つのGPU上の所定の限界残差に対して大きな離散EOT問題の解法である textbfFlashSinkhorn2 (FS2) を提示する。
- 参考スコア(独自算出の注目度): 2.561864336970246
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Streaming GPU solvers for entropic optimal transport (EOT), such as FlashSinkhorn, avoid storing the dense kernel but still evaluate all $n\times m$ point pairs in every Sinkhorn iteration. We present \textbf{FlashSinkhorn~2} (FS2), a solver for squared-Euclidean cost on low-dimensional point clouds that solves large discrete EOT problems to a prescribed marginal residual on a single GPU by coupling two stages. A coarse stage solves on cell centroids, lifts the potentials to every point and, when a sampled marginal check rejects the lift, continues on the centroids, replacing most point-level updates. A block-sparse fine stage then removes the centroid error that coarse updates cannot. Its Morton-ordered blocks support screening and fused tensor-core execution, and a threshold set by the block masses bounds each omitted tile's contribution to every row and column. On synthetic benchmarks, FS2 reaches the target residual on all 32 problems and GeomLoss multiscale on 10. On one A100, FS2 solves discrete EOT between two $1.34\times10^8$-particle measures from a cosmological $N$-body simulation, at an entropic blur equal to the mean interparticle distance, to an all-particle marginal residual below 0.01 in under 2.5 hours. To our knowledge, it is the largest discrete EOT problem solved to this accuracy within hours. For reproducibility, we release an open-source implementation at https://github.com/ot-triton-lab/flash-sinkhorn
- Abstract(参考訳): FlashSinkhornのようなエントロピック最適トランスポート(EOT)のためのGPUソルバのストリーミングは、高密度カーネルの保存は避けるが、Sinkhornのイテレーション毎にすべての$n\times m$pointペアを評価する。
本稿では,低次元点雲上での2乗ユークリッドコストの解法である‘textbf{FlashSinkhorn~2} (FS2) について述べる。
粗いステージは、セルセントロイドを解き、電位を各点に引き上げ、サンプル化された限界チェックがリフトを拒絶すると、セントロイドを継続し、ほとんどのポイントレベルの更新を置き換える。
ブロックスパースなファインステージは、粗い更新ができないセントロイドエラーを取り除く。
モートン順序ブロックはスクリーニングと融合テンソルコアの実行をサポートし、ブロック質量によって設定された閾値は、各省略タイルの行と列への寄与を束縛する。
合成ベンチマークでは、FS2は32の全ての問題でターゲット残差に達し、GeomLossは10でマルチスケールとなった。
1つのA100では、FS2は1.34\times10^8$- Particleの2つの測度を、平均粒子間距離と等しいエントロピックなぼかしで2.5時間以内に0.01以下の全粒子辺縁残差で解いた。
我々の知る限り、これは数時間以内にこの精度で解決された最大の離散EOT問題である。
再現性のために、https://github.com/ot-triton-lab/flash-sinkhornでオープンソース実装をリリースします。
関連論文リスト
- Spectral Anatomy of Quantum Gaussian Process Kernels [38.264196157340216]
我々は,Nystrm近似誤差に束縛されたコーシー=シュワルツテール,有限サンプル分散抽出等式,およびエンフターゲット依存の最適エントロピーのキャラクタリゼーションを証明した。
診断はカーネルに依存しない: ハードウェア効率、マッチゲート、IQPのインハンドRBF/Matérn/RFF/deep-カーネルファミリはすべて同一の$S/log n$曲線に崩壊する。
論文 参考訳(メタデータ) (2026-05-29T07:41:14Z) - Hierarchical Transformer Preconditioning for Interactive Physics Simulation [28.137076331332413]
階層型トランスフォーマープレコンディショナー(Hierarchical Transformer Preconditioner)は、弱い許容率のH行列分割に固定されたニューラルプレコンディショナーである。
ネットワークは低ランクの遠距離因子を通して逆をモデル化する。
高速道路の接続を利用して、奥行きを隔ててコンテキストを伝播する。
論文 参考訳(メタデータ) (2026-05-13T11:02:27Z) - The Wristband Gaussian Loss: Deterministic, Composable Latents via a Sphere-Interval Decomposition [1.0195618602298684]
リストバンドガウス損失(英: Wristband Gaussian Loss)は、サンプリング、KL項、反復輸送のないガウス化点埋め込みに対する決定論的バッチ損失である。
我々は、この反射カーネルの目的を、2つの方法で計算する:$O(N2 d)$で最寄りの3イメージのペアランケーションと、角およびラジアルマーサーモードに結合するスペクトルノイマン経路である。
1Dワッサーシュタインの半径項とモーメントペナルティは同じ最適値を持つ有限サンプル加速器として機能し、モンテカルロ$はこれらの成分を単一の標準統計量に変換する。
論文 参考訳(メタデータ) (2026-05-09T07:25:37Z) - Near-Optimal Clustering in Mixture of Markov Chains [74.3828414695655]
我々は、長さ$H$の軌跡を、大きさ$S$の有限状態空間上の未知のエルゴードマルコフ鎖の1つによって生成される、$T$ trajectories of length $H$の問題を研究する。
我々は、連鎖の遷移核間の重み付きKL分散によって支配されるクラスタリングエラー率に基づいて、インスタンス依存で高い確率の低い境界を導出する。
次に,新しい2段階クラスタリングアルゴリズムを提案する。
論文 参考訳(メタデータ) (2025-06-02T05:10:40Z) - Sharper Convergence Guarantees for Asynchronous SGD for Distributed and
Federated Learning [77.22019100456595]
通信周波数の異なる分散計算作業者のトレーニングアルゴリズムを示す。
本研究では,より厳密な収束率を$mathcalO!!(sigma2-2_avg!)とする。
また,不均一性の項は,作業者の平均遅延によっても影響されることを示した。
論文 参考訳(メタデータ) (2022-06-16T17:10:57Z) - Accelerated Gradient Tracking over Time-varying Graphs for Decentralized Optimization [59.65871549878937]
実用的な単一ループ加速勾配追跡には$O(fracgamma1-sigma_gamma)2sqrtfracLepsilon)$が必要であることを証明している。
我々の収束率は$O(frac1epsilon5/7)$と$O(fracLmu)5/7frac1(1-sigma)1.5logfrac1epsilon)$よりも大幅に改善した。
論文 参考訳(メタデータ) (2021-04-06T15:34:14Z) - Differentially Quantized Gradient Methods [53.3186247068836]
微分量子化グラディエントDescence (DQ-GD) が$maxsigma_mathrmGD, rhon 2-R$の線形収縮係数を得ることを示す。
あるクラス内のアルゴリズムは$maxsigma_mathrmGD, 2-R$よりも早く収束できない。
論文 参考訳(メタデータ) (2020-02-06T20:40:53Z)
関連論文リストは本サイト内にある論文のタイトル・アブストラクトから自動的に作成しています。
指定された論文の情報です。
本サイトの運営者は本サイト(すべての情報・翻訳含む)の品質を保証せず、本サイト(すべての情報・翻訳含む)を使用して発生したあらゆる結果について一切の責任を負いません。