arXiv reaDer
確率論的回転モデリングのための SO(3) 多様体上の離散正規化フローの詳細
Delving into Discrete Normalizing Flows on SO(3) Manifold for Probabilistic Rotation Modeling
正規化フロー (NF) は、基本分布の一連の追跡可能な変換によって表現力豊かな分布を構築し、基になるデータの確率モデルを形成するための強力なツールを提供します。回転は、コンピューター ビジョン、グラフィックス、およびロボット工学における重要な量であり、オクルージョンと対称性が発生したときに多くのあいまいさを示す可能性があるため、そのような確率モデルが必要になります。ユークリッド空間の NF については多くの進歩がありましたが、SO(3) 多様体用に調整された不連続または多対 1 マッピングのない効果的な正規化フローはありません。回転多様体のユニークな非ユークリッド特性を考えると、既存の NF を SO(3) 多様体に適応させることは自明ではありません。この論文では、メビウス変換ベースの結合層と四元数アフィン変換を組み合わせることにより、SO(3) の新しい正規化フローを提案します。提案された回転正規化フローを使用すると、SO(3) で任意の分布を効果的に表現できるだけでなく、入力観測を考慮してターゲット分布を条件付きで構築することもできます。広範な実験により、無条件タスクと条件付きタスクの両方で、回転正規化フローがベースラインよりも大幅に優れていることが示されています。
Normalizing flows (NFs) provide a powerful tool to construct an expressive distribution by a sequence of trackable transformations of a base distribution and form a probabilistic model of underlying data. Rotation, as an important quantity in computer vision, graphics, and robotics, can exhibit many ambiguities when occlusion and symmetry occur and thus demands such probabilistic models. Though much progress has been made for NFs in Euclidean space, there are no effective normalizing flows without discontinuity or many-to-one mapping tailored for SO(3) manifold. Given the unique non-Euclidean properties of the rotation manifold, adapting the existing NFs to SO(3) manifold is non-trivial. In this paper, we propose a novel normalizing flow on SO(3) by combining a Mobius transformation-based coupling layer and a quaternion affine transformation. With our proposed rotation normalizing flows, one can not only effectively express arbitrary distributions on SO(3), but also conditionally build the target distribution given input observations. Extensive experiments show that our rotation normalizing flows significantly outperform the baselines on both unconditional and conditional tasks.
updated: Sat Apr 08 2023 06:52:02 GMT+0000 (UTC)
published: Sat Apr 08 2023 06:52:02 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アソシエイト