arXiv reaDer
医学的検出のためのインスタンスレベルの不確実性の調査
Exploring Instance-Level Uncertainty for Medical Detection
不確実性を伴って予測する深層学習の能力は、臨床ルーチンでの採用の鍵として認識されています。さらに、経験的証拠に従って不確実性をモデル化することにより、パフォーマンスの向上が可能になりました。以前の研究では、セグメンテーションおよび分類タスクにおける不確実性の推定について広く議論されてきましたが、主にバウンディングボックスの位置合わせの課題のため、バウンディングボックスベースの検出への適用は制限されていました。この作業では、2つの異なるバウンディングボックスレベル(またはインスタンスレベル)の不確実性推定値、つまり予測分散とモンテカルロ(MC)サンプル分散を使用して2.5D検出CNNを拡張することを検討します。実験は、LUNA16データセットでの肺結節の検出のために実施されます。これは、結節と非結節の間に重大な意味のあいまいさが存在する可能性があるタスクです。結果は、私たちの方法が両方のタイプの分散の組み合わせを利用することにより、評価スコアを84.57%から88.86%に改善することを示しています。さらに、生成された不確実性により、確率しきい値のみを使用する場合と比較して優れた動作点が可能になり、パフォーマンスを89.52%にさらに高めることができることを示します。結節の検出例を視覚化して、この方法の利点をさらに説明します。
The ability of deep learning to predict with uncertainty is recognized as key for its adoption in clinical routines. Moreover, performance gain has been enabled by modelling uncertainty according to empirical evidence. While previous work has widely discussed the uncertainty estimation in segmentation and classification tasks, its application on bounding-box-based detection has been limited, mainly due to the challenge of bounding box aligning. In this work, we explore to augment a 2.5D detection CNN with two different bounding-box-level (or instance-level) uncertainty estimates, i.e., predictive variance and Monte Carlo (MC) sample variance. Experiments are conducted for lung nodule detection on LUNA16 dataset, a task where significant semantic ambiguities can exist between nodules and non-nodules. Results show that our method improves the evaluating score from 84.57% to 88.86% by utilizing a combination of both types of variances. Moreover, we show the generated uncertainty enables superior operating points compared to using the probability threshold only, and can further boost the performance to 89.52%. Example nodule detections are visualized to further illustrate the advantages of our method.
updated: Sun Feb 07 2021 16:06:55 GMT+0000 (UTC)
published: Wed Dec 23 2020 18:46: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アソシエイト