Tiny CNN-Based Supervised Contrastive Learning for Seizure Prediction

Automatic seizure prediction based on ElectroEncephaloGraphy (EEG) ensures the safety of patients with epilepsy and mitigates anxiety. In recent years, significant progress has been made in this field. However, the predictive performance of existing methods encounters a bottleneck that is difficult to overcome. Moreover, there are certain limitations such as significant differences in prediction efficacy among patients or intricate model structures. Given these considerations, Siamese Network (SiaNet) and Triplet Network (TriNet) are proposed based on tiny convolutional neural network and supervised contrastive learning. Short-Time Fourier Transform (STFT) is first applied to the pre-processed data. Then data tuples are constructed and fed into the networks for training. Both networks try to minimize the interval between samples of the same class while maximize the interval between samples of different classes. The two networks consist of multiple branches with shared weights, which can learn from each other via contrastive learning. Promising results are obtained on the CHB-MIT and Siena datasets, with a total of 35 patients. Meanwhile, both models have only 19.351K parameters.