Epileptic seizure prediction: Transfer learning with compact AI models

Research output: Contribution to book/Conference proceedings/Anthology/ReportConference contributionContributedpeer-review

Abstract

Objective: This study investigates the feasibility to apply transfer learning to develop compact, end-to-end algorithms optimized for the permanent, long-term operation of systems for the prediction of epileptic seizures. Methods: Thirteen artificial neural networks (ANNs), originally pretrained on ImageNet for image classification - including six compact models and seven milestone architectures - were repurposed for seizure prediction using two benchmark EEG datasets. The performance of these transferred models was evaluated against ANNs trained from scratch and leading state-of-the-art algorithms. Results: Transfer learning based models significantly outperformed models trained from scratch. Remarkably, lightweight models such as Shufflenet with only 1.4 million parameters and 50 million multiply-accumulate operations, achieved state-of-the-art performances. An input converter was introduced, expanding the effective receptive field and significantly enhancing prediction performance, particularly in compact models. Moreover, transfer learning algorithms performed effectively in challenging cases where EEG-specific ANNs and feature-based baselines failed, demonstrating their ability to capture features beyond domain-specific knowledge. Conclusion: Transfer learning is a promising strategy for developing compact and efficient algorithms well-suited for long-term seizure prediction. Significance: This research creates opportunities to advance seizure prediction technology beyond current methods.

Details

Original languageEnglish
Title of host publicationAICAS 2025 - 2025 7th IEEE International Conference on Artificial Intelligence Circuits and Systems, Proceedings
PublisherInstitute of Electrical and Electronics Engineers (IEEE)
Number of pages5
ISBN (electronic)979-8-3315-2424-1
ISBN (print)979-8-3315-2425-8
Publication statusPublished - 2025
Peer-reviewedYes

Publication series

SeriesIEEE International Conference on Artificial Intelligence Circuits and Systems (AICAS)
ISSN2834-9830

Conference

Title7th IEEE International Conference on Artificial Intelligence Circuits and Systems
Abbreviated titleIEEE AICAS 2025
Conference number7
Duration28 - 30 April 2025
Website
LocationUniversité de Bordeaux
CityBordeaux
CountryFrance

External IDs

ORCID /0000-0001-7436-0103/work/201620651