Application of a Machine Learning Algorithm in Prediction of Abusive Head Trauma in Children

Area of Science:

• Medical Informatics
• Pediatric Emergency Medicine
• Machine Learning in Healthcare

Background:

• Abusive head trauma (AHT) is a serious concern in pediatric care.
• Timely detection of AHT is crucial for intervention and improving outcomes.
• Current diagnostic methods can be time-consuming and may miss subtle indicators.

Purpose of the Study:

• To develop and evaluate a machine learning algorithm for early detection of potential abusive head trauma (AHT).
• To utilize free-text physician notes and demographic data for AHT identification.
• To assess the algorithm's performance in distinguishing AHT cases from other head injuries.

Main Methods:

• Collected first free-text physician notes and demographic data for children under 5.
• Compared AHT cases (diagnosed by Child Protective Team) with a control group (head/neck injury).
• Developed composite scores (differential, sentiment, subjectivity) and trained a Random Forest algorithm with demographic data.

Main Results:

• The Random Forest model combining composite scores and demographics achieved an 84% accuracy.
• Area under the curve (AUC) improved from 0.68 to 0.78 when incorporating demographic data.
• Key predictors included composite score, sentiment, age, and subjectivity; subjectivity trended higher in AHT cases.

Conclusions:

• Machine learning effectively identifies patterns in clinical notes and demographics indicative of AHT.
• This approach offers a promising tool for timely and accurate AHT detection in children.
• Further integration of such algorithms can enhance pediatric patient safety and care.