Skip to main content

ORIGINAL RESEARCH article

Front. Med., 08 September 2023
Sec. Ophthalmology
This article is part of the Research Topic Big Data and Artificial Intelligence in Ophthalmology - Clinical Application and Future Exploration View all 10 articles

DME-DeepLabV3+: a lightweight model for diabetic macular edema extraction based on DeepLabV3+ architecture

Yun Bai&#x;Yun Bai1Jing Li&#x;Jing Li1Lianjun Shi&#x;Lianjun Shi2Qin Jiang
Qin Jiang2*Biao Yan
Biao Yan3*Zhenhua Wang
Zhenhua Wang1*
  • 1College of Information Science, Shanghai Ocean University, Shanghai, China
  • 2The Affiliated Eye Hospital, Nanjing Medical University, Nanjing, China
  • 3Eye Institute, Eye and ENT Hospital, Shanghai Medical College, Fudan University, Shanghai, China

Introduction: Diabetic macular edema (DME) is a major cause of vision impairment in the patients with diabetes. Optical Coherence Tomography (OCT) is an important ophthalmic imaging method, which can enable early detection of DME. However, it is difficult to achieve high-efficiency and high-precision extraction of DME in OCT images because the sources of OCT images are diverse and the quality of OCT images is not stable. Thus, it is still required to design a model to improve the accuracy of DME extraction in OCT images.

Methods: A lightweight model (DME-DeepLabV3+) was proposed for DME extraction using a DeepLabV3+ architecture. In this model, MobileNetV2 model was used as the backbone for extracting low-level features of DME. The improved ASPP with sawtooth wave-like dilation rate was used for extracting high-level features of DME. Then, the decoder was used to fuse and refine low-level and high-level features of DME. Finally, 1711 OCT images were collected from the Kermany dataset and the Affiliated Eye Hospital. 1369, 171, and 171 OCT images were randomly selected for training, validation, and testing, respectively.

Conclusion: In ablation experiment, the proposed DME-DeepLabV3+ model was compared against DeepLabV3+ model with different setting to evaluate the effects of MobileNetV2 and improved ASPP on DME extraction. DME-DeepLabV3+ had better extraction performance, especially in small-scale macular edema regions. The extraction results of DME-DeepLabV3+ were close to ground truth. In comparative experiment, the proposed DME-DeepLabV3+ model was compared against other models, including FCN, UNet, PSPNet, ICNet, and DANet, to evaluate DME extraction performance. DME-DeepLabV3+ model had better DME extraction performance than other models as shown by greater pixel accuracy (PA), mean pixel accuracy (MPA), precision (Pre), recall (Re), F1-score (F1), and mean Intersection over Union (MIoU), which were 98.71%, 95.23%, 91.19%, 91.12%, 91.15%, and 91.18%, respectively.

Discussion: DME-DeepLabV3+ model is suitable for DME extraction in OCT images and can assist the ophthalmologists in the management of ocular diseases.

1. Introduction

Diabetic macular edema (DME) is the major cause of vision loss in the patients with diabetic retinopathy. Increasing prevalence of DME is tightly correlated with the global epidemic of diabetes mellitus (1, 2). DME is usually caused by the rupture of retinal barrier and increased permeability of retinal vessels, which is characterized by the leakage of fluid and other plasma components. The effusion can accumulate in the macula, resulting in edema (3, 4). In the clinical work, the presence and severity of retinopathy are required to be determined according to the size of edema area.

Optical Coherence Tomography (OCT) is a non-contact, non-invasive, and highly sensitive ophthalmic imaging method, which can enable early detection of diabetic macular edema by observing the transverse section of macular degeneration (5). Normal OCT image is shown in Figure 1A and OCT image with DME is shown in Figure 1B. DMEs accumulated in typical relative positions within the main retinal layers. Based on OCT patterns of DME, DME can be classified into three different patterns, including diffuse retinal thickening (DRT), cystoid macular edema (CME), and serous retinal detachment (SRD). CME normally starts to manifest symptoms in the inner retina, while SRD and DRT typically appear in the outer retina. In the severe advanced stages of DR, CMEs can also proliferate from the inner to the outer retina and merge with DRT (6). Thus, rapid and accurate detection of all types of edemas is of great significance for evaluating the progression of diabetic retinopathy. In the clinical work, DME is usually segmented by the well-trained experts (7). However, manual extraction of DME edemas is time-consuming and labor-intensive. Moreover, there is inevitable variability in the extraction results by different experts. With increased prevalence of diabetes, an increasing number of patients require disease management based on OCT images in the clinical practices. Thus, it is highly required to design an automatic method for rapid and accurate detection of DME in OCT images.

