Development of a Machine Learning Model for the Classification of Enterobius vermicularis Egg

Natthanai Chaibutr1,2,3, Pongphan Pongpanitanont4, Sakhone Laymanivong5

  • 1Medical Innovation and Technology Program, School of Allied Health Sciences, Walailak University, Nakhon Si Thammarat 80160, Thailand.

Journal of Imaging
|September 27, 2024
PubMed

Insights

Automated detection of pinworm (Enterobius vermicularis) eggs using convolutional neural networks (CNNs) significantly improves diagnostic accuracy. Data augmentation enhanced CNN performance, offering a more efficient alternative to traditional methods.

Area of Science:

  • Medical diagnostics
  • Computational biology
  • Parasitology

Background:

  • Enterobius vermicularis infections are common globally, particularly in children.
  • Current diagnosis relies on the time-consuming scotch tape technique, requiring expert microscopic examination.
  • Automated detection using artificial intelligence offers potential for improved efficiency and accuracy.

Purpose of the Study:

  • To enhance the automated detection of Enterobius vermicularis eggs using convolutional neural networks (CNNs).
  • To benchmark a proposed CNN model against leading models for pinworm egg detection.
  • To evaluate the impact of data augmentation on diagnostic performance.

Main Methods:

  • Digitization and augmentation of 40,000 microscopic images of E. vermicularis eggs and artifacts.
  • Training and validation using an 80:20 split and five-fold cross-validation.
  • Development and benchmarking of a CNN model, including the Xception architecture.

Main Results:

  • The proposed CNN model achieved 90.0% accuracy, precision, recall, and F1-score after data augmentation.
  • Data augmentation improved model stability, increasing ROC-AUC from 0.77 to 0.97.
  • The Xception model demonstrated superior performance with 99.0% accuracy, precision, recall, and F1-score.

Conclusions:

  • Data augmentation and advanced CNN architectures significantly improve the accuracy and efficiency of E. vermicularis diagnosis.
  • AI-powered tools can provide a reliable and faster alternative to traditional pinworm detection methods.
  • The developed CNN models show comparable performance to larger models despite smaller file sizes.

Related Concept Videos