arXiv reaDer
OT-Net: 再利用可能なニューラル最適輸送ソルバー
OT-Net: A Reusable Neural Optimal Transport Solver
最適トランスポート(OT)の普及に伴い、その計算は必須となり、さまざまなアルゴリズムが登場しています。しかし、既存の手法は効率が低いか、不連続なマップを表現できません。したがって、新しい再利用可能なニューラル OT ソルバー OT-Net が提示されます。これは、最初にニューラル ネットワークを介してブレニエの高さ表現を学習してそのポテンシャルを取得し、次にポテンシャルの勾配を計算することによって OT マップを取得します。このアルゴリズムには 2 つのメリットがあります。1) 不連続なマップを簡単に表現できるため、不連続なサポートを持つターゲット分布と一致し、明確な境界を実現できます。これにより、生成されたモデルのモード崩壊を十分に排除できます。 2) OT マップは、新しいターゲット サンプルが追加されたときに提案アルゴリズムによって直接計算でき、マップの効率と再利用性が大幅に向上します。さらに、アルゴリズムの理論上の誤差限界が分析され、画像生成、色転送、およびドメイン適応におけるアプローチの実験的成功を実証しました。
With the widespread application of optimal transport (OT), its calculation becomes essential, and various algorithms have emerged. However, the existing methods either have low efficiency or cannot represent discontinuous maps. A novel reusable neural OT solver OT-Net is thus presented, which first learns Brenier's height representation via the neural network to obtain its potential, and then gained the OT map by computing the gradient of the potential. The algorithm has two merits, 1) it can easily represent discontinuous maps, which allows it to match any target distribution with discontinuous supports and achieve sharp boundaries. This can well eliminate mode collapse in the generated models. 2) The OT map can be calculated straightly by the proposed algorithm when new target samples are added, which greatly improves the efficiency and reusability of the map. Moreover, the theoretical error bound of the algorithm is analyzed, and we have demonstrated the empirical success of our approach in image generation, color transfer, and domain adaptation.
updated: Wed Jun 14 2023 04:11:38 GMT+0000 (UTC)
published: Wed Jun 14 2023 04:11:38 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アソシエイト