コンピューター ビジョンの多くの困難な問題に対して高い予測精度を達成しているにもかかわらず、最近の研究では、ディープ ニューラル ネットワーク (DNN) は自信過剰な予測を行う傾向があり、予測が不十分になることが示唆されています。 DNN キャリブレーションを改善するための既存の試みのほとんどは分類タスクに限定されており、ドメイン内予測のキャリブレーションに限定されています。驚くべきことに、視覚ベースのセキュリティに敏感で安全性が重要なアプリケーションにおいて極めて重要な位置を占める物体検出方法の調整を研究する試みは、ほとんど、あるいはまったく行われていない。この論文では、最新の物体検出方法を校正するための新しいトレーニング時間手法を提案します。予測の不確実性を活用して、マルチクラスの信頼性とボックスの位置特定を共同で校正することができます。私たちは、いくつかのドメイン内およびドメイン外の検出ベンチマークについて広範な実験を実行しています。結果は、私たちが提案したトレーニング時キャリブレーション方法が、ドメイン内予測とドメイン外予測の両方でキャリブレーション誤差を低減する点で、いくつかのベースラインを常に上回っていることを示しています。コードとモデルは https://github.com/bimsarapathiraja/MCCL で入手できます。
Albeit achieving high predictive accuracy across many challenging computer vision problems, recent studies suggest that deep neural networks (DNNs) tend to make overconfident predictions, rendering them poorly calibrated. Most of the existing attempts for improving DNN calibration are limited to classification tasks and restricted to calibrating in-domain predictions. Surprisingly, very little to no attempts have been made in studying the calibration of object detection methods, which occupy a pivotal space in vision-based security-sensitive, and safety-critical applications. In this paper, we propose a new train-time technique for calibrating modern object detection methods. It is capable of jointly calibrating multiclass confidence and box localization by leveraging their predictive uncertainties. We perform extensive experiments on several in-domain and out-of-domain detection benchmarks. Results demonstrate that our proposed train-time calibration method consistently outperforms several baselines in reducing calibration error for both in-domain and out-of-domain predictions. Our code and models are available at https://github.com/bimsarapathiraja/MCCL.