arXiv reaDer
教師なし点群登録のためのサンプリングネットワークガイド付きクロスエントロピー法
Sampling Network Guided Cross-Entropy Method for Unsupervised Point Cloud Registration
この論文では、点群登録タスクをマルコフ決定過程としてモデル化することにより、教師なし3D登録のためのクロスエントロピー法(CEM)を組み込んだエンドツーエンドのディープモデルを提案します。私たちのモデルは、サンプリングネットワークモジュールと微分可能なCEMモジュールで構成されています。サンプリングネットワークモジュールでは、点群のペアが与えられると、サンプリングネットワークは変換空間全体の事前サンプリング分布を学習します。学習したサンプリング分布は、微分可能なCEMモジュールの「適切な」初期化として使用できます。微分可能なCEMモジュールでは、最初に、点群登録タスクの報酬関数として、最大コンセンサス基準ベースのアライメントメトリックを提案します。次に、報酬関数に基づいて、状態ごとに、融合スコア関数を作成して、サンプリングされた変換を評価します。ここで、変換の現在および将来の報酬に重みを付けます。特に、サンプリングされた変換の将来の報酬は、変換された状態に対して反復最接近点(ICP)アルゴリズムを実行することによって取得されます。スコアが最も高い上位k個の変換を選択することにより、サンプリング分布を繰り返し更新します。さらに、CEMを微分可能にするために、sparsemax関数を使用してハードトップkの選択を置き換えます。最後に、Geman-McClure推定器ベースの損失を定式化して、エンドツーエンドの登録モデルをトレーニングします。広範な実験結果は、ベンチマークデータセットに対する私たちの方法の優れた登録パフォーマンスを示しています。
In this paper, by modeling the point cloud registration task as a Markov decision process, we propose an end-to-end deep model embedded with the cross-entropy method (CEM) for unsupervised 3D registration. Our model consists of a sampling network module and a differentiable CEM module. In our sampling network module, given a pair of point clouds, the sampling network learns a prior sampling distribution over the transformation space. The learned sampling distribution can be used as a "good" initialization of the differentiable CEM module. In our differentiable CEM module, we first propose a maximum consensus criterion based alignment metric as the reward function for the point cloud registration task. Based on the reward function, for each state, we then construct a fused score function to evaluate the sampled transformations, where we weight the current and future rewards of the transformations. Particularly, the future rewards of the sampled transforms are obtained by performing the iterative closest point (ICP) algorithm on the transformed state. By selecting the top-k transformations with the highest scores, we iteratively update the sampling distribution. Furthermore, in order to make the CEM differentiable, we use the sparsemax function to replace the hard top-k selection. Finally, we formulate a Geman-McClure estimator based loss to train our end-to-end registration model. Extensive experimental results demonstrate the good registration performance of our method on benchmark datasets.
updated: Tue Sep 14 2021 12:16:02 GMT+0000 (UTC)
published: Tue Sep 14 2021 12:16: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アソシエイト