Scientific Reports (Jun 2024)

Addressing data limitations in seizure prediction through transfer learning

  • Fábio Lopes,
  • Mauro F. Pinto,
  • António Dourado,
  • Andreas Schulze-Bonhage,
  • Matthias Dümpelmann,
  • César Teixeira

DOI
https://doi.org/10.1038/s41598-024-64802-1
Journal volume & issue
Vol. 14, no. 1
pp. 1 – 11

Abstract

Read online

Abstract According to the literature, seizure prediction models should be developed following a patient-specific approach. However, seizures are usually very rare events, meaning the number of events that may be used to optimise seizure prediction approaches is limited. To overcome such constraint, we analysed the possibility of using data from patients from an external database to improve patient-specific seizure prediction models. We present seizure prediction models trained using a transfer learning procedure. We trained a deep convolutional autoencoder using electroencephalogram data from 41 patients collected from the EPILEPSIAE database. Then, a bidirectional long short-term memory and a classifier layers were added on the top of the encoder part and were optimised for 24 patients from the Universitätsklinikum Freiburg individually. The encoder was used as a feature extraction module. Therefore, its weights were not changed during the patient-specific training. Experimental results showed that seizure prediction models optimised using pretrained weights present about four times fewer false alarms while maintaining the same ability to predict seizures and achieved more 13% validated patients. Therefore, results evidenced that the optimisation using transfer learning was more stable and faster, saving computational resources. In summary, adopting transfer learning for seizure prediction models represents a significant advancement. It addresses the data limitation seen in the seizure prediction field and offers more efficient and stable training, conserving computational resources. Additionally, despite the compact size, transfer learning allows to easily share data knowledge due to fewer ethical restrictions and lower storage requirements. The convolutional autoencoder developed in this study will be shared with the scientific community, promoting further research.