A Novel Data Augmentation Active Learning Algorithm in a Small-Sample Setting

Guoqing Cai,Shoulin Huang,Ting Ma
DOI: https://doi.org/10.2139/ssrn.4265349
2022-01-01
SSRN Electronic Journal
Abstract:Attaining robust performance in a small-sample setting (SSS) is an urgent issue for practical applications of brain-computer interface systems (BCIs). Active learning (AL) is a potential semi-supervised approach in SSS. However, most of AL methods select part of the unlabeled trials for labeling and incorporates them directly into the training set, in which mislabeled trials will hamper the robustness of the classification models. To boost AL’s robustness in SSS, we propose a novel data augmentation AL method (DAAL) wherein data augmentation (DA) is embedded in the trial-incorporation process. We assess the effectiveness of selected trials and utilize DA to extract useful information while alleviating negative impacts. Experimental results on the two datasets demonstrate DAAL can yield higher classification performance and stronger robustness even when AL deteriorates severely. The visualization of features illustrates DAAL can combine the advantages of DA and AL. Consequently, we conclude that DAAL is a promising framework for applications of BCIs , especially in the online system.
What problem does this paper attempt to address?