Diagnosis of Tobacco Addiction using Medical Signal: An EEG-based Time-Frequency Domain Analysis Using Machine Learning

Abstract

Addiction such as tobacco smoking affects the human brain and thus causes significant changes in the brainwaves. The changes in brain wave due to smoking can be identified by focusing on changes in electroencephalogram pattern, extracting different time-frequency domain features. In this aspect, a laboratory-based study has been presented in this paper, for assessing the brain signal changes due to the tobacco addiction. Four classifier models, namely, Logistic Regression (LR), K- Nearest Neighbor (KNN), Support Vector Machine (SVM) and Random Forest Classifier (RFC) were trained and tested for assessing the performance of the time domain, frequency domain and fusion of time-frequency domain features, with a five-fold cross-validation. Four different performance measures (sensitivity, specificity, accuracy, and area under the receiver operating characteristic curve) were used to measure the overall performance, and the results suggested that the classifiers based on time-frequency domain features perform the best while using combinedly. Using the utilized fusion of the time-frequency domain features, the classification models can identify the smoker group with an accuracy ranged from (86.5-91.3%), where the RFC shows the best accuracy of 91.3%, which is higher than the three other classifiers models.