CADDA: Class-wise automatic differentiable data augmentation for EEG signals - Inria - Institut national de recherche en sciences et technologies du numérique Accéder directement au contenu
Communication Dans Un Congrès Année : 2022

CADDA: Class-wise automatic differentiable data augmentation for EEG signals

Résumé

Data augmentation is a key element of deep learning pipelines, as it informs the network during training about transformations of the input data that keep the label unchanged. Manually finding adequate augmentation methods and parameters for a given pipeline is however rapidly cumbersome. In particular, while intuition can guide this decision for images, the design and choice of augmentation policies remains unclear for more complex types of data, such as neuroscience signals. Besides, class-dependent augmentation strategies have been surprisingly unexplored in the literature, although it is quite intuitive: changing the color of a car image does not change the object class to be predicted, but doing the same to the picture of an orange does. This paper investigates gradient-based automatic data augmentation algorithms amenable to class-wise policies with exponentially larger search spaces. Motivated by supervised learning applications using EEG signals for which good augmentation policies are mostly unknown, we propose a new differentiable relaxation of the problem. In the class-agnostic setting, results show that our new relaxation leads to optimal performance with faster training than competing gradient-based methods, while also outperforming gradient-free methods in the class-wise setting. This work proposes also novel differentiable augmentation operations relevant for sleep stage classification.
Fichier principal
Vignette du fichier
cadda.pdf (828.99 Ko) Télécharger le fichier
MNIST_learnig_curve.pdf (14.4 Ko) Télécharger le fichier
MNIST_transforms.pdf (14.09 Ko) Télécharger le fichier
adda-vs-dada-diag.pdf (11.75 Ko) Télécharger le fichier
adda-vs-dada-vs-fraa-diag-new.pdf (16.5 Ko) Télécharger le fichier
adda-vs-dada-vs-fraa-diag.pdf (13.13 Ko) Télécharger le fichier
cadda.synctex.gz (207.15 Ko) Télécharger le fichier
class-wise-aug.pdf (24.45 Ko) Télécharger le fichier
fixed_all_curves_res.pdf (17.8 Ko) Télécharger le fichier
fixed_density_matching_burning.pdf (18.89 Ko) Télécharger le fichier
fixed_gradient-based-comparison+cw.pdf (21.72 Ko) Télécharger le fichier
fixed_gradient-based-comparison+cw_shaded.pdf (20.07 Ko) Télécharger le fichier
fixed_gradient-based-comparison+cw_with_f1.pdf (21.68 Ko) Télécharger le fichier
fixed_manual_transform_selection.pdf (22.86 Ko) Télécharger le fichier
fixed_manual_transform_selection_MASS.pdf (23.28 Ko) Télécharger le fichier
fixed_new_transforms_selected_supp.pdf (29.81 Ko) Télécharger le fichier
freq-shift.pdf (25.62 Ko) Télécharger le fichier
gradient-based-comparison+cw_shaded.pdf (20.64 Ko) Télécharger le fichier
gradient-based-comparison+cw_with_f1.pdf (21.5 Ko) Télécharger le fichier
mass_magnitudes.pdf (20.97 Ko) Télécharger le fichier
physionet_magnitudes.pdf (20.64 Ko) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03483570 , version 1 (16-12-2021)
hal-03483570 , version 2 (17-12-2021)
hal-03483570 , version 3 (09-06-2022)

Identifiants

Citer

Cédric Rommel, Thomas Moreau, Joseph Paillard, Alexandre Gramfort. CADDA: Class-wise automatic differentiable data augmentation for EEG signals. ICLR 2022 - Tenth International Conference on Learning Representations, Apr 2022, Virtual event, France. ⟨hal-03483570v3⟩
147 Consultations
156 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More