A roadmap for applying machine learning when working with privacy-sensitive data: predicting non-response to treatment for eating disorders

Abstract

ABSTRACT Objectives Applying machine-learning methodology to clinical data could present a promising avenue for predicting outcomes in patients receiving treatment for psychiatric disorders. However, preserving privacy when working with patient data remains a critical concern. Methods In showcasing how machine-learning can be used to build a clinically relevant prediction model on clinical data, we apply two commonly used machine-learning algorithms (Random Forest and least absolute shrinkage and selection operator) to routine outcome monitoring data collected from 593 patients with eating disorders to predict absence of reliable improvement 12 months after entering outpatient treatment. Results An RF model trained on data collected at baseline and after three months made 31.3% fewer errors in predicting lack of reliable improvement at 12 months, in comparison with chance. Adding data from a six-month follow-up resulted in only marginal improvements to accuracy. Conclusion We were able to build and validate a model that could aid clinicians and researchers in more accurately predicting treatment response in patients with EDs. We also demonstrated how this could be done without compromising privacy. ML presents a promising approach to developing accurate prediction models for psychiatric disorders such as ED.