Deep learning-based classification of multichannel bio-signals using directedness transfer learning

Abstract

<div lang="en" class="abs"><p class="abs_header">Abstract</p><p>The problem with processing of multivariate/multichannel signals lies in adapting of existing classifiers on data. Reformulating time-series data as visual clues and assigning visual patterns to different categories help the classification of time series in a wide range of applications. These series-to-image transformations have benefits including better noise robustness and more options regarding augmentation. They also provide the possibility of achieving discriminative features by employing transfer learning paradigm in cases dealing with highly small training datasets. In this respect, this work aimed to encode spectral-phase information into a bi-dimensional map. Transferring knowledge was done using bi-dimensional transformation capitalizing on the direction and propagation pattern of one channel influence on the others. EEG data from patients diagnosed with delirium (N = 15) recorded using a 10-channel BrainStatus device were used for this analysis. Considering leave-one-subject-out cross-validation, classification outcomes demonstrated that directedness transfer learning via Alexnet yields a promising performance showing 97.17% precision and outperforming other approaches. Comparison with nine different deep networks pretrained on ImageNet database was included. Directedness transfer learning resulted in precision of 95.29 ± 1.46 (µ ± σ)% among all networks. For further evaluation, directedness bi-dimensional transformation was also compared with six other 2D maps. Applying different networks resulted in average precision of (91.99 ± 2.23)% for polar-, (91.69 ± 1.57)% for correlation-, (90.46 ± 1.71)% for Spectrogram-, (87.82 ± 2.16)% for Wavelet-, (84.24 ± 1.72)% for Wigner-Ville- and (82.84 ± 2.46)% for Mel-frequency Cepstrum maps. To conclude, the proposed technique shows significant benefit in compressing spatio-spectral patterns of multichannel signals in just a unified visual representation.</p></div>