Semi-supervised segmentation of retinoblastoma tumors in fundus images

Amir Rahdar1, Mohamad Javad Ahmadi1, Masood Naseripour2

  • 1Chashmyar Company, Tehran, Iran.

Scientific Reports
|August 10, 2023
PubMed

Insights

This study introduces a semi-supervised machine learning model for early retinoblastoma detection. The model achieves 93% accuracy in segmenting ocular abnormalities from fundus images, aiding in early cancer diagnosis.

Area of Science:

  • Ophthalmology
  • Medical Imaging
  • Artificial Intelligence

Background:

  • Retinoblastoma is the primary intraocular malignancy in young children.
  • Early detection of retinoblastoma offers a >90% cure rate.
  • Accurate tumor segmentation is crucial for effective treatment planning.

Purpose of the Study:

  • To develop a semi-supervised machine learning model for retinoblastoma segmentation in fundus images.
  • To achieve segmentation accuracy comparable to expert ophthalmologists.
  • To create a cost-effective automated detection system.

Main Methods:

  • Utilized Gaussian mixture models for initial abnormality detection in ~4200 fundus images.
  • Trained a cost-effective model using initial detection results for improved efficiency.
  • Employed semi-supervised machine learning for image segmentation.

Main Results:

  • The proposed model achieved an average accuracy of 93% using the Sørensen-Dice coefficient.
  • Demonstrated high precision in extracting detailed tumor boundaries from fundus images.
  • The model provides results comparable to those of medical experts.

Conclusions:

  • Semi-supervised machine learning offers a viable approach for accurate retinoblastoma segmentation.
  • The developed model can assist in early diagnosis and treatment planning for retinoblastoma.
  • Automated segmentation can improve the efficiency and consistency of retinoblastoma detection.