Detecting olfactory impairment through objective diagnosis: Catboost classifier on EEG data

초록

Detecting olfactory impairment using an objective diagnosis kit has been a challenge. Recently, machine learning and deep learning models have been used on EEG data with promising results. The goal of our study was to detect olfactory impairment through a machine learning classifier with EEG data. This was done by identifying the important EEG data factors affecting olfactory impairment. Finally, we compared our model to other machine learning and deep learning algorithms in order to identify possibilities for further research. Downsampling and extracting various waves from EEG data were conducted for data preprocessing. Then, an independent component analysis was performed to remove artifacts. Through this processing, a dataset in CSV format was obtained. Next, we built a CatBoost classifier model because it is recent boost model and has high performance for classification. It identified whether a subject had olfactory impairment or not. After training with the CatBoost algorithm, we compared it to different machine learning and deep learning algorithms. The CatBoost model showed 87.56 % accuracy, while other machine learning algorithms such as the random forest classifier, gradient boosting classifier, XG boosting classifier, k-nearest-neighbor classifier, decision tree classifier, Gaussian NB, and logistic regressor revealed 82.22 %, 78.89 %, 78.22 %, 75.78 %, 74 %, 69.78 %, and 41.11 % accuracy, respectively. With deep learning models, which consisted of bi-directional long short term memory, long short term memory and a deep neural network, the performance was 63.11 %, 51.33 %, and 60 %. The CatBoost model showed feature importance, which revealed that the gamma wave on the Cz channel was about 20, which was the highest among the other variables.

키워드