arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2206.00536eess.IVcs.CV

Impact of loss function in Deep Learning methods for accurate retinal vessel segmentation

Tecnologico de Monterrey(蒙特雷理工学院);Hospital Sant Joan de Déu(圣胡安德迪乌斯医院);Universitat Politècnica de Catalunya(加泰罗尼亚理工大学);Research Centre for Biomedical Engineering (CREB)(生物医学工程研究中心(CREB))

Daniela Herrera, Gilberto Ochoa-Ruiz, Miguel Gonzalez-Mendoza, Christian Mata

英文摘要:

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

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