Optics and Precision Engineering, Volume. 30, Issue 17, 2147(2022)
Automatic classification of retinopathy with attention ConvNeXt
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.
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
Category: Information Sciences
Received: May. 31, 2022
Accepted: --
Published Online: Oct. 20, 2022
The Author Email: HUANG Wenbo (huangwenbo@sina.com)