arXiv reaDer
正確な網膜血管セグメンテーションのための深層学習法における損失関数の影響
Impact of loss function in Deep Learning methods for accurate retinal vessel segmentation
眼底画像で調べた網膜血管網は、眼だけでなく複数の病気の診断に貢献します。このシステムのセグメンテーションは、形態学的特徴の定量化を支援することにより、これらの画像を分析するという特殊なタスクに役立つ可能性があります。その関連性のために、この問題に自動的に取り組むために、いくつかのディープラーニングベースのアーキテクチャがテストされています。ただし、複雑な網膜血管系のセグメンテーションに対する損失関数の選択の影響は体系的に評価されていません。この作業では、深層学習アーキテクチャ(U-Net、Attention U-Net、およびNested UNet)を使用した損失関数Binary Cross Entropy、Dice、Tversky、およびCombo損失とDRIVEデータセットの比較を示します。それらのパフォーマンスは、AUC、平均二乗誤差、サイコロスコア、ハウスドルフ距離の4つのメトリックを使用して評価されます。モデルは、同じ数のパラメーターとエポックでトレーニングされました。サイコロスコアとAUCを使用した場合、最良の組み合わせはSA-UNetとコンボ損失であり、それぞれ平均0.9442と0.809でした。ハウスドルフ距離と平均二乗誤差の最良の平均は、それぞれ平均6.32と0.0241のダイス損失関数を備えたネストされたU-Netを使用して得られました。結果は、損失関数の選択に有意差があることを示しました。
The retinal vessel network studied through fundus images contributes to the diagnosis of multiple diseases not only found in the eye. The segmentation of this system may help the specialized task of analyzing these images by assisting in the quantification of morphological characteristics. Due to its relevance, several Deep Learning-based architectures have been tested for tackling this problem automatically. However, the impact of loss function selection on the segmentation of the intricate retinal blood vessel system hasn't been systematically evaluated. In this work, we present the comparison of the loss functions Binary Cross Entropy, Dice, Tversky, and Combo loss using the deep learning architectures (i.e. U-Net, Attention U-Net, and Nested UNet) with the DRIVE dataset. Their performance is assessed using four metrics: the AUC, the mean squared error, the dice score, and the Hausdorff distance. The models were trained with the same number of parameters and epochs. Using dice score and AUC, the best combination was SA-UNet with Combo loss, which had an average of 0.9442 and 0.809 respectively. The best average of Hausdorff distance and mean square error were obtained using the Nested U-Net with the Dice loss function, which had an average of 6.32 and 0.0241 respectively. The results showed that there is a significant difference in the selection of loss function
updated: Wed Jun 01 2022 14:47:18 GMT+0000 (UTC)
published: Wed Jun 01 2022 14:47:18 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アソシエイト