LMBiS-Net: A lightweight bidirectional skip connection based multipath CNN for retinal blood vessel segmentation

被引:3
作者
Abbasi, Mufassir Matloob [1 ]
Iqbal, Shahzaib [1 ]
Aurangzeb, Khursheed [2 ]
Alhussein, Musaed [2 ]
Khan, Tariq M. [3 ]
机构
[1] Abasyn Univ Islamabad Campus AUIC, Dept Elect Engn, Islamabad Campus, Islamabad 44000, Pakistan
[2] King Saud Univ, Coll Comp & Informat Sci, Dept Comp Engn, POB 51178, Riyadh 11543, Saudi Arabia
[3] Univ New South Wales, Sch Comp Sci & Engn, Sydney, NSW, Australia
关键词
Retina blood vessel segmentation; Bidirectional skip connections; Multipath connections; U-NET; NETWORK; ARCHITECTURE; IMAGES;
D O I
10.1038/s41598-024-63496-9
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Blinding eye diseases are often related to changes in retinal structure, which can be detected by analysing retinal blood vessels in fundus images. However, existing techniques struggle to accurately segment these delicate vessels. Although deep learning has shown promise in medical image segmentation, its reliance on specific operations can limit its ability to capture crucial details such as the edges of the vessel. This paper introduces LMBiS-Net, a lightweight convolutional neural network designed for the segmentation of retinal vessels. LMBiS-Net achieves exceptional performance with a remarkably low number of learnable parameters (only 0.172 million). The network used multipath feature extraction blocks and incorporates bidirectional skip connections for the information flow between the encoder and decoder. In addition, we have optimised the efficiency of the model by carefully selecting the number of filters to avoid filter overlap. This optimisation significantly reduces training time and improves computational efficiency. To assess LMBiS-Net's robustness and ability to generalise to unseen data, we conducted comprehensive evaluations on four publicly available datasets: DRIVE, STARE, CHASE_DB1, and HRF The proposed LMBiS-Net achieves significant performance metrics in various datasets. It obtains sensitivity values of 83.60%, 84.37%, 86.05%, and 83.48%, specificity values of 98.83%, 98.77%, 98.96%, and 98.77%, accuracy (acc) scores of 97.08%, 97.69%, 97.75%, and 96.90%, and AUC values of 98.80%, 98.82%, 98.71%, and 88.77% on the DRIVE, STARE, CHEASE_DB, and HRF datasets, respectively. In addition, it records F1 scores of 83.43%, 84.44%, 83.54%, and 78.73% on the same datasets. Our evaluations demonstrate that LMBiS-Net achieves high segmentation accuracy (acc) while exhibiting both robustness and generalisability across various retinal image datasets. This combination of qualities makes LMBiS-Net a promising tool for various clinical applications.
引用
收藏
页数:13
相关论文
共 65 条
[31]   ESDMR-Net: A lightweight network with expand-squeeze and dual multiscale residual connections for medical image segmentation [J].
Khan, Tariq M. ;
Naqvi, Syed S. ;
Meijering, Erik .
ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 133
[32]  
Khan TM, 2023, Arxiv, DOI arXiv:2309.03535
[33]   T-Net: A Resource-Constrained Tiny Convolutional Neural Network for Medical Image Segmentation [J].
Khan, Tariq M. ;
Robles-Kelly, Antonio ;
Naqvi, Syed S. .
2022 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2022), 2022, :1799-1808
[34]   Width-wise vessel bifurcation for improved retinal vessel segmentation [J].
Khan, Tariq M. ;
Khan, Mohammad A. U. ;
Rehman, Naveed Ur ;
Naveed, Khuram ;
Afridi, Imran Uddin ;
Naqvi, Syed Saud ;
Raazak, Imran .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 71
[35]   Residual Multiscale Full Convolutional Network (RM-FCN) for High Resolution Semantic Segmentation of Retinal Vasculature [J].
Khan, Tariq M. ;
Robles-Kelly, Antonio ;
Naqvi, Syed S. ;
Arsalan, Muhammad .
STRUCTURAL, SYNTACTIC, AND STATISTICAL PATTERN RECOGNITION, S+SSPR 2020, 2021, 12644 :324-333
[36]   A Semantically Flexible Feature Fusion Network for Retinal Vessel Segmentation [J].
Khan, Tariq M. ;
Robles-Kelly, Antonio ;
Naqvi, Syed S. .
NEURAL INFORMATION PROCESSING, ICONIP 2020, PT IV, 2020, 1332 :159-167
[37]   A region growing and local adaptive thresholding-based optic disc detection [J].
Khan, Tariq M. ;
Mehmood, Mehwish ;
Naqvi, Syed S. ;
Butt, Muhammad Fasih Uddin .
PLOS ONE, 2020, 15 (01)
[38]  
Laibacher T, 2019, Arxiv, DOI arXiv:1811.07738
[39]  
Lei T, 2020, INT CONF ACOUST SPEE, P1379, DOI [10.1109/ICASSP40776.2020.9053454, 10.1109/icassp40776.2020.9053454]
[40]   PyConvU-Net: a lightweight and multiscale network for biomedical image segmentation [J].
Li, Changyong ;
Fan, Yongxian ;
Cai, Xiaodong .
BMC BIOINFORMATICS, 2021, 22 (01)