Optics and Precision Engineering, Volume. 30, Issue 17, 2147(2022)

Automatic classification of retinopathy with attention ConvNeXt

Wenbo HUANG1,*... Yuxiang HUANG1, Yuan YAO2 and Yang YAN1 |Show fewer author(s)
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

    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.

    Tools

    Get Citation

    Copy Citation Text

    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

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Information Sciences

    Received: May. 31, 2022

    Accepted: --

    Published Online: Oct. 20, 2022

    The Author Email: HUANG Wenbo (huangwenbo@sina.com)

    DOI:10.37188/OPE.20223017.2147

    Topics