arXiv reaDer
RAMA:GPUでの高速マルチカットアルゴリズム
RAMA: A Rapid Multicut Algorithm on GPU
機械学習やコンピュータービジョンで広く使用されている古典的なグラフクラスタリング問題であるマルチカット(別名相関クラスタリング)問題に対して、高度に並列なプライマルデュアルアルゴリズムを提案します。私たちのアルゴリズムは、再帰的に実行される3つのステップで構成されます。コストおよび(3)行列-行列の乗算により、コストが大幅に削減された収縮エッジ。私たちのアルゴリズムは、最適までの距離を推定する一次解と下限を生成します。アルゴリズムをGPUに実装し、CPUで実行される従来のシーケンシャルアルゴリズムと比較して、ソリューションの品質を犠牲にすることなく、実行速度が1桁から2桁向上することを示しています。プライマルとデュアルのギャップが小さい場合、数秒で最大O(10 ^ 8)変数の非常に大規模なベンチマーク問題を解決できます。私たちのコードはhttps://github.com/pawelswoboda/RAMAで入手できます。
We propose a highly parallel primal-dual algorithm for the multicut (a.k.a. correlation clustering) problem, a classical graph clustering problem widely used in machine learning and computer vision. Our algorithm consists of three steps executed recursively: (1) Finding conflicted cycles that correspond to violated inequalities of the underlying multicut relaxation, (2) Performing message passing between the edges and cycles to optimize the Lagrange relaxation coming from the found violated cycles producing reduced costs and (3) Contracting edges with high reduced costs through matrix-matrix multiplications. Our algorithm produces primal solutions and lower bounds that estimate the distance to optimum. We implement our algorithm on GPUs and show resulting one to two orders-of-magnitudes improvements in execution speed without sacrificing solution quality compared to traditional sequential algorithms that run on CPUs. We can solve very large scale benchmark problems with up to O(10^8) variables in a few seconds with small primal-dual gaps. Our code is available at https://github.com/pawelswoboda/RAMA.
updated: Fri Mar 11 2022 15:28:06 GMT+0000 (UTC)
published: Sat Sep 04 2021 10:33:59 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト