arXiv reaDer
事前にトレーニングされたnnU-NetモデルがCovid-19肺病変のセグメンテーションでサイレントに失敗した場合の検出
Detecting when pre-trained nnU-Net models fail silently for Covid-19 lung lesion segmentation
コンピュータ断層撮影における肺病変の自動セグメンテーションは、Covid-19パンデミック時の臨床医の負担を軽減する可能性があります。しかし、予測深層学習モデルは、分布外(OOD)データで黙って失敗するため、臨床ルーチンでは信頼されていません。特徴空間におけるマハラノビス距離を利用した軽量OOD検出法を提案します。提案されたアプローチは、モデルアーキテクチャやトレーニング手順を変更することなく、最先端のセグメンテーションパイプラインにシームレスに統合できるため、事前にトレーニングされたモデルの新しいデータへの適合性を評価するために使用できます。複数の機関のデータセットでトレーニングされたパッチベースのnnU-Netアーキテクチャを使用してメソッドを検証し、モデルが誤ってセグメント化したサンプルを効果的に検出することを確認します。
Automatic segmentation of lung lesions in computer tomography has the potential to ease the burden of clinicians during the Covid-19 pandemic. Yet predictive deep learning models are not trusted in the clinical routine due to failing silently in out-of-distribution (OOD) data. We propose a lightweight OOD detection method that exploits the Mahalanobis distance in the feature space. The proposed approach can be seamlessly integrated into state-of-the-art segmentation pipelines without requiring changes in model architecture or training procedure, and can therefore be used to assess the suitability of pre-trained models to new data. We validate our method with a patch-based nnU-Net architecture trained with a multi-institutional dataset and find that it effectively detects samples that the model segments incorrectly.
updated: Wed Jul 14 2021 11:45:47 GMT+0000 (UTC)
published: Tue Jul 13 2021 10:48:08 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アソシエイト