mlsanityData to 7 Oct 2026

HMS-VesselNet: Hierarchical Multi-Scale Attention Network with Topology-Preserving Loss for Retinal Vessel Segmentation

arXiv 2603.21891 ↗First posted 23 Mar 2026eess.IV cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Retinal vessel segmentation methods based on standard overlap losses tend to miss thin peripheral vessels because these structures occupy very few pixels and have low contrast against the background. We propose HMS-VesselNet, a hierarchical multi-scale network that processes fundus images across four parallel branches at different resolutions and combines their outputs using learned fusion weights. The training loss combines Dice, binary cross-entropy, and centerline Dice to jointly optimize area overlap and vessel continuity. Hard example mining is applied from epoch 20 onward to concentrate gradient updates on the most difficult training images. Tested on 68 images from DRIVE, STARE, and CHASE_DB1 using 5-fold cross-validation, the model achieves a mean Dice of 88.72 +/- 0.67%, Sensitivity of 90.78 +/- 1.42%, and AUC of 98.25 +/- 0.21%. In leave-one-dataset-out experiments, AUC remains above 95% on each unseen dataset. The largest improvement is in the recall of thin peripheral vessels, which are the structures most frequently missed by standard methods and most critical for early detection of diabetic retinopathy.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 21 of its 45 references that are in the map. First posted on arXiv 23 Mar 2026.

Builds on

21 of 45 references.

References · 21

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  3. 2016Densely Connected Convolutional Networkscited by 6,691
  4. 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
  5. 2016Feature Pyramid Networks for Object Detectioncited by 5,021
  6. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
  7. 2014Fully Convolutional Networks for Semantic Segmentationcited by 4,845
  8. 2017Squeeze-and-Excitation Networkscited by 4,763
  9. 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,590
  10. 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,552
  11. 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,420
  12. 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,218
9 more
  1. 2019When Does Label Smoothing Help?cited by 985
  2. 2018Albumentations: fast and flexible image augmentationscited by 558
  3. 2020clDice -- A Novel Topology-Preserving Loss Function for Tubular Structure Segmentationcited by 286
  4. 2019Topology-Preserving Deep Image Segmentationcited by 198
  5. 2019A Topological Loss Function for Deep-Learning based Image Segmentation using Persistent Homologycited by 134
  6. 2018DUNet: A deformable network for retinal vessel segmentationcited by 89
  7. 2020SA-UNet: Spatial Attention U-Net for Retinal Vessel Segmentationcited by 70
  8. 2020Genetic U-Net: Automatically Designed Deep Networks for Retinal Vessel Segmentation Using a Genetic Algorithmcited by 9
  9. 2025LFRA-Net: A Lightweight Focal and Region-Aware Attention Network for Retinal Vessel Segmentatiocited by 3

Built on by

Not cited yet.