Label Alignment Improves EEG-based Machine Learning-based Classification of Traumatic Brain Injury.
Journal:
Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual International Conference
Published Date:
Jul 1, 2022
Abstract
Machine learning and deep learning algorithms have paved the way for improved analysis of biomedical data which has led to a better understanding of various biological conditions. However, one major hindrance to leveraging the potential of machine learning models is the requirement of huge datasets. In the biomedical domain, this becomes extremely difficult due to uncertainties in collecting high-quality data as well as, in the case of human subjects data, privacy. Further, when it comes to biomedical data, inter-subject variability has been a long-entrenched issue. The data obtained from different individuals will differ to a considerable extent that it becomes difficult to find population differences in small datasets. In this work, we investigate the use of label alignment techniques on an EEG-based Traumatic Brain Injury (TBI) classification task to overcome inter-subject variability, thereby increasing the classification accuracy. We show an increase in accuracy of around 6% in some cases as compared to our previous results. In the end, we also propose a methodology to incorporate TBI data from a different species (e.g., mice) after domain adaptation, which might further improve the performance by increasing the amount of training datasets available for the classification model.