深層学習を伴う医用画像における予測の不確実性を考慮することが最も重要です。モンテカルロドロップアウトを使用した変分ベイズ推定による偶然性と認識論の両方の不確実性の推定を回帰タスクに適用し、予測の不確実性が体系的に過小評価されていることを示します。単一のスカラー値でσスケーリングを適用します。両方のタイプの不確かさに対するシンプルでありながら効果的な校正方法。私たちのアプローチのパフォーマンスは、さまざまな最先端の畳み込みネットワークアーキテクチャを使用して、さまざまな一般的な医療回帰データセットで評価されます。私たちの実験では、σスケーリングは予測の不確実性を確実に再較正することができます。実装が簡単で、精度を維持できます。回帰における適切に調整された不確実性により、信頼性の低い予測の確実な拒否または分布外のサンプルの検出が可能になります。ソースコードはhttps://github.com/mlaves/well-calibrated-regression-uncertaintyで入手できます。
The consideration of predictive uncertainty in medical imaging with deep learning is of utmost importance. We apply estimation of both aleatoric and epistemic uncertainty by variational Bayesian inference with Monte Carlo dropout to regression tasks and show that predictive uncertainty is systematically underestimated. We apply σ scaling with a single scalar value; a simple, yet effective calibration method for both types of uncertainty. The performance of our approach is evaluated on a variety of common medical regression data sets using different state-of-the-art convolutional network architectures. In our experiments, σ scaling is able to reliably recalibrate predictive uncertainty. It is easy to implement and maintains the accuracy. Well-calibrated uncertainty in regression allows robust rejection of unreliable predictions or detection of out-of-distribution samples. Our source code is available at https://github.com/mlaves/well-calibrated-regression-uncertainty