journal article·2024Open access

ProNet Adaptive Retinal Vessel Segmentation Algorithm Based on Improved UperNet Network

Sijia Zhu, Pinxiu Wang, Ke Shen

Computers, Materials & Continua· Volume 78· Issue 1

Abstract

This paper proposes a new network structure, namely the ProNet network. Retinal medical image segmentation can help clinical diagnosis of related eye diseases and is essential for subsequent rational treatment. The baseline model of the ProNet network is UperNet (Unified perceptual parsing Network), and the backbone network is ConvNext (Convolutional Network). A network structure based on depth-separable convolution and 1 × 1 convolution is used, which has good performance and robustness. We further optimise ProNet mainly in two aspects. One is data enhancement using increased noise and slight angle rotation, which can significantly increase the diversity of data and help the model better learn the patterns and features of the data and improve the model's performance. Meanwhile, it can effectively expand the training data set, reduce the influence of noise and abnormal data in the data set on the model, and improve the accuracy and reliability of the model. Another is the loss function aspect, and we finally use the focal loss function. The focal loss function is well suited for complex tasks such as object detection. The function will penalise the loss carried by samples that the model misclassifies, thus enabling better training of the model to avoid these errors while solving the category imbalance problem as a way to improve image segmentation density and segmentation accuracy. From the experimental results, the evaluation metrics mIoU (mean Intersection over Union) enhanced by 4.47%, and mDice enhanced by 2.92% compared to the baseline network. Better generalization effects and more accurate image segmentation are achieved.
Source: openalex_inverted_index · Quality: baseline

Research context

OpenAlex topics: Retinal Imaging and Analysis · Glaucoma and retinal disorders · Gaze Tracking and Assistive Technology

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

References (37)

Show source-provided references
1. A new uncertainty measure via belief Renyi entropy in Dempster-Shafer theory and its application to decision making
2. Distance measures of picture fuzzy sets and interval-valued picture fuzzy sets with their applications10.3934/math.20231525 ↗
3. Credal-based fuzzy number data clustering10.1007/s41066-023-00410-0 ↗
4. Image segmentation based on adaptive K-means algorithm
5. Classification accuracy in the measurement of MMPI/MMPI–2: Profile similarity10.1207/s15327752jpa7101_5 ↗
6. Deformable models in medical image analysis: A survey10.1016/S1361-8415(96)80007-7 ↗
7. Mask R-CNN
8. SegNet: A deep convolutional encoder-decoder architecture for image segmentation10.1109/TPAMI.2016.2644615 ↗
9. U-Net: Convolutional networks for biomedical image segmentation
10. V-Net: Fully convolutional neural networks for volumetric medical image segmentation
11. Progressive knowledge-embedded unified perceptual parsing for scene understanding
12. Deep residual learning for image recognition
13. ImageNet: A large-scale hierarchical image database
14. Pyramid scene parsing network
15. Feature pyramid networks for object detection
16. A ConvNet for the 2020s
17. Attention is all you need
18. J. L. Ba, J. R. Kiros and G. E. Hinton, “Layer normalization,” 2016. 10.48550/arXiv.1607.06450 (accessed 04/03/2023)
19. Swin transformer: Hierarchical vision transformer using shifted windows
20. Inverted residuals and linear bottlenecks: Mobile networks for classification, detection and segmentation
21. Focal loss for dense object detection
22. A survey of loss functions for semantic segmentation
23. The lovasz-softmax loss: A tractable surrogate for the optimization of the intersection-over-union measure in neural networks
24. Measuring retinal vessel tortuosity in 10-year-old children: Validation of the Computer-assisted image analysis of the retina (CAIAR) program10.1167/iovs.08-3018 ↗
25. Deep learning on computerized analysis of chronic obstructive pulmonary disease
26. Bi-directional ConvLSTM U-Net with densley connected convolutions
27. Ridge-based vessel segmentation in color images of the retina10.1109/TMI.2004.825627 ↗
28. U-Net: Convolutional networks for biomedical image segmentation
29. A new supervised retinal vessel segmentation method based on the robust hybrid feature10.1016/j.bspc.2016.05.006 ↗
30. FC-RCCN: Fully convolutional residual continuous CRF network for semantic segmentation10.1016/j.patrec.2018.08.030 ↗
31. Blood vessel segmentation of fundus images bymajor vessel extraction and subimage classification
32. Joint optic disc and cup segmentation based on multi-label deep network and polar transformation10.1109/TMI.2018.2791488 ↗
33. Attention guided network for retinal image segmentation
34. Factorized U-net for retinal vessel segmentation
35. DR-VNet: Retinal vessel segmentation via dense residual UNet
36. Do you need sharpened details? Asking MMDC-Net: Multi-layer multi-scale dilated convolution network for retinal vessel segmentation10.1016/j.compbiomed.2022.106198 ↗
37. Ridge–based vessel segmentation in color images of the retina