LDMRes-Net: A Lightweight Neural Network for Efficient Medical Image Segmentation on IoT and Edge Devices

被引:9
作者
Iqbal, Shahzaib [1 ]
Khan, Tariq M. [2 ]
Naqvi, Syed S. [3 ]
Naveed, Asim [4 ]
Usman, Muhammad
Khan, Haroon Ahmed
Razzak, Imran
机构
[1] COMSATS Univ Islamabad CUI, Dept Elect & Comp Engn, Islamabad 45550, Pakistan
[2] Univ New South Wales, Sch Comp Sci & Engn, Sydney, NSW 2052, Australia
[3] COMSATS Univ Islamabad CUI, Dept Elect & Comp Engn, Islamabad 39161, Pakistan
[4] Univ Engn & Technol Lahore, Comp Sci Dept, Lahore 110460, Pakistan
关键词
Image segmentation; Medical diagnostic imaging; Computer architecture; Retina; Computational modeling; Training; Image edge detection; Medical image segmentation; retinal features segmentation; light-weight deep networks; dual multiscale residual block; BLOOD-VESSEL SEGMENTATION; U-NET; RETINAL IMAGES; ARCHITECTURE; OPTIMIZATION; CONNECTIONS;
D O I
10.1109/JBHI.2023.3331278
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In this study, we propose LDMRes-Net, a lightweight dual-multiscale residual block-based convolutional neural network tailored for medical image segmentation on IoT and edge platforms. Conventional U-Net-based models face challenges in meeting the speed and efficiency demands of real-time clinical applications, such as disease monitoring, radiation therapy, and image-guided surgery. In this study, we present the Lightweight Dual Multiscale Residual Block-based Convolutional Neural Network (LDMRes-Net), which is specifically designed to overcome these difficulties. LDMRes-Net overcomes these limitations with its remarkably low number of learnable parameters (0.072 M), making it highly suitable for resource-constrained devices. The model's key innovation lies in its dual multiscale residual block architecture, which enables the extraction of refined features on multiple scales, enhancing overall segmentation performance. To further optimize efficiency, the number of filters is carefully selected to prevent overlap, reduce training time, and improve computational efficiency. The study includes comprehensive evaluations, focusing on the segmentation of the retinal image of vessels and hard exudates crucial for the diagnosis and treatment of ophthalmology. The results demonstrate the robustness, generalizability, and high segmentation accuracy of LDMRes-Net, positioning it as an efficient tool for accurate and rapid medical image segmentation in diverse clinical applications, particularly on IoT and edge platforms. Such advances hold significant promise for improving healthcare outcomes and enabling real-time medical image analysis in resource-limited settings.
引用
收藏
页码:3860 / 3871
页数:12
相关论文
共 94 条
[1]  
[Anonymous], 2021, Int. J. Comput. Vis., V129, P3051
[2]  
[Anonymous], 2022, IEEE J. Sel. Areas Commun., V40, P3122
[3]  
[Anonymous], 2023, IEEE J. Biomed. Health Inform.
[4]   Prompt Deep Light-Weight Vessel Segmentation Network (PLVS-Net) [J].
Arsalan, Muhammad ;
Khan, Tariq M. ;
Naqvi, Syed Saud ;
Nawaz, Mehmood ;
Razzak, Imran .
IEEE-ACM TRANSACTIONS ON COMPUTATIONAL BIOLOGY AND BIOINFORMATICS, 2023, 20 (02) :1363-1371
[5]   Aiding the Diagnosis of Diabetic and Hypertensive Retinopathy Using Artificial Intelligence-Based Semantic Segmentation [J].
Arsalan, Muhammad ;
Owais, Muhammad ;
Mahmood, Tahir ;
Cho, Se Woon ;
Park, Kang Ryoung .
JOURNAL OF CLINICAL MEDICINE, 2019, 8 (09)
[6]   Bi-Directional ConvLSTM U-Net with Densley Connected Convolutions [J].
Azad, Reza ;
Asadi-Aghbolaghi, Maryam ;
Fathy, Mahmood ;
Escalera, Sergio .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW), 2019, :406-415
[7]   SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation [J].
Badrinarayanan, Vijay ;
Kendall, Alex ;
Cipolla, Roberto .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) :2481-2495
[8]   Fast Retinal Vessel Detection and Measurement Using Wavelets and Edge Location Refinement [J].
Bankhead, Peter ;
Scholfield, C. Norman ;
McGeown, J. Graham ;
Curtis, Tim M. .
PLOS ONE, 2012, 7 (03)
[9]  
Chen J., 2021, arXiv
[10]   DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J].
Chen, Liang-Chieh ;
Papandreou, George ;
Kokkinos, Iasonas ;
Murphy, Kevin ;
Yuille, Alan L. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) :834-848