NUK - logo
E-viri
Celotno besedilo
Recenzirano Odprti dostop
  • Class imbalance should not ...
    Thölke, Philipp; Mantilla-Ramos, Yorguin-Jose; Abdelhedi, Hamza; Maschke, Charlotte; Dehgan, Arthur; Harel, Yann; Kemtur, Anirudha; Mekki Berrada, Loubna; Sahraoui, Myriam; Young, Tammy; Bellemare Pépin, Antoine; El Khantour, Clara; Landry, Mathieu; Pascarella, Annalisa; Hadid, Vanessa; Combrisson, Etienne; O’Byrne, Jordan; Jerbi, Karim

    NeuroImage (Orlando, Fla.), 08/2023, Letnik: 277
    Journal Article

    •Class imbalance is common issue in the application of machine learning (ML) to neuroscience and can have severe consequences if not handled properly.•The impact of increasing data imbalance on ML performance is assessed for various levels of imbalance using simulated data, as well as EEG, MEG and fMRI recordings.•In highly imbalanced data, the commonly used Accuracy metric yields misleadingly high performances that result from systematically predicting the majority class.•The balanced accuracy (BAcc) metric is recommended as a default evaluation metric for ML, when seeking to minimize overall classification error.•A list of recommendations for dealing with imbalanced data is provided, and open-source code is made available to allow for further investigation. Machine learning (ML) is increasingly used in cognitive, computational and clinical neuroscience. The reliable and efficient application of ML requires a sound understanding of its subtleties and limitations. Training ML models on datasets with imbalanced classes is a particularly common problem, and it can have severe consequences if not adequately addressed. With the neuroscience ML user in mind, this paper provides a didactic assessment of the class imbalance problem and illustrates its impact through systematic manipulation of data imbalance ratios in (i) simulated data and (ii) brain data recorded with electroencephalography (EEG), magnetoencephalography (MEG) and functional magnetic resonance imaging (fMRI). Our results illustrate how the widely-used Accuracy (Acc) metric, which measures the overall proportion of successful predictions, yields misleadingly high performances, as class imbalance increases. Because Acc weights the per-class ratios of correct predictions proportionally to class size, it largely disregards the performance on the minority class. A binary classification model that learns to systematically vote for the majority class will yield an artificially high decoding accuracy that directly reflects the imbalance between the two classes, rather than any genuine generalizable ability to discriminate between them. We show that other evaluation metrics such as the Area Under the Curve (AUC) of the Receiver Operating Characteristic (ROC), and the less common Balanced Accuracy (BAcc) metric - defined as the arithmetic mean between sensitivity and specificity, provide more reliable performance evaluations for imbalanced data. Our findings also highlight the robustness of Random Forest (RF), and the benefits of using stratified cross-validation and hyperprameter optimization to tackle data imbalance. Critically, for neuroscience ML applications that seek to minimize overall classification error, we recommend the routine use of BAcc, which in the specific case of balanced data is equivalent to using standard Acc, and readily extends to multi-class settings. Importantly, we present a list of recommendations for dealing with imbalanced data, as well as open-source code to allow the neuroscience community to replicate and extend our observations and explore alternative approaches to coping with imbalanced data.