Real-Time Jaundice Detection in Neonates Based on Machine Learning Models

Abstract

Introduction: Despite the many attempts made by researchers to diagnose jaundice non-invasively using machine learning techniques, the low amount of data used to build their models remains the key factor limiting the performance of their models. Objective: To build a system to diagnose neonatal jaundice non-invasively based on machine learning algorithms created based on a dataset comprising 767 infant images using a computer device and a USB webcam. Methods: The first stage of the proposed system was to evaluate the performance of four machine learning algorithms, namely support vector machine (SVM), k nearest neighbor (k-NN), random forest (RF), and extreme gradient boost (XGBoost), based on a dataset of 767 infant images. The algorithm with the best performance was chosen as the classifying algorithm in the developed application. The second stage included designing an application that enables the user to perform jaundice detection for a patient under test with the minimum effort required by capturing the patient’s image using a USB webcam. Results: The obtained results of the first stage of the machine learning algorithms evaluation process indicated that XGBoost outperformed the rest of the algorithms by obtaining an accuracy of 99.63%. The second-best algorithm was the RF algorithm, which had an accuracy of 98.99%. Following RF, with a slight difference, was the k-NN algorithm. It achieved an accuracy of 98.25%. SVM scored the lowest performance among the above three algorithms, with an accuracy of 96.22%. Based on these obtained results, the XGBoost algorithm was chosen to be the classifier of the proposed system. In the second stage, the jaundice application was designed based on the model created by the XGBoost algorithm. This application ensured it was user friendly with as fast a processing time as possible. Conclusion: Early detection of neonatal jaundice is crucial due to the severity of its complications. A non-invasive system using a USB webcam and an XGBoost machine learning technique was proposed. The XGBoost algorithm achieved 99.63% accuracy and successfully diagnosed 10 out of 10 NICU infants with very little processing time. This denotes the efficiency of machine learning algorithms in healthcare in general and in monitoring systems specifically.