ボクセルごとの条件付きMultinoulli分布が想定される識別畳み込みニューラルネットワーク(CNN)は、多くの脳病変セグメンテーションタスクでうまく機能しています。臨床診療で使用される訓練された識別CNNの場合、患者の放射線学的特徴がモデルに入力されます。この場合、セグメンテーションの条件付き分布が生成されます。予測の不確実性を把握することは、モデルを放棄するか、競合するモデルから選択するかを決定するのに役立ちます。ただし、実際には、グラウンドトゥルースセグメンテーションを知ることはできないため、真のモデル分散を知ることはできません。この作業では、識別CNNのセグメンテーションサンプリングを使用して、磁気共鳴(MR)画像のみに基づいて新しい患者のサンプル間ダイス分布を分析することにより、トレーニング済みモデルの堅牢性を評価します。さらに、サンプル間のダイス観測が独立しており、特定の条件下で有限の平均と分散で同一に分布していることを示すことにより、特定の患者のCNNモデルを拒否するか受け入れるかを決定するための厳密な信頼ベースの決定ルールが提案されます。 ISLES 2015(SISS)データセットに適用されたモデルは、7つの予測を非ロバストとして識別し、残りの脳で計算された平均ダイス係数は12%向上しました。
Discriminative convolutional neural networks (CNNs), for which a voxel-wise conditional Multinoulli distribution is assumed, have performed well in many brain lesion segmentation tasks. For a trained discriminative CNN to be used in clinical practice, the patient's radiological features are inputted into the model, in which case a conditional distribution of segmentations is produced. Capturing the uncertainty of the predictions can be useful in deciding whether to abandon a model, or choose amongst competing models. In practice, however, we never know the ground truth segmentation, and therefore can never know the true model variance. In this work, segmentation sampling on discriminative CNNs is used to assess a trained model's robustness by analyzing the inter-sample Dice distribution on a new patient solely based on their magnetic resonance (MR) images. Furthermore, by demonstrating the inter-sample Dice observations are independent and identically distributed with a finite mean and variance under certain conditions, a rigorous confidence based decision rule is proposed to decide whether to reject or accept a CNN model for a particular patient. Applied to the ISLES 2015 (SISS) dataset, the model identified 7 predictions as non-robust, and the average Dice coefficient calculated on the remaining brains improved by 12 percent.