3D変換グループSO(3)、SE(3)、およびSim(3)を含む計算グラフのバックプロパゲーションを実行する問題に対処します。 3D変換グループは、3Dビジョンやロボット工学で広く使用されていますが、ベクトル空間を形成せず、代わりに滑らかな多様体上にあります。ユークリッド空間に3D変換を埋め込む標準的なバックプロパゲーションアプローチには、数値的な問題があります。 3D変換のグループ構造を活用し、多様体の接空間でバックプロパゲーションを実行する新しいライブラリを紹介します。私たちのアプローチは、数値的に安定していて、実装が簡単で、さまざまなタスクのセットに有益であることを示しています。プラグアンドプレイのPyTorchライブラリは、https://github.com/princeton-vl/lietorchで入手できます。
We address the problem of performing backpropagation for computation graphs involving 3D transformation groups SO(3), SE(3), and Sim(3). 3D transformation groups are widely used in 3D vision and robotics, but they do not form vector spaces and instead lie on smooth manifolds. The standard backpropagation approach, which embeds 3D transformations in Euclidean spaces, suffers from numerical difficulties. We introduce a new library, which exploits the group structure of 3D transformations and performs backpropagation in the tangent spaces of manifolds. We show that our approach is numerically more stable, easier to implement, and beneficial to a diverse set of tasks. Our plug-and-play PyTorch library is available at https://github.com/princeton-vl/lietorch.