arXiv reaDer
物体検出の不確実性をモデル化するためのモンテカルロDropBlock
Monte Carlo DropBlock for Modelling Uncertainty in Object Detection
ディープラーニングの進歩により、オブジェクトの検出やセグメンテーションなどのコンピュータービジョンの問題により、パフォーマンスが大幅に向上しました。ただし、自動運転車などの多くの実際のアプリケーションでは、オブジェクトの誤った予測に関連するリスクが非常に高くなります。 YOLOモデルなどのオブジェクト検出用の標準的な深層学習モデルは、多くの場合、予測に自信がなく、分布外データの予測の不確実性を考慮していません。この作業では、モンテカルロDropBlock(MC-DropBlock)ベースの推論を使用して、オブジェクト検出およびセグメンテーションタスクの不確実性をモデル化するための効率的かつ効果的なアプローチを提案します。提案されたアプローチは、YOLOなどの深層学習モデルの畳み込み層にトレーニング時間とテスト時間の間にドロップブロックを適用します。これが、モデルの認識論的不確実性を捉えることができるベイズ畳み込みニューラルネットワークにつながることを示します。さらに、ガウス尤度を使用して偶然性の不確実性をキャプチャします。分布外実験を使用して、オブジェクト検出およびセグメンテーションタスクの不確実性をモデル化する上で提案されたアプローチの有効性を示します。実験結果は、MC-DropBlockが、オブジェクトの検出とセグメンテーションにおけるYOLOモデルの一般化、キャリブレーション、および不確実性モデリング機能を改善することを示しています。
With the advancements made in deep learning, computer vision problems like object detection and segmentation have seen a great improvement in performance. However, in many real-world applications such as autonomous driving vehicles, the risk associated with incorrect predictions of objects is very high. Standard deep learning models for object detection such as YOLO models are often overconfident in their predictions and do not take into account the uncertainty in predictions on out-of-distribution data. In this work, we propose an efficient and effective approach to model uncertainty in object detection and segmentation tasks using Monte-Carlo DropBlock (MC-DropBlock) based inference. The proposed approach applies drop-block during training time and test time on the convolutional layer of the deep learning models such as YOLO. We show that this leads to a Bayesian convolutional neural network capable of capturing the epistemic uncertainty in the model. Additionally, we capture the aleatoric uncertainty using a Gaussian likelihood. We demonstrate the effectiveness of the proposed approach on modeling uncertainty in object detection and segmentation tasks using out-of-distribution experiments. Experimental results show that MC-DropBlock improves the generalization, calibration, and uncertainty modeling capabilities of YOLO models in object detection and segmentation.
updated: Sun Aug 08 2021 11:34:37 GMT+0000 (UTC)
published: Sun Aug 08 2021 11:34:37 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アソシエイト