Prediction of ADHD from a small dataset using an adaptive EEG theta/beta ratio and PCA feature extraction

• Main Authors: Sase, Takumi, Othman, Marini
• Format: Conference or Workshop Item
• Language: English
• Published: Springer Nature Switzerland 2022
• Subjects: QA75 Electronic computers. Computer science
• Online Access: http://irep.iium.edu.my/97972/3/97972_Prediction%20of%20ADHD%20from%20a%20Small%20Dataset.pdf; http://irep.iium.edu.my/97972/
• Institution: Universiti Islam Antarabangsa Malaysia

EEG Theta/beta ratio (TBR) is conventionally used as a biomarker in childhood Attention-Deficit/Hyperactivity Disorder (ADHD) prediction and treatment. Due to the heterogeneity of ADHD symptoms, several studies have applied machine learning algorithms for enhancing the recognition of ADHD. These methods, however, have limited performance in a small dataset. In this paper, we propose an adaptive EEG feature extraction approach using TBR and PCA. Repeated TBR-PCA feature extraction, SVM classification and statistical testing were applied on a small EEG sample with ADHD/typically developing (TD) labels. The steps were repeated with an update of the feature extraction technique until a high accuracy is achieved, allowing the small samples to be correctly identified (r = 0.833, one-sided, Bonferroni-corrected p < 0.0166). Within subjects EEG samples analyses performed better compared to between subject analyses, with accuracy getting worse with the increase of EEG segments. The contribution of this work is two-fold: the practical application allows for a reliable adoption of machine learning in non-invasive EEG screening of small ADHD dataset, while the theoretical contribution extends beyond the eyes closed resting state condition considered in this study and provides a methodological approach when working with limited samples.