FIGURE 1
www.frontiersin.org

Figure 1. Optical coherence tomography images in diabetic patients and healthy controls. (A) Normal OCT image; (B) OCT image with DME.

Image extraction is processed and analyzed according to the features, including image color, spatial structure, and texture information (8). Image extraction models can divide an image into several specific regions, such as threshold-based extraction model (9, 10), region-based extraction model (11, 12), and edge detection-based extraction model (13). With the development of deep learning, several models have been developed to extract DME, such as fully convolutional network (FCN), U-Net, and PSPNet. Based on these deep learning models, several scholars have also developed the improved models for DME extraction. Table 1 showed the strengths and weaknesses of different models for DME extraction.

TABLE 1
www.frontiersin.org

Table 1. Strengths and weaknesses of different models for DME extraction.

The sources of OCT images are diverse and the quality of OCT images is not always stable. Moreover, the size and distribution of DMEs are not uniform and the borders of DMEs are blurred. Thus, it is still required to design a novel model to improve the accuracy of DME extraction in OCT images. In this study, we proposed a lightweight automatic model (DME-DeepLabV3+) based on the DeepLabV3+ architecture. The major contributions of the proposed DME-DeepLabV3+ are shown below:

Taking MobileNetV2 as the backbone, the ability of DME-DeepLabV3+ is improved in extracting the low-level features of DME.

Improving ASPP by the sawtooth wave-like dilation rate, DME-DeepLabV3+ avoids grid effects, learns more local information, and extracts high-level features of DME better.

Based on the decoder, DME-DeepLabV3+ fuses the low-level and high-level features of DME, and refines the results of DME extraction.

2. Materials and methods

The flowchart of the proposed model, DME-DeepLabV3+, was shown in Figure 2.

FIGURE 2
www.frontiersin.org

Figure 2. Flowchart of DME extraction by DME-DeepLabV3+ model.

Low-level features of DME extraction by MobileNetV2; High-level features of DME extraction by the improved ASPP; Fusion and refinement of low-level and high-level features of DME by the decoder.

2.1. Low-level features of DME extraction by MobileNetV2

DeepLabV3+ is a deep learning model for image extraction with deep convolutional nets, which takes Xception as the backbone network (21). Xception uses numerous parameters, complicated operations, and high computer performance requirements (22), which leads to several challenges for DME extraction, such as fault-extraction and over-extraction problems. MobileNetV2 is a lightweight network, which shows a great advantage to solve the fault-extraction and over-extraction problems (23). In DME-DeepLabV3+ model, we used MobileNetV2 as the backbone to simplify model structure, which could improve the extraction efficiency and reduce the problems of fault-extraction and over-extraction.

MobileNetV2 used depthwise separable convolution to reduce the number of parameters and complex operations. Depthwise separable convolution consisted of DepthWise (DW) and PointWise (PW), whereas DW performed convolution operations on each channel of the input layer and PW fused the features and obtained the feature information with stronger expressive ability. MobileNetV2 used Inverted Residual to improve the memory efficiency.

In Inverted Residual, the dimension of DME features was increased by 1 × 1 convolution. Next, DME features were extracted by 3 × 3 DW convolution, and the dimension of DME features was reduced by 1 × 1 convolution (Figure 3). When the stride was 1, DME output features were consistent with the input features and shortcuts were used to add the elements of DME input and output. When the stride was 2, no shortcut was required. At the same time, a linear bottleneck neural network was used in the last 1 × 1 convolutional layer of Inverted Residual, which could reduce the loss of low-dimensional feature of DME information.

FIGURE 3
www.frontiersin.org

Figure 3. Structure of Inverted Residual. (A) Stride = 1; (B) Stride = 2.

Compared with DeepLabV3+ with Xception as the backbone network, DME-DeepLabV3+ with MobileNetV2 as the backbone network not only improved the accuracy but also improved the efficiency in DME extraction.

2.2. High-level features of DME extraction by the improved ASPP

ASPP consists of atrous convolution with different dilation rates, which strikes the best trade-off between multi-scale feature extraction and context assimilation, especially for small objects (24). DME has multi-scale features, especially with several small areas of edema. ASPP was then used to extract high-level features of DME. However, the dilation rate in ASPP had a grid effect, which not only lost the semantic information but also ignored the consistency of local information in edema regions (25). Here, we replaced the original dilation rate with the sawtooth wave-like dilation rate to improve ASPP for extracting the high-level features of DME. A sawtooth wave-like dilation rate was formed by the repeated combination of two sets of the same “rising edge” type dilation rate.

