HCSP-Net:A Novel Model of Age-Related Macular Degeneration Classification Based on Color Fundus Photography

作     者：Cheng Wan Jiani Zhao Xiangqian Hong Weihua Yang Shaochong Zhang 

作者机构：College of Electronic and Information Engineering/College of Integrated CircuitsNanjing University of Aeronautics and AstronauticsNanjing211106China Shenzhen Eye InstituteShenzhen Eye HospitalJinan UniversityShenzhen518040China 

出 版 物：《Computers, Materials & Continua》 (计算机、材料和连续体（英文）)

年 卷 期：2024年第79卷第4期

页      面：391-407页

核心收录：

学科分类：08[工学] 0805[工学-材料科学与工程（可授工学、理学学位）] 0812[工学-计算机科学与技术（可授工学、理学学位）] 

基　　金：Shenzhen Fund for Guangdong Provincial High-Level Clinical Key Specialties(SZGSP014) Sanming Project of Medicine in Shenzhen(SZSM202311012) Shenzhen Science and Technology Planning Project(KCXFZ20211020163813019). 

主　　题：Computer-aided diagnosis deep learning age-related macular degeneration transformer 

摘      要：Age-related macular degeneration(AMD)ranks third among the most common causes of blindness.As the most conventional and direct method for identifying AMD,color fundus photography has become prominent owing to its consistency,ease of use,and good quality in extensive clinical practice.In this study,a convolutional neural network(CSPDarknet53)was combined with a transformer to construct a new hybrid model,HCSP-Net.This hybrid model was employed to tri-classify color fundus photography into the normal macula(NM),dry macular degeneration(DMD),and wet macular degeneration(WMD)based on clinical classification manifestations,thus identifying and resolving AMD as early as possible with color fundus photography.To further enhance the performance of this model,grouped convolution was introduced in this study without significantly increasing the number of parameters.HCSP-Net was validated using an independent test set.The average precision of HCSPNet in the diagnosis of AMD was 99.2%,the recall rate was 98.2%,the F1-Score was 98.7%,the PPV(positive predictive value)was 99.2%,and the NPV(negative predictive value)was 99.6%.Moreover,a knowledge distillation approach was also adopted to develop a lightweight student network(SCSP-Net).The experimental results revealed a noteworthy enhancement in the accuracy of SCSP-Net,rising from 94%to 97%,while remarkably reducing the parameter count to a quarter of HCSP-Net.This attribute positions SCSP-Net as a highly suitable candidate for the deployment of resource-constrained devices,which may provide ophthalmologists with an efficient tool for diagnosing AMD.