机构:[1]Department of Biomedical Engineering, Beijing International Science and Technology Cooperation Base for Intelligent Physiological Measurement and Clinical Transformation, Beijing University of Technology, Beijing, China[2]Beijing Tongren Eye Center, Beijing Ophthalmology & Visual Sciences Key Lab, Beijing Tongren Hospital, Capital Medical University, Beijing, China首都医科大学附属北京同仁医院首都医科大学附属同仁医院[3]Fan Gongxiu Honors College, Beijing University of Technology, Beijing, China[4]Sports and Medicine Integrative Innovation Center, Capital University of Physical Education and Sports, Beijing, China[5]Department of Ophthalmology, Beijing Boai Hospital, China Rehabilitation Research Center, School of Rehabilitation Medicine, Capital Medical University, Beijing, China
Automatic multilabel classification of multiple fundus diseases is of importance for ophthalmologists. This study aims to design an effective multilabel classification model that can automatically classify multiple fundus diseases based on color fundus images.We proposed a multilabel fundus disease classification model based on a convolutional neural network to classify normal and seven categories of common fundus diseases. Specifically, an attention mechanism was introduced into the network to further extract information features from color fundus images. The fundus images with eight categories of labels were applied to train, validate, and test our model. We employed the validation accuracy, area under the receiver operating characteristic curve (AUC), and F1-score as performance metrics to evaluate our model.Our proposed model achieved better performance with a validation accuracy of 94.27%, an AUC of 85.80%, and an F1-score of 86.08%, compared to two state-of-the-art models. Most important, the number of training parameters has dramatically dropped by three and eight times compared to the two state-of-the-art models.This model can automatically classify multiple fundus diseases with not only excellent accuracy, AUC, and F1-score but also significantly fewer training parameters and lower computational cost, providing a reliable assistant in clinical screening.The proposed model can be widely applied in large-scale multiple fundus disease screening, helping to create more efficient diagnostics in primary care settings.
第一作者机构:[1]Department of Biomedical Engineering, Beijing International Science and Technology Cooperation Base for Intelligent Physiological Measurement and Clinical Transformation, Beijing University of Technology, Beijing, China
共同第一作者:
通讯作者:
通讯机构:[1]Department of Biomedical Engineering, Beijing International Science and Technology Cooperation Base for Intelligent Physiological Measurement and Clinical Transformation, Beijing University of Technology, Beijing, China[4]Sports and Medicine Integrative Innovation Center, Capital University of Physical Education and Sports, Beijing, China[5]Department of Ophthalmology, Beijing Boai Hospital, China Rehabilitation Research Center, School of Rehabilitation Medicine, Capital Medical University, Beijing, China[*1]Department of Biomedical Engineering, Beijing International Science and Technology Cooperation Base for Intelligent Physiological Measurement and Clinical Transformation, Beijing University of Technology, 100 Pingleyuan, Chaoyang District, Beijing 100124, China[*2]Sports and Medicine Integrative Innovation Center, Capital University of Physical Education and Sports, 11 North Third Ring West Road, Beijing 100191, China.
推荐引用方式(GB/T 7714):
Lu Zhenzhen,Miao Jingpeng,Dong Jingran,et al.Automatic Multilabel Classification of Multiple Fundus Diseases Based on Convolutional Neural Network With Squeeze-and-Excitation Attention[J].TRANSLATIONAL VISION SCIENCE & TECHNOLOGY.2023,12(1):22.doi:10.1167/tvst.12.1.22.
APA:
Lu Zhenzhen,Miao Jingpeng,Dong Jingran,Zhu Shuyuan,Wu Penghan...&Feng Jihong.(2023).Automatic Multilabel Classification of Multiple Fundus Diseases Based on Convolutional Neural Network With Squeeze-and-Excitation Attention.TRANSLATIONAL VISION SCIENCE & TECHNOLOGY,12,(1)
MLA:
Lu Zhenzhen,et al."Automatic Multilabel Classification of Multiple Fundus Diseases Based on Convolutional Neural Network With Squeeze-and-Excitation Attention".TRANSLATIONAL VISION SCIENCE & TECHNOLOGY 12..1(2023):22