Skip to main navigation Skip to search Skip to main content

An Empirical Analysis of Loss Functions for Deep Retinal Vessel Segmentation

  • Ayoub Fatihi
  • , Toufique Ahmed Soomro
  • , Tareq A. Alawneh
  • , Ahmed J. Afifi
  • , Faisal Bin Ubaid
  • , Herbert Jelinek
  • , Lihong Zheng
  • , Shafique Ahmed Soomro
  • , Junbin Gao
  • University of Lausanne
  • Charles Sturt University
  • The University of Larkano
  • Al-Balqa Applied University
  • Helmholtz-Zentrum Dresden-Rossendorf
  • Sukkur IBA University
  • Indus University, Pakistan
  • University of Sydney

Research output: Contribution to journalArticlepeer-review

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.

Original languageEnglish
Article number11
JournalACM Transactions on Computing for Healthcare
Volume7
Issue number1
DOIs
StatePublished - 14 Jan 2026

Keywords

  • cross-entropy
  • deep learning
  • diagnosis
  • dice loss
  • loss functions
  • medical image analysis
  • retinal vessel segmentation
  • treatment
  • weighted cross-entropy

Fingerprint

Dive into the research topics of 'An Empirical Analysis of Loss Functions for Deep Retinal Vessel Segmentation'. Together they form a unique fingerprint.

Cite this