Prediction of antiepileptic drug treatment outcomes of patients with newly diagnosed epilepsy by machine learning

Abstract

ObjectiveThe objective of this study was to build a supervised machine learning-based classifier, which can accurately predict the outcomes of antiepileptic drug (AED) treatment of patients with newly diagnosed epilepsy. MethodsWe collected information from 287 patients with newly diagnosed epilepsy between 2009 and 2017 at the Second Affiliated Hospital of Zhejiang University. Patients were prospectively followed up for at least 3 years. A number of features, including demographic features, medical history, and auxiliary examinations (electroencephalogram [EEG] and magnetic resonance imaging [MRI]) are selected to distinguish patients with different remission outcomes. Seizure outcomes classified as remission and never remission. In addition, remission is further divided into early remission and late remission. Five classical machine learning algorithms, i.e., Decision Tree, Random Forest, Support Vector Machine, XGBoost, and Logistic Regression, are selected and trained by our dataset to get classification models. ResultsOur study shows that 1) compared with the other four algorithms, the XGBoost algorithm based machine learning model achieves the best prediction performance of the AED treatment outcomes between remission and never remission patients with an F1 score of 0.947 and an area under the curve (AUC) value of 0.979; 2) The best discriminative factor for remission and never remission patients is higher number of seizures before treatment (>3); 3) XGBoost-based machine learning model also offers the best prediction between early remission and later remission patients, with an F1 score of 0.836 and an AUC value of 0.918; 4) multiple seizure type has the highest dependence to the categories of early and late remission patients. SignificancesOur XGBoost-based machine learning classifier accurately predicts the most probable AED treatment outcome of a patient after he/she finishes all the standard examinations for the epilepsy disease. The classifier's prediction result could help disease guide counseling and eventually improve treatment strategies.