Figures 4, 5 show the illustration of the atrous convolution principle of DeepLabV3+ and DME-DeepLabV3+, respectively. Figures 4A,B show RF (receptive field) and the number of calculation times of DeepLabV3+. Figures 5A,B show RF and the number of calculation times of DME-DeepLabV3+. The results show that there was about 73% of information loss due to the grid effect in DeepLabV3+ model. In DME-DeepLabV3+ model, each pixel was effectively used and involved in further computations. Compared with DeepLabV3+ model, increased dilation rate in DME-DeepLabV3+ model can avoid the grid effects and learn more local information.

FIGURE 4
www.frontiersin.org

Figure 4. Illustration of atrous convolution principle of DeepLabV3 + with  dilation rate = [1, 6, 12, 18] and RF  =  75  ×  75. (A) Effective pixels in RF, which were marked in blue; (B) The number of calculation times of each pixel.

FIGURE 5
www.frontiersin.org

Figure 5. Illustration of atrous convolution principle of DME-DeepLabV3+ with dilation rate = [1, 3, 9, 1, 3, 9] and RF  =  53  ×  53. (A) Effective pixels in RF, which were marked in blue; (B) The number of calculation times of each pixel.

2.3. Fusion and refinement of low-level and high-level features by the decoder

Low-level and high-level features of DME were extracted by MobileNetV2 and the improved ASPP, respectively. All features of DME were fused and refined by the decoder. The decoder is mainly composed of ordinary convolution and fusion layers. It fuses the features extracted from the encoder, uses the up-sampling to restore the feature dimension, and outputs the prediction results of the same size with less information loss as possible (26). In the decoder, low-level features with fewer channels were obtained by 1 × 1 convolution. Bilinear up-sampling of high-level features were conducted by a factor of 4. The concatenation features were obtained by concatenating the low-level features and high-level features and a feature concatenation was refined by a few 3 × 3 convolutions. Finally, the results of DME extraction were output following another bilinear up-sampling by a factor of 4.

2.4. Ethical statement

The design and conduct of this study adhere to the intent and principles of the Declaration of Helsinki. The protocols were also reviewed and approved by the ethical committee of Eye Hospital (Nanjing medical university). Informed consents were obtained from all participants.

2.5. Datasets

The datasets contained 1711 OCT images, including 416 images (512 × 512 pixels) selected from the Kermany dataset (27) and 1,295 OCT datasets (938 × 420 pixels) collected from the Affiliated Eye Hospital, Nanjing Medical University. All patients were required to undergo OCT scanning by a spectral domain OCT (RTVue, Optovue Inc., United States). These OCT images were centered on the macula with an axial resolution of 10 μm and a 24-bit depth and were acquired in 2 s, covering 4 × 4 mm area. Inclusion criteria were as follows: the presence of macular edema in at least one eye and clear optical media allowing OCT imaging with good quality. Subsets of 1,369, 171, and 171 OCT images were randomly selected for training, validation, and testing, respectively. Each OCT image was individually labeled by three experienced clinicians who had more than 10-year clinical working experience. The annotation results were binarized by MATLAB software, where the background was labeled as 0 and the DME labeled as 1. Due to the limited human energy, some artificial deviations were inevitable. For these images, a senior expert was consulted and thorough rounds of discussion and adjudication were conducted to ensure the accuracy of the labeling. The original OCT images and ground truth are shown in Figure 6.

FIGURE 6
www.frontiersin.org

Figure 6. Original OCT images and DME labeling by three experienced clinicians.

3. Implementation

The hardware configurations used for this study are shown below: Windows 10, NVIDIA GeForce RTX 3060. The software environment is the deep-learning framework PyTorch 1.10.0, CUDA 11.3, and the programming language Python 3.9.

4. Evaluation metrics

Seven metrics were calculated to estimate the extraction performance of DME-DeepLabV3+, including pixel accuracy (PA), mean pixel accuracy (MPA), precision (Pre), recall (Re), F1-score (F1), mean intersection over union (MIoU), and frames per second (FPS).

