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
    • Université de Lausanne (UNIL)
    • Charles Sturt University
    • Al-Balqa Applied University
    • Helmholtz Institute Freiberg for Resource Technology
    • Sukkur IBA University
    • Charles Sturt University, Wagga Wagga
    • Indus University
    • 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 languageBritish English
    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