Domain adaptation with optimal transport improves EEG sleep stage classifiers - Archive ouverte HAL Access content directly
Conference Papers Year : 2018

Domain adaptation with optimal transport improves EEG sleep stage classifiers

Abstract

—Low sample size and the absence of labels on certain data limits the performances of predictive algorithms. To overcome this problem, it is sometimes possible to learn a model on a large labeled auxiliary dataset. Yet, this assumes that the two datasets exhibit similar statistical properties which is rarely the case in practice: there is a discrepancy between the large dataset, called the source, and the dataset of interest, called the target. Improving the prediction performance on the target domain by reducing the distribution discrepancy, between the source and the target domains, is known as Domain Adaptation (DA). Presently, Optimal transport DA (OTDA) methods yield state-of-the-art performances on several DA problems. In this paper, we consider the problem of sleep stage classification, and use OTDA to improve the performances of a convolutional neural network. We use features learnt from the electroencephalogram (EEG) and the electrooculogram (EOG) signals. Our results demonstrate that the method significantly improves the network predictions on the target data.
Fichier principal
Vignette du fichier
main.pdf (246.05 Ko) Télécharger le fichier
Origin Files produced by the author(s)
Loading...

Dates and versions

hal-01814190 , version 1 (13-06-2018)

Identifiers

  • HAL Id : hal-01814190 , version 1

Cite

Stanislas Chambon, Mathieu N. Galtier, Alexandre Gramfort. Domain adaptation with optimal transport improves EEG sleep stage classifiers. Pattern Recognition in Neuroimaging, Jun 2018, Singapour, Singapore. ⟨hal-01814190⟩
388 View
754 Download

Share

Gmail Mastodon Facebook X LinkedIn More