arXiv reaDer
レイヤーアンサンブル:セグメンテーションの深層学習におけるシングルパスの不確実性推定
Layer Ensembles: A Single-Pass Uncertainty Estimation in Deep Learning for Segmentation
深層学習における不確実性の推定は、臨床現場でAIアルゴリズムを安全に利用する必要があるため、医療画像分析の主要な研究分野になっています。不確実性推定のほとんどのアプローチでは、複数のネットワークのテストまたはトレーニング中にネットワークの重みを複数回サンプリングする必要があります。これにより、時間と計算リソースの面でトレーニングとテストのコストが高くなります。本論文では、単一のネットワークを使用し、ネットワークの予測不確実性を推定するために単一のパスのみを必要とする新しい不確実性推定方法であるレイヤーアンサンブルを提案します。さらに、画像レベルの不確実性メトリックを導入します。これは、エントロピーや分散などの一般的に使用されるピクセル単位のメトリックと比較して、セグメンテーションタスクにとってより有益です。 2Dおよび3D、バイナリおよびマルチクラスの医療画像セグメンテーションタスクに関するアプローチを評価します。私たちの方法は、単一のネットワークと単一のパスのみを必要とする最先端のディープアンサンブルで競争力のある結果を示しています。
Uncertainty estimation in deep learning has become a leading research field in medical image analysis due to the need for safe utilisation of AI algorithms in clinical practice. Most approaches for uncertainty estimation require sampling the network weights multiple times during testing or training multiple networks. This leads to higher training and testing costs in terms of time and computational resources. In this paper, we propose Layer Ensembles, a novel uncertainty estimation method that uses a single network and requires only a single pass to estimate predictive uncertainty of a network. Moreover, we introduce an image-level uncertainty metric, which is more beneficial for segmentation tasks compared to the commonly used pixel-wise metrics such as entropy and variance. We evaluate our approach on 2D and 3D, binary and multi-class medical image segmentation tasks. Our method shows competitive results with state-of-the-art Deep Ensembles, requiring only a single network and a single pass.
updated: Wed Mar 16 2022 18:46:53 GMT+0000 (UTC)
published: Wed Mar 16 2022 18:46:53 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アソシエイト