A Mutual Information-Based Many-Objective Optimization Method for EEG Channel Selection in the Epileptic Seizure Prediction Task

Abstract

Epileptic seizure prediction using multi-channel electroencephalogram (EEG) signals is very important in clinical therapy. A large number of channels lead to high computational complexity with low model performance. To improve the performance and reduce the overfitting that arises due to the use of unrelevant channels, the present paper proposed a channel selection method to study the brain region activation related to epileptic seizure. Our method is bio-inspired and cognitive since it integrates the novel binary many-objective particle swarm optimization with a ConvLSTM model. The proposed method has two advantages. First, it performed a new initialization strategy based on channel weighting with mutual information, thereby promoting the fast convergence of the optimization algorithm. Second, it captures spatio-temporal information from raw EEG segments thanks to the ConvLSTM model. The selected sub-channels are optimized as many-objective optimization problem that includes maximizing F1-score, sensitivity, specificity, and minimizing the ratio rate of selected channels. Our results have shown a performance of up to \(97.94\%\) with only one EEG channel. Interestingly, when using all the EEG channels available, lower performance was achieved compared to the case when EEG channels were selected by our approach. This study revealed that it is possible to predict epileptic seizures using a few channels, which provides evidence for the future development of portable EEG seizure prediction devices.