Conditional-GAN Based Data Augmentation for Deep Learning Task Classifier Improvement Using fNIRS Data

• Published in: Frontiers in big data Vol. 4; p. 659146
• Main Authors: Wickramaratne, Sajila D., Mahmud, Md.Shaad
• Format: Journal Article
• Language: English
• Published: Frontiers Media S.A 29.07.2021
• Subjects: Big Data; CGAN; classification; CNN; deep learning; functional near-infrared spectroscopy; GAN
• ISSN: 2624-909X; 2624-909X
• DOI: 10.3389/fdata.2021.659146

Functional near-infrared spectroscopy (fNIRS) is a neuroimaging technique used for mapping the functioning human cortex. fNIRS can be widely used in population studies due to the technology’s economic, non-invasive, and portable nature. fNIRS can be used for task classification, a crucial part of functioning with Brain-Computer Interfaces (BCIs). fNIRS data are multidimensional and complex, making them ideal for deep learning algorithms for classification. Deep Learning classifiers typically need a large amount of data to be appropriately trained without over-fitting. Generative networks can be used in such cases where a substantial amount of data is required. Still, the collection is complex due to various constraints. Conditional Generative Adversarial Networks (CGAN) can generate artificial samples of a specific category to improve the accuracy of the deep learning classifier when the sample size is insufficient. The proposed system uses a CGAN with a CNN classifier to enhance the accuracy through data augmentation. The system can determine whether the subject’s task is a Left Finger Tap, Right Finger Tap, or Foot Tap based on the fNIRS data patterns. The authors obtained a task classification accuracy of 96.67% for the CGAN-CNN combination.