Application of transformers for predicting epilepsy treatment response
Choong, J. H.; Hakeem, H.; Chen, Z.; Brodie, M.; Lawn, N.; Drummond, T.; Kwan, P.; Ge, Z.
Show abstract
There is growing interest in machine learning based approaches to assist clinicians in treatment selection. In the treatment of epilepsy, a common neurological disorder that affects 70 million people worldwide, previous research has employed scoring methods generated from traditional machine learning methods based on pre-treatment patient characteristics to classify those with drug-resistant epilepsy (DRE). In this study, we used an attention-based approach in predicting the response to different antiseizure medications (ASMs) in individuals with newly diagnosed epilepsy. By applying a conventional transformer to model the patients response, we can use the predicted probability to determine the success rate of specific ASMs. Applying the transformer allowed the model to place attention on patient information and past treatments to model future drug responses. We trained a conventional transformer model based on one cohort of 1536 patients with newly diagnosed epilepsy, compared its performance with other trained models using RNN and LSTM, and applied it to a validation cohort of 736 patients. In the development cohort, the transformer model showed the highest accuracy (81%) and AUC (0.85), and maintained similar accuracy and AUC (74% and 0.79, respectively) in the validation cohort.
Matching journals
The top 7 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
- Evaluating the generalisability of region-naïve machine learning algorithms for the identification of epilepsy in low-resource settings 94%
- Longitudinally Tracking Personal Physiomes for Precision Management of Childhood Epilepsy 91%
- Enhancing Fairness in Disease Prediction by Optimizing Multiple Domain Adversarial Networks 91%
Similar papers in this journal
Similar papers in this journal
- NLP-based tools for localization of the Epileptogenic Zone in patients with drug-resistant focal epilepsy 96%
- Dynamic network properties of the interictal brain determine whether seizures appear focal or generalised 94%
- Dynamic Multiday Seizure Cycles in a Tetanus Toxin Rat Model of Epilepsy: Evolving Rhythms and Implications for Prediction 92%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.