• Laser & Optoelectronics Progress
  • Vol. 57, Issue 24, 241025 (2020)
Tianfu Zhang1, Shuncong Zhong1、*, Chaoming Lian1, Ning Zhou1, and Maosong Xie2
Author Affiliations
  • 1School of Mechanical Engineering and Automation, Fuzhou University, Fuzhou, Fujian 350108, China
  • 2First Affiliated Hospital of Fujian Medical University, Fuzhou, Fujian 350000, China
  • show less
    DOI: 10.3788/LOP57.241025 Cite this Article Set citation alerts
    Tianfu Zhang, Shuncong Zhong, Chaoming Lian, Ning Zhou, Maosong Xie. Deep Learning Feature Fusion-Based Retina Image Classification[J]. Laser & Optoelectronics Progress, 2020, 57(24): 241025 Copy Citation Text show less
    Convolutional neural network model
    Fig. 1. Convolutional neural network model
    Convolutional layer structure comparison. (a) Ordinary convolution; (b) depth separable convolution
    Fig. 2. Convolutional layer structure comparison. (a) Ordinary convolution; (b) depth separable convolution
    Sample graphs of the dataset. (a) CNV; (b) DME; (c) DRUSEN; (d) NORMAL
    Fig. 3. Sample graphs of the dataset. (a) CNV; (b) DME; (c) DRUSEN; (d) NORMAL
    Image preprocessing. (a) Original image of OCT retina; (b)mean-shift removed speckle image
    Fig. 4. Image preprocessing. (a) Original image of OCT retina; (b)mean-shift removed speckle image
    Validation result curves. (a) Validation accuracy curve; (b) validation loss curve
    Fig. 5. Validation result curves. (a) Validation accuracy curve; (b) validation loss curve
    Confusion matrix. (a) Confusion matrix without weighted loss function; (b) confusion matrix with weighted loss function
    Fig. 6. Confusion matrix. (a) Confusion matrix without weighted loss function; (b) confusion matrix with weighted loss function
    Visual heat maps
    Fig. 7. Visual heat maps
    Confusion matrix. (a) Confusion matrix for GAPNet model; (b) confusion matrix for RongheNet model
    Fig. 8. Confusion matrix. (a) Confusion matrix for GAPNet model; (b) confusion matrix for RongheNet model
    Image typeNumber of images
    TrainValidationTestTotal
    CNV260437441372137205
    DME79432270113511348
    DRUSEN603117238628616
    NORMAL184205263263226315
    Total5843716697835083484
    Table 1. Division of OCT dataset
    Network typeCategory weightAccuracyMAPrecisionRecall
    SlimNetNo89.982.588.082.0
    SlimNetYes90.287.285.087.0
    BnL2NetYes93.091.692.093.0
    GAPNetYes94.993.392.093.0
    RongheNetYes97.297.095.097.0
    Table 2. Comparison of classification indexes of different training methods unit: %
    MethodMAPrecisionRecall
    Wang et al[6]89.690.888.6
    Bhowmik et al[7]92.694.094.0
    Yu et al[8]94.194.690.6
    Ours97.095.097.0
    Table 3. Performance comparison of different models unit: %
    Tianfu Zhang, Shuncong Zhong, Chaoming Lian, Ning Zhou, Maosong Xie. Deep Learning Feature Fusion-Based Retina Image Classification[J]. Laser & Optoelectronics Progress, 2020, 57(24): 241025
    Download Citation