PA = i = 0 k p ii i = 0 k j = 0 k p ij     (1)
MPA = 1 k + 1 i = 0 k p ii j = 0 k p ij     (2)
Pre = TP TP + FP     (3)
Re = TP TP + FN     (4)
F 1 = 2 × Pre × Re Pre + Re     (5)
MIoU = 1 k + 1 i = 0 k p ii j = 0 k p ij + j = 0 k p ji p ii     (6)
FPS = frameNum elapsedTime     (7)

TP, FP, and FN denote the true positive region, false positive region, and false negative region, respectively. p i i is the number of edema area pixels which was correctly classified as edema areas; p i j is the number of background area pixels which are misclassified as edema areas; p j i is the number of edema area pixels which are incorrectly classified as the background; k is the labeling results of different classes, where k = 0 expressed as background class and k = 1 as DME class; f r a m e N u m is the number of OCT images that are input to the model when performing inference; e l a p s e d T i m e is the time consumed by the model when performing inference. PA is the overall pixel accuracy. MPA is the average pixel accuracy of DME and background. Pre and Re are the proportion of real DME regions in the samples predicted as DME and the proportion of correct predictions in all DME, respectively. F1-score (F1) is a balanced metric and determined by precision and recall. MIoU is a metric to measure the similarity of ground truth and prediction. FPS is the number of OCT images inferred per second.

5. Results

To evaluate the performance of DME extraction of DME-DeepLabV3+ model, two comparative experiments were performed. In experiment 1, DME extraction performance of DME-DeepLabV3+ model was evaluated by comparing against DeepLabV3+ model under different settings. In experiment 2, DME extraction performance of DME-DeepLabV3+ model was evaluated by comparing against other end-to-end models, including FCN, UNet, PSPNet, ICNet, and DANet.

5.1. Experiment 1 (ablation experiment)

To evaluate the effects of MobileNetV2 and the improved ASPP on DME extraction performance, the proposed DME-DeepLabV3+ model was compared against DeepLabV3+ model with different settings, including DeepLabV3+, DeepLabV3+ with MobileNetV2 (MobileNetV2-DeepLabV3+), DeepLabV3+ with the improved ASPP (Improved ASPP-DeepLabV3+). Figure 7 showed the DME extraction results by DeepLabV3+ model with different settings, where red, blue, and white DME regions represented true positive (TP) regions, false positive (FP) regions and false negative (FN) regions, respectively. DeepLabV3+ model led to some missed and false extraction of DME. MobileNetV2-DeepLabV3+ and improved ASPP-DeepLabV3+ reduced the missed and false extraction of DME. However, the missed extraction still existed in small edematous regions as shown in OCT2. DME-DeepLabV3+ had better extraction performance, especially in small-scale macular edema regions. The extraction results of DME-DeepLabV3+ were close to the ground truth.

FIGURE 7
www.frontiersin.org

Figure 7. DME extraction results by DeepLabV3+ model with different settings Red, blue, and white DME regions represented TP regions, FP regions and FN regions.

Table 2 showed the results of evaluation metrics for DeepLabV3+ under different settings. Compared with DeepLabV3+ model, the MobileNetV2-DeepLabV3+ enhanced the scores of PA, MPA, Pre, Re, F1, MIoU, and FPS of DME extraction results, which were 98.69(0.28↑), 94.95(1.50↑), 91.02(0.57↑), 91.09(3.31↑), 91.06(1.97↑), 91.05(1.78↑), and 9.24(3.40↑), respectively. FPS increased by about 58%. The improved ASPP-DeepLabV3+ enhanced the scores of PA, MPA, Re, F1, and MIoU of DME extraction results, which were 98.47(0.06↑), 93.90(0.45↑), 88.49(0.71↑), 89.45(0.36↑), and 89.61(0.34↑), respectively. DME-DeepLabV3+ enhanced the scores of PA, MPA, Pre, Re, F1, and MIoU of DME extraction results, which were 98.71(0.30↑), 95.23(1.78↑), 91.19(0.74↑), 91.12(3.34↑), 91.15(2.06↑), and 91.18(1.91↑), respectively. FPS was 9.03, which was lower than that of MobileNetV2-DeepLabV3+ (0.21↓).

TABLE 2
www.frontiersin.org

Table 2. Evaluation metrics of DME extraction by DeepLabV3+ model with different settings.

5.2. Experiment 2 (comparative experiment)

