PV-RCNN: 3D物体検出のためのポイント・ボクセル特徴セットアブストラクション
1分まとめ
ボクセルCNNの効率的な学習とPointNetベースのセットアブストラクションの柔軟な受容野を統合した2段階検出器を提案し、KITTIテストセットで従来のLiDARのみの手法を1.72ポイント(Moderate)上回るなど、当時の最高性能(SOTA)を達成した。
- 3DボクセルCNNとPointNetベースのセットアブストラクションを、キーポイントを介して2段階で統合する3D物体検出器を提案した。
- KITTIテストセットのCar 3D Detection(Moderate、R40)で81.43を達成し、従来のLiDARのみの手法(STD)を1.72ポイント上回った。
- Waymo Open Dataset(バージョン1.0)のVehicle LEVEL 1 3D mAPで70.30を達成し、従来手法(MVF)の62.93を7.37ポイント上回った。
- 評価はKITTIとWaymoに限定され、Waymoの一部の比較対象は著者らによる再現実装の結果を含む。当時の計算コストは高く、その後の改良によって性能はさらに向上している。
課題
グリッドベース(ボクセルCNN)は計算効率が良い反面、空間解像度が低下して正確な位置決めに課題があった。ポイントベース(PointNet)は柔軟な受容野を持つが計算コストが高かった。また、従来のRoIプーリングは、スパースで低解像度のボクセル特徴を扱うため、ゼロが多く情報量の少ない特徴しか抽出できなかった。
提案手法
3Dスパース畳み込みを用いたボクセルCNNをバックボーンに高品質な3Dプロポーザルを生成する。次に、FPSでサンプリングした少数のキーポイント(KITTIで2048個、Waymoで4096個)に対して、Voxel Set Abstraction(VSA)モジュールによりマルチスケールのボクセル特徴を集約する。生の点群特徴とBEV特徴も統合し、予測キーポイント重み付け(PKW)モジュールで前景キーポイントを重み付ける。提案リファインメントでは、各プロポーザルに6×6×6のグリッドポイントを設定し、RoIグリッドプーリングにより複数半径(0.8m、1.6m)でキーポイント特徴を集約する。IoUガイドによる信頼度予測とボックスリファインメントを2層MLPで行う。
評価と結果
KITTIテストセット(Car 3D、R40)ではEasy 90.25、Moderate 81.43、Hard 76.82を達成。Moderateは従来のLiDARのみの手法(STD)を1.72上回った。KITTI valセット(Car、Moderate、R11)では83.90。Waymo Open Dataset(v1.0)のVehicle LEVEL 1 3D mAPは70.30(従来手法のMVFは62.93)。アブレーションでは、RoI-gridモジュールをRoI-aware poolingに置き換えるとModerateが84.83から82.97に低下、PKWモジュールを削除すると82.95に低下した。
KITTIテストセットにおけるCar 3D Detectionの比較(R40)
| Method | Easy | Moderate | Hard |
|---|---|---|---|
| SECOND | 83.34 | 72.55 | 65.82 |
| PointRCNN | 86.96 | 75.64 | 70.70 |
| STD | 87.95 | 79.71 | 75.09 |
| PV-RCNN | 90.25 | 81.43 | 76.82 |
KITTIテストセットのCar 3D Detection(Moderate、R40)
Waymo Open Dataset(バージョン1.0)のVehicle LEVEL 1 3D mAP
アブレーション実験(KITTI val、Car、Moderate、R40)
新規性
ボクセルCNNとPointNetベースのセットアブストラクションを、サンプリングしたキーポイントを媒介として2段階で統合し、効率的な学習と柔軟な受容野を両立した点。
限界
著者による明記はないが、以下の制約が読み取れる。評価はKITTIとWaymoに限定されており、データセットのバイアスは未知数である。Waymoの比較対象(PointPillars、MVF)は一部、著者らによる再現実装の結果であり、公式リーダーボードの値とは異なる可能性がある。学習には多数のGPU(KITTIで8台、Waymoで32台)を使用しており、当時の計算コストは高い。また、その後の研究により、ボクセルエンコーダーの改良やTransformerベースの手法、高度なデータ拡張によって、これらのデータセットでの性能はさらに向上している。
産業へのインパクト
この論文は、ボクセルベースとポイントベースの特徴学習を統合するという、その後の3D物体検出の標準的な設計パターンを確立した。当時は「グリッドかポイントか」の二択だったが、両者の利点を活かす道を示した。KITTIとWaymoは現在も標準的なベンチマークであり、指標(mAP、mAPH)も現在まで引き継がれている。この研究を基点に、その後の多くの手法がボクセルとポイントのハイブリッド構造を採用した。