Machine learning for predicting medical outcomes associated with acute lithium poisoning.

Journal: Scientific reports
PMID:

Abstract

The use of machine learning algorithms and artificial intelligence in medicine has attracted significant interest due to its ability to aid in predicting medical outcomes. This study aimed to evaluate the effectiveness of the random forest algorithm in predicting medical outcomes related to acute lithium toxicity. We analyzed cases recorded in the National Poison Data System (NPDS) between January 1, 2014, and December 31, 2018. We highlighted instances of acute lithium toxicity in patients with ages ranging from 0 to 89 years. A random forest model was employed to predict serious medical outcomes, including those with a major effect, moderate effect, or death. Predictions were made using the pre-defined NPDS coding criteria. The model's predictive performance was assessed by computing accuracy, recall (sensitivity), and F1-score. Of the 11,525 reported cases of lithium poisoning documented during the study, 2,760 cases were categorized as acute lithium overdose. One hundred thirty-nine individuals experienced severe outcomes, whereas 2,621 patients endured minor outcomes. The random forest model exhibited exceptional accuracy and F1-scores, achieving values of 99%, 98%, and 98% for the training, validation, and test datasets, respectively. The model achieved an accuracy rate of 100% and a sensitivity rate of 96% for important results. In addition, it achieved a 96% accuracy rate and a sensitivity rate of 100% for minor outcomes. The SHapley Additive exPlanations (SHAP) study found factors, including drowsiness/lethargy, age, ataxia, abdominal pain, and electrolyte abnormalities, significantly influenced individual predictions. The random forest algorithm achieved a 98% accuracy rate in predicting medical outcomes for patients with acute lithium intoxication. The model demonstrated high sensitivity and precision in accurately predicting significant and minor outcomes. Further investigation is necessary to authenticate these findings.

Authors

  • Omid Mehrpour
    Data Science Institute, Southern Methodist University, Dallas, TX, USA. omehrpour@smu.edu.
  • Varun Vohra
    Department of Emergency Medicine, Michigan Poison & Drug Information Center, Wayne State University School of Medicine, Detroit, MI, USA.
  • Samaneh Nakhaee
    Medical Toxicology and Drug Abuse Research Center (MTDRC), Birjand University of Medical Sciences (BUMS), Birjand, Iran.
  • Seyed Ali Mohtarami
    Department of Computer Engineering and Information Technology (PNU), Tehran, Iran.
  • Farshad M Shirazi
    Arizona Poison & Drug Information Center, the University of Arizona, College of Pharmacy and University of Arizona, College of Medicine, Tucson, AZ, USA.