Supervised Contrastive Learning Framework for Electroencephalography-based Air-writing Recognition

TL;DR

This paper introduces a supervised contrastive learning framework that significantly enhances EEG-based air-writing recognition accuracy by improving neural signal representations across different architectures.

Contribution

The study demonstrates the effectiveness of supervised contrastive learning in boosting EEG-based air-writing recognition accuracy over traditional methods.

Findings

Supervised contrastive learning improves classification accuracy for EEG-based air-writing.

EEGNet and DeepConvNet architectures benefit from contrastive learning, with accuracy increases of approximately 10-15%.

Higher accuracy is achieved using ICA-derived neural components compared to raw EEG signals.

Abstract

Electroencephalography (EEG) - based air-writing recognition offers a human-computer interaction paradigm by decoding neural activity associated with handwriting movements. Despite its potential, reliable EEG-based air-writing recognition remains challenging due to low signal-to-noise ratio and pronounced inter-subject variability. In this study, we examine the use of supervised contrastive learning to improve representation learning for EEG-based air-writing recognition. The analysis is conducted on preprocessed EEG signals and independent component analysis (ICA)-derived neural components obtained from five participants, with trials segmented from -1 to 2 s relative to movement on-set. EEGNet and DeepConvNet architectures are evaluated under both conventional cross-entropy training and a supervised contrastive learning framework using a subject-dependent five-fold cross-validation scheme. The results indicate that supervised contrastive learning consistently improves classification accuracy across architectures and feature representations. For preprocessed EEG signals, the mean accuracy increases from 33.45% to 43.77% and from 29.14% to 38.06% with EEGNet and DeepConvNet, respectively. Using ICA components, higher mean accuracies of 49.21% and 43.32% are achieved with EEGNet and DeepConvNet, respectively. These results suggest that the supervised contrastive learning framework offers an efficient extension to existing EEG-based air-writing recognition approaches.