We evaluate DME extraction performance of DME-DeepLabV3+ model by comparing against other models, including FCN, UNet, PSPNet, ICNet, and DANet. Figure 8 showed DME extraction results by different models, where red, blue, and white DME regions represented true positive (TP) regions, false positive (FP) regions, and false negative (FN) regions, respectively. As shown in Figure 8, DME extraction results of DME-DeepLabV3+ were close to the ground truth. A part of the background was extracted falsely by FCN, U-Net, or DANet models. Compared with FCN and U-Net, PSPNet and ICNet reduced the fault-extraction, but the small-scale macular edema was over-extracted. Table 3 showed the parameter configurations of different models and Table 4 showed the results of DME evaluation metrics. Compared with FCN, U-Net, PSPNet, and DANet models, DME-DeepLabV3+ achieved higher scores of PA, MPA, and FPS. As for Pre, Re, F1, and MIoU, DME-DeepLabV3+ substantially exceeded other models. Compared with ICNet, DME-DeepLabV3+ achieved a better trade-off in the accuracy and efficiency for DME extraction.

FIGURE 8
www.frontiersin.org

Figure 8. DME extraction results by different models Red, blue, and white DME regions represented TP regions, FP regions and FN regions.

TABLE 3
www.frontiersin.org

Table 3. Parameter configurations of different DME extraction models.

TABLE 4
www.frontiersin.org

Table 4. Evaluation metrics of DME extraction by different models.

6. Conclusion and discussion

With the increased incidence of diabetes, DME has become a major cause of visual impairment in diabetic patients (28). DME occurs as a result of the disruption of blood-retinal barrier and consequent increase in vascular permeability (29). OCT allows longitudinal, functional, and microstructural analysis of human macula (30). However, manual labeling DME is time-consuming and labor-intensive (31). Automatic extraction of DME based on machine learning can help physicians assess disease severity, determine treatment options, and improve life quality of patients (32). Thus, it is urgent to develop an efficient model for DME detection. In this study, we proposed a lightweight model based on DeepLabV3+, termed DME-DeepLabV3+, to extract DME in OCT images. MobileNetV2 architecture was used as the backbone to extract the low-level features of DME and reduce the model complexity to enhance DME detection accuracy. With the help of improved ASPP structure, DME-DeepLabV3+ avoided the grid effects and learned more local information. Finally, the decoder was used to fuse the low-level and high-level features of DME and refined the results of DME extraction.

OCT image modality has been widely used for detecting DME due to its non-invasive and high-resolution features. Considering the clinical characteristics that are present in OCT images such as thickness, reflectivity or intraretinal fluid accumulation, DMEs have been categorized into three different types: SRD, DRT, and CME. Traditional DME detection is based on the low-level hand-crafted features, which require significant domain knowledge and are sensitive to the variations of lesions. Given great variability of morphology, shape, and relative ME position, it is difficult to detect all three ME types simultaneously. Our proposed model can achieve automatic and simultaneous detection of all three types of ME (SRD, DRT, and CME) in the ophthalmological field. However, the accuracy of DRT detection is still not good as SRD or CME detection. DRT is characterized by a sponge-like retinal swelling of the macula with reduced intraretinal reflectivity. In addition, DRT is characterized by uniform thickening of inner retinal layers but without macroscopic optical empty spaces. Thus, further improvement of our proposed model is still required for enhancing the accuracy of the automatic detection of DRT edemas.

In clinical practice, layer segmentation and fluid area segmentation can provide qualitative information and visualization of retinal structure, which is important for DME assessment and monitoring. Although commercial OCT devices with on-board proprietary segmentation software are available, the definition of retinal boundaries varies between the manufacturers, making the quantitative retinal thickness difficult. In addition, proprietary software is difficult to be used for image analysis from other OCT devices, which poses a great challenge for effective diagnosis of DME (33). Although automated methods for layer segmentation have been proposed, most of them usually ignore the priority of mutually exclusive relationships between different layers, which can also affect the accuracy of DME assessment (34). In future study, we will improve our model to consider both layer segmentation and fluid area segmentation for better monitoring the progression of DME in retinal diseases.

Both microaneurysm (MA) formation and DME lesions are the important signs of DR. Early and accurate detection of DME and MAs can reduce the risk of DR. Due to the small size of MA lesions and low contrast between MA lesion and retinal background, automated MA detection is still challenging. Many imaging modalities have been used to detect MAs, including color fundus images, optical coherence tomography angiography (OCTA), and fluorescein fundus angiography (FFA). However, MAs are situated on the capillaries, which are not often visible in color fundus images. Although FFA can capture the small changes of retinal vessels, FFA is an invasive method compared with other imaging modalities. OCTA can provide the detailed visualization of vascular perfusion and allow for the examination of retinal vasculature in 3D (35). In future study, we would also improve our model by considering the segmentation of FFA for better monitoring the progression of DME in retinal diseases. We would design modules with better feature extraction capabilities, such as embedding attention mechanism to the model, strengthening key information, suppressing useless information, and better capturing contextual information, to improve the generalization of the model for the diagnosis of retinal diseases.

