• Optics and Precision Engineering
  • Vol. 30, Issue 17, 2147 (2022)
Wenbo HUANG1,*, Yuxiang HUANG1, Yuan YAO2, and Yang YAN1
Author Affiliations
  • 1School of Computer Science and Technology, Changchun Normal University, Changchun30032, China
  • 2Bureau of Major Tasks, Chinese Academy of Sciences, Beijing100864, China
  • show less
    DOI: 10.37188/OPE.20223017.2147 Cite this Article
    Wenbo HUANG, Yuxiang HUANG, Yuan YAO, Yang YAN. Automatic classification of retinopathy with attention ConvNeXt[J]. Optics and Precision Engineering, 2022, 30(17): 2147 Copy Citation Text show less

    Abstract

    Due to the small differences in image features between classes and the relative fuzzy classification threshold of retinopathy, automatic classification algorithms are challenged by problems related to low recognition and classification accuracy. This paper proposes an automatic classification model for retinopathy based on an improved ConvNeXt network. Aiming at solving the problem of insufficient data in the data set, the horizontal flip left and right transformation method is used to expand the data, and related data sets are introduced to balance data distribution. To solve problems related to image blurring and uneven illumination in the fundus image, the Graham method was used to predict the image. The characteristics of the lesions are also highlighted. In this paper, an attention-fused ConvNeXt network was proposed to assist doctors in diagnosing retinopathy, an efficient channel attention mechanism was introduced, and an E-Block module was designed to channel interaction information while avoiding dimensionality reduction. The transfer learning method was used to train all layer parameters of the network, and the dropout method was added to avoid the overfitting problem caused by the strong learning ability of the ConvNeXt network. The results show that the sensitivity, specificity, and accuracy of the proposed model are 95.20%, 98.80%, and 95.21%, respectively. Compared with the ConvNeXt and other networks, the performance indexes of this network model for automatic classification of retinopathy.
    Wenbo HUANG, Yuxiang HUANG, Yuan YAO, Yang YAN. Automatic classification of retinopathy with attention ConvNeXt[J]. Optics and Precision Engineering, 2022, 30(17): 2147
    Download Citation