An Empirical Analysis of Loss Functions for Deep Retinal Vessel Segmentation
Retinal vessel segmentation underpins computer-assisted screening and monitoring of ocular and systemic disease. While encoder–decoder networks such as U-Net are widely used, their behavior is strongly shaped by the training objective. This work presents a controlled empirical study of loss functions for vessel segmentation using a U-Net architecture that employs strided convolutions in the encoder, together with a consistent pre-processing pipeline based on morphological enhancement and principal component analysis. We compare cross-entropy, weighted cross-entropy, and Dice losses on the DRIVE and STARE datasets under identical settings, reporting pixel-wise and overlap-based measures to reflect both detection and spatial agreement. The configuration with weighted cross-entropy provides a balanced outcome, achieving sensitivity and accuracy of 0.873 and 0.969 on DRIVE, and 0.821 and 0.961 on STARE. Rather than proposing architectural novelty, the contribution of this study is a reproducible data-driven comparison that clarifies the tradeoffs each loss imposes on recall, specificity, and boundary fidelity, offering practical guidance for selecting objectives in retinal vessel segmentation.
Authors
- Toufique Ahmed Soomro (ORCID: https://orcid.org/0000-0002-8560-0026)
- Herbert F. Jelinek (ORCID: https://orcid.org/0000-0001-5457-6193)
- Junbin Gao (ORCID: https://orcid.org/0000-0001-9803-0256)
- Ayoub Fatihi (ORCID: https://orcid.org/0000-0001-8572-6553)
- Lihong Zheng (ORCID: https://orcid.org/0000-0001-5728-4356)
- Shafique Ahmed Soomro (ORCID: https://orcid.org/0009-0009-1610-0245)
- Faisal Bin Ubaid (ORCID: https://orcid.org/0009-0009-4111-143X)
- Ahmed J. Afifi (ORCID: https://orcid.org/0000-0001-6782-6753)
- Tareq A. Alawneh (ORCID: https://orcid.org/0000-0002-2400-1599)
Institutions
- The University of Sydney (AU)
- Charles Sturt University (AU)
- Khalifa University of Science and Technology (AE)
- Al-Balqa Applied University (JO)
- Indus University (PK)
- Helmholtz Institute Freiberg for Resource Technology (DE)
- Sukkur IBA University (PK)
- University of Lausanne (CH)
Publication Details
- Journal
- ACM Transactions on Computing for Healthcare
- Published
- 2025-12-04
- DOI
- https://doi.org/10.1145/3779294
- Primary Topic
- Retinal Imaging and Analysis
- Type
- article
- Field-Weighted Citation Impact
- 0.00