arXiv reaDer
EvCenterNet: 証拠学習を使用したオブジェクト検出の不確実性推定
EvCenterNet: Uncertainty Estimation for Object Detection using Evidential Learning
不確実性の推定は、高レベルの意思決定や経路計画など、いくつかのダウンストリーム タスクに貴重な情報を提供するため、自動運転などの安全性が重要な設定では非常に重要です。この作業では、証拠学習を利用して分類と回帰の両方の不確実性を直接推定する、新しい不確実性認識 2D オブジェクト検出フレームワークである EvCenterNet を提案します。オブジェクト検出に証拠学習を採用するために、スパースヒートマップ入力用の証拠と焦点損失関数の組み合わせを考案します。証拠学習によって発生したクラスの不均衡に対処するために、回帰とヒートマップ予測にクラスバランスの重み付けを導入します。さらに、予測されたヒートマップの不確実性を積極的に利用して、最も不確実な点に焦点を当てることで検出性能を向上させる学習スキームを提案します。 KITTI データセットでモデルをトレーニングし、BDD100K や nuImages などの困難な分散データセットで評価します。私たちの実験は、私たちのアプローチが精度を向上させ、基本モデルに関連する実行時間の損失を最小限に抑えることを示しています。
Uncertainty estimation is crucial in safety-critical settings such as automated driving as it provides valuable information for several downstream tasks including high-level decision-making and path planning. In this work, we propose EvCenterNet, a novel uncertainty-aware 2D object detection framework utilizing evidential learning to directly estimate both classification and regression uncertainties. To employ evidential learning for object detection, we devise a combination of evidential and focal loss functions for the sparse heatmap inputs. We introduce class-balanced weighting for regression and heatmap prediction to tackle the class imbalance encountered by evidential learning. Moreover, we propose a learning scheme to actively utilize the predicted heatmap uncertainties to improve the detection performance by focusing on the most uncertain points. We train our model on the KITTI dataset and evaluate it on challenging out-of-distribution datasets including BDD100K and nuImages. Our experiments demonstrate that our approach improves the precision and minimizes the execution time loss in relation to the base model.
updated: Mon Mar 06 2023 11:07:11 GMT+0000 (UTC)
published: Mon Mar 06 2023 11:07:11 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アソシエイト