Data availability statement

The raw data supporting the conclusions of this article will be made available by the authors, without undue reservation.

Author contributions

BY, ZW, and QJ were responsible for the conceptualization and data collection. BY, ZW, and YB were responsible for the experiment design and manuscript writing. JL and LS conducted the data collection and data entry. BY and ZW were responsible for overall supervision and manuscript revision. All authors contributed to the article and approved the submitted version.

Funding

This research was generously supported by the grants from the National Natural Science Foundation of China (Grant Nos. 82171074 and 82070983).

Conflict of interest

The authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.

The reviewer ZH declared a shared parent affiliation with the authors LS, QJ, and BY to the handling editor at the time of review.

Publisher’s note

All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article, or claim that may be made by its manufacturer, is not guaranteed or endorsed by the publisher.

References

1. Schmidt-Erfurth, U, Garcia-Arumi, J, Bandello, F, Berg, K, Chakravarthy, U, Gerendas, BS, et al. Guidelines for the Management of Diabetic Macular Edema by the European Society of Retina Specialists (Euretina). Ophthalmologica. (2017) 237:185–222. doi: 10.1159/000458539

PubMed Abstract | CrossRef Full Text | Google Scholar

2. Steinmetz, JD, Bourne, RR, Briant, PS, Flaxman, SR, Taylor, HR, Jonas, JB, et al. Causes of blindness and vision impairment in 2020 and trends over 30 years, and prevalence of avoidable blindness in relation to vision 2020: the right to sight: an analysis for the global burden of disease study. Lancet Glob Health. (2021) 9:e144–60. doi: 10.1016/S2214-109X(20)30489-7

PubMed Abstract | CrossRef Full Text | Google Scholar

3. Tan, GS, Cheung, N, Simo, R, Cheung, GC, and Wong, TY. Diabetic Macular Oedema. Lancet Diabetes Endocrinol. (2017) 5:143–55. doi: 10.1016/S2213-8587(16)30052-3

CrossRef Full Text | Google Scholar

4. Das, A, McGuire, PG, and Rangasamy, S. Diabetic macular edema: pathophysiology and novel therapeutic targets. Ophthalmology. (2015) 122:1375–94. doi: 10.1016/j.ophtha.2015.03.024

PubMed Abstract | CrossRef Full Text | Google Scholar

5. Van Melkebeke, L, Barbosa-Breda, J, Huygens, M, and Stalmans, I. Optical coherence tomography angiography in Glaucoma: a review. Ophthalmic Res. (2018) 60:139–51. doi: 10.1159/000488495

PubMed Abstract | CrossRef Full Text | Google Scholar

6. Wu, Q, Zhang, B, Hu, Y, Liu, B, Cao, D, Yang, D, et al. Detection of morphologic patterns of diabetic macular edema using a deep learning approach based on optical coherence tomography images. Retina. 41:1110–7. doi: 10.1097/iae.0000000000002992

PubMed Abstract | CrossRef Full Text | Google Scholar

7. Wang, Z, Zhong, Y, Yao, M, Ma, Y, Zhang, W, Li, C, et al. Automated segmentation of macular edema for the diagnosis of ocular disease using deep learning method. Sci Rep. (2021) 11:1–12. doi: 10.1038/s41598-021-92458-8

PubMed Abstract | CrossRef Full Text | Google Scholar

8. Liu, X, Song, L, Liu, S, and Zhang, Y. A review of deep-learning-based medical image segmentation methods. Sustainability. (2021) 13:1224. doi: 10.3390/su13031224

CrossRef Full Text | Google Scholar

9. Makandar, A, and Halalli, B. Threshold based segmentation technique for mass detection in mammography. J Comput. (2016) 11:472–8. doi: 10.17706/jcp.11.6.472-478

CrossRef Full Text | Google Scholar

10. Zebari, DA, Zeebaree, DQ, Abdulazeez, AM, Haron, H, and Hamed, HNA. Improved threshold based and trainable fully automated segmentation for breast Cancer boundary and pectoral muscle in mammogram images. IEEE Access. (2020) 8:203097–116. doi: 10.1109/ACCESS.2020.3036072

