Automatic zoning for retinopathy of prematurity with semi-supervised feature calibration adversarial learning

Yuanyuan Peng1, Zhongyue Chen1, Weifang Zhu1

  • 1MIPAV Lab, School of Electronics and Information Engineering, Soochow University, Suzhou, Jiangsu 215006, China.

Insights

This study introduces a new semi-supervised learning network for retinopathy of prematurity (ROP) zoning, improving diagnosis accuracy for this leading cause of childhood blindness. The novel approach effectively uses unlabeled data to enhance ROP classification.

Area of Science:

  • Ophthalmology
  • Medical Imaging
  • Artificial Intelligence

Background:

  • Retinopathy of prematurity (ROP) is a significant cause of childhood blindness in premature infants.
  • Current automated ROP diagnosis primarily focuses on screening and disease presence, with limited research on ROP zoning.
  • High-quality annotations for ROP zoning are scarce, hindering model development and effective utilization of unlabeled data.

Purpose of the Study:

  • To develop a novel semi-supervised feature calibration adversarial learning network (SSFC-ALN) for accurate 3-level ROP zoning.
  • To address the challenge of limited labeled data in ROP diagnosis by leveraging unlabeled data through semi-supervised learning.
  • To improve the classification performance of ROP zoning by incorporating an attention-based feature calibration module.

Main Methods:

  • Proposed a semi-supervised feature calibration adversarial learning network (SSFC-ALN) comprising a generative U-net and a compound network with a discriminator.
  • Integrated an attention mechanism-based feature calibration module (FCM) into the compound network to enhance focus on relevant image features.
  • Employed a 3-fold cross-validation strategy on a dataset of 1013 fundus images from 108 patients.

Main Results:

  • The SSFC-ALN achieved high classification performance for 3-level ROP zoning.
  • The proposed method demonstrated superior results compared to other state-of-the-art classification techniques.
  • The integration of the FCM and adversarial learning effectively improved the diagnostic accuracy.

Conclusions:

  • The developed SSFC-ALN is a promising approach for automated ROP zoning, particularly in scenarios with limited labeled data.
  • The study highlights the effectiveness of semi-supervised learning and attention mechanisms in improving ROP diagnosis.
  • This work contributes to advancing objective and accurate assessment of ROP severity through automated fundus image analysis.

Related Concept Videos