ADS-Net: attention-awareness and deep supervision based network for automatic detection of retinopathy of prematurity

Yuanyuan Peng1, Zhongyue Chen1, Weifang Zhu1

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

Insights

A new deep learning model, ADS-Net, effectively detects and grades retinopathy of prematurity (ROP) in premature infants. This automated system aids ophthalmologists in diagnosing this severe eye condition, improving patient outcomes.

Area of Science:

  • Ophthalmology
  • Medical Imaging
  • Artificial Intelligence

Background:

  • Retinopathy of prematurity (ROP) is a severe ocular complication in premature infants.
  • Accurate ROP detection and grading are crucial for timely intervention.
  • Existing methods face challenges in learning discriminative features from fundus images due to data complexity.

Purpose of the Study:

  • To develop an automated system for ROP detection and grading.
  • To address limitations in feature representation for medical image analysis.
  • To improve the accuracy and efficiency of ROP diagnosis.

Main Methods:

  • A novel Attention-Aware and Deep Supervision network (ADS-Net) was proposed.
  • A Multi-Semantic Feature Aggregation (MsFA) module using self-attention was designed to handle image complexities.
  • Deeply supervised loss was employed to optimize the deep model training.

Main Results:

  • ADS-Net achieved a Kappa index of 0.9552 for ROP screening and 0.9037 for ROP grading on a per-image basis.
  • The proposed method demonstrated superior performance compared to state-of-the-art classification networks.
  • Experimental results validated the effectiveness of ADS-Net in ROP detection and grading tasks.

Conclusions:

  • The ADS-Net provides a safe, objective, and cost-effective approach for ROP diagnosis.
  • The integration of attention mechanisms and deep supervision enhances feature learning and model performance.
  • This automated system shows significant potential to assist ophthalmologists in managing ROP.