Open Access iconOpen Access

ARTICLE

crossmark

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

Cheng Wan1, Jiani Zhao1, Xiangqian Hong2, Weihua Yang2,*, Shaochong Zhang2,*

1 College of Electronic and Information Engineering/College of Integrated Circuits, Nanjing University of Aeronautics and Astronautics, Nanjing, 211106, China
2 Shenzhen Eye Institute, Shenzhen Eye Hospital, Jinan University, Shenzhen, 518040, China

* Corresponding Authors: Weihua Yang. Email: email; Shaochong Zhang. Email: email

(This article belongs to the Special Issue: Deep Learning in Computer-Aided Diagnosis Based on Medical Image)

Computers, Materials & Continua 2024, 79(1), 391-407. https://doi.org/10.32604/cmc.2024.048307

Abstract

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 HCSP-Net 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.

Keywords


Cite This Article

APA Style
Wan, C., Zhao, J., Hong, X., Yang, W., Zhang, S. (2024). Hcsp-net: A novel model of age-related macular degeneration classification based on color fundus photography. Computers, Materials & Continua, 79(1), 391-407. https://doi.org/10.32604/cmc.2024.048307
Vancouver Style
Wan C, Zhao J, Hong X, Yang W, Zhang S. Hcsp-net: A novel model of age-related macular degeneration classification based on color fundus photography. Comput Mater Contin. 2024;79(1):391-407 https://doi.org/10.32604/cmc.2024.048307
IEEE Style
C. Wan, J. Zhao, X. Hong, W. Yang, and S. Zhang "HCSP-Net: A Novel Model of Age-Related Macular Degeneration Classification Based on Color Fundus Photography," Comput. Mater. Contin., vol. 79, no. 1, pp. 391-407. 2024. https://doi.org/10.32604/cmc.2024.048307



cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 240

    View

  • 214

    Download

  • 0

    Like

Share Link