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

Institutions

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
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

An Empirical Analysis of Loss Functions for Deep Retinal Vessel Segmentation

Toufique Ahmed Soomro, Herbert F. Jelinek, Junbin Gao, Ayoub Fatihi et al.
ACM Transactions on Computing for Healthcare
Retinal Imaging and Analysis
article

An Empirical Analysis of Loss Functions for Deep Retinal Vessel Segmentation

Toufique Ahmed Soomro, Herbert F. Jelinek, Junbin Gao, Ayoub Fatihi, Lihong Zheng, Shafique Ahmed Soomro, Faisal Bin Ubaid, Ahmed J. Afifi, Tareq A. Alawneh
article en

Abstract

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.

ACM Transactions on Computing for HealthcareVol. 7(1)
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)
Openalex Percentile: Top 47%
Retinal Imaging and Analysis
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.