Diabetic Retinopathy Assessment through Multitask Learning Approach on Heterogeneous Fundus Image Datasets

基于异构眼底图像数据集的多任务学习方法对糖尿病视网膜病变进行评估

阅读:2

Abstract

OBJECTIVE: To develop and validate an artificial intelligence (AI)-based system, Diabetic Retinopathy Analysis Model Assistant (DRAMA), for diagnosing diabetic retinopathy (DR) across multisource heterogeneous datasets and aimed at improving the diagnostic accuracy and efficiency. DESIGN: This was a cross-sectional study conducted at Zhejiang University Eye Hospital and approved by the ethics committee. SUBJECTS: The study included 1500 retinal images from 957 participants aged 18 to 83 years. The dataset was divided into 3 subdatasets: color fundus photography, ultra-widefield imaging, and portable fundus camera. Images were annotated by 3 experienced ophthalmologists. METHODS: The AI system was built using EfficientNet-B2, pretrained on the ImageNet dataset. It performed 11 multilabel tasks, including image type identification, quality assessment, lesion detection, and diabetic macular edema (DME) detection. The model used LabelSmoothingCrossEntropy and AdamP optimizer to enhance robustness and convergence. The system's performance was evaluated using metrics such as accuracy, sensitivity, specificity, and area under the curve (AUC). External validation was conducted using datasets from different clinical centers. MAIN OUTCOME MEASURES: The primary outcomes measured were the accuracy, sensitivity, specificity, and AUC of the AI system in diagnosing DR. RESULTS: After excluding 218 poor-quality images, DRAMA demonstrated high diagnostic accuracy, with EfficientNet-B2 achieving 87.02% accuracy in quality assessment and 91.60% accuracy in lesion detection. Area under the curves were >0.95 for most tasks, with 0.93 for grading and DME detection. External validation showed slightly lower accuracy in some tasks but outperformed in identifying hemorrhages and DME. Diabetic Retinopathy Analysis Model Assistant diagnosed the entire test set in 86 ms, significantly faster than the 90 to 100 minutes required by humans. CONCLUSIONS: Diabetic Retinopathy Analysis Model Assistant, an AI-based multitask model, showed high potential for clinical integration, significantly improving the diagnostic efficiency and accuracy, particularly in resource-limited settings. FINANCIAL DISCLOSURES: The author(s) have no proprietary or commercial interest in any materials discussed in this article.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。