Score: 0

Accuracy-Constrained CNN Pruning for Efficient and Reliable EEG-Based Seizure Detection

Published: September 5, 2025 | arXiv ID: 2509.05190v1

By: Mounvik K, N Harshit

Potential Business Impact:

Finds seizures faster with smaller, smarter computer brain.

Business Areas:
Image Recognition Data and Analytics, Software

Deep learning models, especially convolutional neural networks (CNNs), have shown considerable promise for biomedical signals such as EEG-based seizure detection. However, these models come with challenges, primarily due to their size and compute requirements in environments where real-time detection or limited resources are available. In this study, we present a lightweight one-dimensional CNN model with structured pruning to improve efficiency and reliability. The model was trained with mild early stopping to address possible overfitting, achieving an accuracy of 92.78% and a macro-F1 score of 0.8686. Structured pruning of the baseline CNN involved removing 50% of the convolutional kernels based on their importance to model predictions. Surprisingly, after pruning the weights and memory by 50%, the new network was still able to maintain predictive capabilities, while modestly increasing precision to 92.87% and improving the macro-F1 score to 0.8707. Overall, we present a convincing case that structured pruning removes redundancy, improves generalization, and, in combination with mild early stopping, achieves a promising way forward to improve seizure detection efficiency and reliability, which is clear motivation for resource-limited settings.

Page Count
10 pages

Category
Computer Science:
Machine Learning (CS)