arXiv reaDer
ドメインシフト下での物体検出におけるキャリブレーションの改善に向けて
Towards Improving Calibration in Object Detection Under Domain Shift
セーフティ クリティカルなアプリケーションでのディープ ニューラル ネットワークの使用が増加しているため、トレーニング済みのモデルを十分に調整する必要があります。現在のほとんどのキャリブレーション手法は、ドメイン内予測のキャリブレーションの改善に焦点を当てながら、分類の問題に対処しています。多くの意思決定システムで同様のスペースと重要性を占める視覚オブジェクト検出器のキャリブレーションへの対処には、ほとんどまたはまったく注意が払われていません。この論文では、特にドメインシフトの下で、現在のオブジェクト検出モデルのキャリブレーションを研究します。この目的のために、最初にオブジェクト検出用のプラグアンドプレイのトレインタイム キャリブレーション ロスを導入します。検出器のキャリブレーションを改善するための補助損失関数として使用できます。第二に、一般的に使用される自己訓練ベースのドメイン適応検出器を暗黙的に調整できる、オブジェクト検出のための新しい不確実性定量化メカニズムを考案します。私たちの研究には、1 段階と 2 段階の物体検出器の両方が含まれています。私たちの損失が、ドメイン内とドメイン外の両方の検出のキャリブレーションを顕著なマージンで改善することを示しています。最後に、さまざまなドメイン シフト シナリオでドメイン適応オブジェクト検出器を調整する際の手法の有用性を示します。
The increasing use of deep neural networks in safety-critical applications requires the trained models to be well-calibrated. Most current calibration techniques address classification problems while focusing on improving calibration on in-domain predictions. Little to no attention is paid towards addressing calibration of visual object detectors which occupy similar space and importance in many decision making systems. In this paper, we study the calibration of current object detection models, particularly under domain shift. To this end, we first introduce a plug-and-play train-time calibration loss for object detection. It can be used as an auxiliary loss function to improve detector's calibration. Second, we devise a new uncertainty quantification mechanism for object detection which can implicitly calibrate the commonly used self-training based domain adaptive detectors. We include in our study both single-stage and two-stage object detectors. We demonstrate that our loss improves calibration for both in-domain and out-of-domain detections with notable margins. Finally, we show the utility of our techniques in calibrating the domain adaptive object detectors in diverse domain shift scenarios.
updated: Thu Sep 15 2022 20:32:28 GMT+0000 (UTC)
published: Thu Sep 15 2022 20:32:28 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アソシエイト