CrossRef Full Text | Google Scholar

11. Biratu, ES, Schwenker, F, Debelee, TG, Kebede, SR, Negera, WG, and Molla, HT. Enhanced region growing for brain tumor Mr image segmentation. J Imaging. (2021) 7:22. doi: 10.3390/jimaging7020022

PubMed Abstract | CrossRef Full Text | Google Scholar

12. Liu, J, Yan, S, Lu, N, Yang, D, Fan, C, Lv, H, et al. Automatic segmentation of foveal avascular zone based on adaptive watershed algorithm in retinal optical coherence tomography angiography images. J Innov Opt Health Sci. (2022) 15:2242001. doi: 10.1142/S1793545822420019

CrossRef Full Text | Google Scholar

13. Chatterjee, S, Suman, A, Gaurav, R, Banerjee, S, Singh, AK, Ghosh, BK, et al. Retinal blood vessel segmentation using edge detection method. J Phys Conf Ser. (2021) 1717:012008. doi: 10.1088/1742-6596/1717/1/012008

CrossRef Full Text | Google Scholar

14. Long, J, Shelhamer, E, and Darrell, T. Fully convolutional networks for semantic segmentation. IEEE Trans Pattern Anal Mach Intell. (2017) 39:640–51. doi: 10.1109/cvpr.2015.7298965

CrossRef Full Text | Google Scholar

15. Ronneberger, O, Fischer, P, and Brox, T. U-net: convolutional networks for biomedical image segmentation. International conference on medical image computing and computer-assisted intervention. Berlin: Springer (2015).

Google Scholar

16. Zhao, H, Shi, J, Qi, X, Wang, X, and Jia, J. Pyramid scene parsing network. Proceedings of the IEEE conference on computer vision and pattern recognition. (2017).

Google Scholar

17. Ben-Cohen, A, Mark, D, Kovler, I, Zur, D, Barak, A, Iglicki, M, et al. Retinal layers segmentation using fully convolutional network in Oct images. Israel: RSIP Vision, 1–8. (2017).

Google Scholar

18. Ruan, Y, Xue, J, Li, T, Liu, D, Lu, H, Chen, M, et al. Multi-phase level set algorithm based on fully convolutional networks (Fcn-Mls) for retinal layer segmentation in Sd-Oct images with central serous Chorioretinopathy (Csc). Biomed Opt Express. (2019) 10:3987–4002. doi: 10.1364/BOE.10.003987

PubMed Abstract | CrossRef Full Text | Google Scholar

19. Orlando, JI, Seeböck, P, Bogunović, H, Klimscha, S, Grechenig, C, Waldstein, S, et al., U2-net: a Bayesian U-net model with epistemic uncertainty feedback for photoreceptor layer segmentation in pathological Oct scans. 2019 IEEE 16th International Symposium on Biomedical Imaging (ISBI 2019) (2019).

Google Scholar

20. Wu, J, Chen, J, Xiao, Z, and Geng, L. Automatic layering of retinal Oct images with dual attention mechanism. 2021 3rd international conference on intelligent medicine and image processing (2021).

Google Scholar

21. Chen, L-C, Zhu, Y, Papandreou, G, Schroff, F, and Adam, H. Encoder-decoder with Atrous separable convolution for semantic image segmentation. Proceedings of the European conference on computer vision (ECCV) (2018).

Google Scholar

22. Zhang, R, Du, L, Xiao, Q, and Liu, J. Comparison of backbones for semantic segmentation network. J Phys Conf Ser. (2020) 1544:012196. doi: 10.1088/1742-6596/1544/1/012196

CrossRef Full Text | Google Scholar

23. Sandler, M, Howard, A, Zhu, M, Zhmoginov, A, and Chen, L-C. Mobilenetv2: Inverted Residuals and Linear Bottlenecks. Proceedings of the IEEE conference on computer vision and pattern recognition (2018).

Google Scholar

24. Chen, L-C, Papandreou, G, Kokkinos, I, Murphy, K, and Yuille, AL. Deeplab: semantic image segmentation with deep convolutional nets, Atrous convolution, and fully connected Crfs. IEEE Trans Pattern Anal Mach Intell. (2017) 40:834–48. doi: 10.1109/tpami.2017.2699184

PubMed Abstract | CrossRef Full Text | Google Scholar

25. Wang, P, Chen, P, Yuan, Y, Liu, D, Huang, Z, Hou, X, et al. Understanding convolution for semantic segmentation. 2018 IEEE winter conference on applications of computer vision (WACV) (2018).

