arXiv reaDer
3DIoUMatch:半教師あり3Dオブジェクト検出のためのIoU予測の活用
3DIoUMatch: Leveraging IoU Prediction for Semi-Supervised 3D Object Detection
3Dオブジェクト検出は、取得が困難な3D注釈に大きく依存する、重要でありながら要求の厳しいタスクです。必要な監視量を減らすために、半教師あり3Dオブジェクト検出の新しい方法である3DIoUMatchを提案します。人気のある点群ベースのオブジェクト検出器であるVoteNetをバックボーンとして採用し、教師と生徒の相互学習フレームワークを活用して、疑似ラベルの形式でラベル付きからラベルなしの列車セットに情報を伝達します。ただし、タスクの複雑さが高いため、疑似ラベルには重大なノイズが発生し、直接使用できないことがわかります。そのために、信頼性ベースのフィルタリングメカニズムを導入します。私たちのアプローチの鍵は、新しい微分可能な3DIoU推定モジュールです。このモジュールは、ローカライズが不十分なプロポーザルのフィルタリングや、IoUガイドによるバウンディングボックスの重複排除に使用されます。推論時に、このモジュールは、テスト時間の最適化を通じてローカリゼーションを改善するためにさらに利用されます。私たちの方法は、ScanNetとSUN-RGBDの両方のベンチマークで最先端の方法を一貫して大幅に改善しています。たとえば、ScanNetで10%のラベル付きデータのみを使用してトレーニングする場合、3DIoUMatchは、従来技術に比べてmAP @ 0.25で7.7の絶対的な改善を達成し、mAP @ 0.5で8.5の絶対的な改善を達成します。
3D object detection is an important yet demanding task that heavily relies on difficult to obtain 3D annotations. To reduce the required amount of supervision, we propose 3DIoUMatch, a novel method for semi-supervised 3D object detection. We adopt VoteNet, a popular point cloud-based object detector, as our backbone and leverage a teacher-student mutual learning framework to propagate information from the labeled to the unlabeled train set in the form of pseudo-labels. However, due to the high task complexity, we observe that the pseudo-labels suffer from significant noise and are thus not directly usable. To that end, we introduce a confidence-based filtering mechanism. The key to our approach is a novel differentiable 3D IoU estimation module. This module is used for filtering poorly localized proposals as well as for IoU-guided bounding box deduplication. At inference time, this module is further utilized to improve localization through test-time optimization. Our method consistently improves state-of-the-art methods on both ScanNet and SUN-RGBD benchmarks by significant margins. For example, when training using only 10% labeled data on ScanNet, 3DIoUMatch achieves 7.7 absolute improvement on mAP@0.25 and 8.5 absolute improvement on mAP@0.5 upon the prior art.
updated: Tue Dec 08 2020 11:06:26 GMT+0000 (UTC)
published: Tue Dec 08 2020 11:06:26 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アソシエイト