ドロップアウトは、通常、トレーニングフェーズで正則化方法として、および深層学習の不確実性を定量化するために使用されます。トレーニング中および推論ステップ中にドロップアウトを使用し、不確実性を低減および定量化しながら、精度を向上させるために複数の予測を平均化することを提案します。結果は、3方向スキャンのみから得られた部分異方性(FA)および平均拡散係数(MD)マップについて評価されます。私たちの方法では、特にトレーニングデータセットが小さい場合に、ドロップアウトのないネットワーク出力と比較して精度を大幅に向上させることができます。さらに、目に見えない病理またはアーチファクトの診断に役立つ可能性のある信頼マップが生成されます。
Dropout is conventionally used during the training phase as regularization method and for quantifying uncertainty in deep learning. We propose to use dropout during training as well as inference steps, and average multiple predictions to improve the accuracy, while reducing and quantifying the uncertainty. The results are evaluated for fractional anisotropy (FA) and mean diffusivity (MD) maps which are obtained from only 3 direction scans. With our method, accuracy can be improved significantly compared to network outputs without dropout, especially when the training dataset is small. Moreover, confidence maps are generated which may aid in diagnosis of unseen pathology or artifacts.