Google Scholar

26. Badrinarayanan, V, Kendall, A, and Cipolla, R. Segnet: a deep convolutional encoder-decoder architecture for image segmentation. IEEE Trans Pattern Anal Mach Intell. (2017) 39:2481–95. doi: 10.1109/tpami.2016.2644615

PubMed Abstract | CrossRef Full Text | Google Scholar

27. Kermany, DS, Goldbaum, M, Cai, W, Valentim, CC, Liang, H, Baxter, SL, et al. Identifying medical diagnoses and treatable diseases by image-based deep learning. Cells. (2018) 172:1122–1131.e9. doi: 10.1016/j.cell.2018.02.010

PubMed Abstract | CrossRef Full Text | Google Scholar

28. Kim, EJ, Lin, WV, Rodriguez, SM, Chen, A, Loya, A, and Weng, CY. Treatment of diabetic macular edema. Curr Diab Rep. (2019) 19:1–10. doi: 10.1007/s11892-019-1188-4

CrossRef Full Text | Google Scholar

29. Noma, H, Yasuda, K, and Shimura, M. Involvement of cytokines in the pathogenesis of diabetic macular edema. Int J Mol Sci. (2021) 22:3427. doi: 10.3390/ijms22073427

PubMed Abstract | CrossRef Full Text | Google Scholar

30. Daruich, A, Matet, A, Moulin, A, Kowalczuk, L, Nicolas, M, Sellam, A, et al. Mechanisms of macular edema: beyond the surface. Prog Retin Eye Res. (2018) 63:20–68. doi: 10.1016/j.preteyeres.2017.10.006

PubMed Abstract | CrossRef Full Text | Google Scholar

31. Pekala, M, Joshi, N, Liu, TA, Bressler, NM, DeBuc, DC, and Burlina, P. Deep learning based retinal Oct segmentation. Comput Biol Med. (2019) 114:103445. doi: 10.1016/j.compbiomed.2019.103445

CrossRef Full Text | Google Scholar

32. de Moura, J, Samagaio, G, Novo, J, Almuina, P, Fernández, MI, and Ortega, M. Joint diabetic macular edema segmentation and characterization in Oct images. J Digit Imaging. (2020) 33:1335–51. doi: 10.1007/s10278-020-00360-y

PubMed Abstract | CrossRef Full Text | Google Scholar

33. Alex, V, Motevasseli, T, Freeman, WR, Jayamon, JA, Bartsch, D-UG, and Borooah, S. Assessing the validity of a cross-platform retinal image segmentation tool in Normal and diseased retina. Sci Rep. (2021) 11:21784. doi: 10.21203/rs.3.rs-396609/v1

CrossRef Full Text | Google Scholar

34. Wei, H, and Peng, P. The segmentation of retinal layer and fluid in Sd-Oct images using Mutex Dice loss based fully convolutional networks. IEEE Access. (2020) 8:60929–39. doi: 10.1109/ACCESS.2020.2983818

CrossRef Full Text | Google Scholar

35. Hervella, ÁS, Rouco, J, Novo, J, and Ortega, M. Retinal microaneurysms detection using adversarial pre-training with unlabeled multimodal images. Inf Fusion. (2022) 79:146–61. doi: 10.1016/j.inffus.2021.10.003

CrossRef Full Text | Google Scholar

Keywords: diabetic macular edema, optical coherence tomography, deep learning, DeepLabV3+, extraction model

Citation: Bai Y, Li J, Shi L, Jiang Q, Yan B and Wang Z (2023) DME-DeepLabV3+: a lightweight model for diabetic macular edema extraction based on DeepLabV3+ architecture. Front. Med. 10:1150295. doi: 10.3389/fmed.2023.1150295

Received: 24 January 2023; Accepted: 25 August 2023;
Published: 08 September 2023.

Edited by:

Darren Shu Jeng Ting, University of Nottingham, United Kingdom

Reviewed by:

Zizhong Hu, Nanjing Medical University, China
Rizwan Ali Naqvi, Sejong University, Republic of Korea

Copyright © 2023 Bai, Li, Shi, Jiang, Yan and Wang. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) and the copyright owner(s) are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

*Correspondence: Biao Yan, yanbiao1982@hotmail.com; Zhenhua Wang, zh-wang@shou.edu.cn; Qin Jiang, jiangqin710@126.com

These authors have contributed equally to this work

Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.