ott-jax/ott

何を解決するか

アクセラレータ対応のスケーラブルなライブラリを提供し、最適輸送（OT）問題を効率的に解決できます。これにより、コスト関数に基づいて2つの点群または点の集合を効率的にマッチングでき、ハンガリアンアルゴリズムなどの従来のアルゴリズムの $O(n^3)$ 複雑さを克服できます。

動作方法

JAXに基づいて構築されており、特にSinkhornアルゴリズムを含むさまざまな離散ソルバーを実装しています。このアルゴリズムはスケジューリング、モーメンタム、低ランク拡張によって最適化されています。また、ソース測度とターゲット測度間の最適輸送マップを近似するニューラルネットワークアプローチもサポートしています。基本的なマッチングに加え、Gromov-WassersteinやWassersteinバリアンツを扱い、多様な設定で点群を比較できます。

対象ユーザー

点群マッチング、数学的最適化、データサイエンスに取り組む研究者や開発者で、GPUやTPU上で高性能な計算を必要とする方々。

特徴

• アクセラレータ上で高性能実行を可能にするJAX駆動。
• 複数の高速化テクニックを備えたSinkhornアルゴリズムの実装。
• 输送マップのニューラルネットワークベースの近似をサポート。
• Gromov-WassersteinおよびWassersteinバリアンツの処理が可能。
• カスタムコスト関数や目的関数の柔軟な定義が可能。