skip to main content
10.1145/3606283.3606291acmotherconferencesArticle/Chapter ViewAbstractPublication PagesicgspConference Proceedingsconference-collections
research-article

Improved Classification of Mental Workload Using One Dimensional Convolutional Neural Network and SMOTE Technique

Published:11 August 2023Publication History

ABSTRACT

Mental Workload (MWL) can be measured as the amount of mental strength necessary to complete a work. Mental workload measurement can help in the reduction of mental stress, tension, anxiety, strain, and worry, in daily life or job-specific tasks. This might have a significant impact on tasks like studies/learning and driving etc. In this paper, the classification of mental workload levels through deep learning (one-dimensional convolutional neural networks (1DCNN))has been done. STEW: Simultaneous Task EEG Workload Dataset is used in this study. Electroencephalogram (EEG) signals are captured during the mental task. Based on the subject's rating after completing the mental task, the workload is categorized as low mental workload or high mental workload. To balance and increase thedataset size Synthetic Minority Oversampling Technique (SMOTE) has been used. Different classification measures are used to evaluate the performance of the deep learning model used. Finally, the mental workload classification accuracy of 97.77 was achieved.

References

  1. W. L. Lim, O. Sourina and L. P. Wang, "STEW: Simultaneous Task EEG Workload Data Set," in IEEE Transactions on Neural Systems and Rehabilitation Engineering, vol. 26, no. 11, pp. 2106-2114, Nov. 2018, doi: 10.1109/TNSRE.2018.2872924.Google ScholarGoogle ScholarCross RefCross Ref
  2. P. Zhang, X. Wang, J. Chen, W. You and W. Zhang, "Spectral and Temporal Feature Learning With Two-Stream Neural Networks for Mental Workload Assessment," in IEEE Transactions on Neural Systems and Rehabilitation Engineering, vol. 27, no. 6, pp. 1149-1159, June 2019, doi: 10.1109/TNSRE.2019.2913400.Google ScholarGoogle ScholarCross RefCross Ref
  3. J. Song, D. Zhuang, G. Song and X. Wanyan, "Pilot mental workload measurement and evaluation under dual task," 2011 4th International Conference on Biomedical Engineering and Informatics (BMEI), 2011, pp. 809-812, doi: 10.1109/BMEI.2011.6098376.Google ScholarGoogle ScholarCross RefCross Ref
  4. K. Kitamura , "A ship Navigator's mental workload using salivary NO3− compared with R-R interval: Simulator-based experiment," 2014 World Automation Congress (WAC), 2014, pp. 98-102, doi: 10.1109/WAC.2014.6935693.Google ScholarGoogle ScholarCross RefCross Ref
  5. L. G. Lim , "A Unified Analytical Framework With Multiple fNIRS Features for Mental Workload Assessment in the Prefrontal Cortex," in IEEE Transactions on Neural Systems and Rehabilitation Engineering, vol. 28, no. 11, pp. 2367-2376, Nov. 2020, doi: 10.1109/TNSRE.2020.3026991.Google ScholarGoogle ScholarCross RefCross Ref
  6. P. Zarjam, J. Epps and N. H. Lovell, "Beyond Subjective Self-Rating: EEG Signal Classification of Cognitive Workload," in IEEE Transactions on Autonomous Mental Development, vol. 7, no. 4, pp. 301-310, Dec. 2015, doi: 10.1109/TAMD.2015.2441960.Google ScholarGoogle ScholarDigital LibraryDigital Library
  7. S. B. Shafiei, A. S. Elsayed, A. A. Hussein, U. Iqbal and K. A. Guru, "Evaluating the Mental Workload During Robot-Assisted Surgery Utilizing Network Flexibility of Human Brain," in IEEE Access, vol. 8, pp. 204012-204019, 2020, doi: 10.1109/ACCESS.2020.3036751.Google ScholarGoogle ScholarCross RefCross Ref
  8. A. Henelius, K. Hirvonen, A. Holm, J. Korpela and K. Muller, "Mental workload classification using heart rate metrics," 2009 Annual International Conference of the IEEE Engineering in Medicine and Biology Society, 2009, pp. 1836-1839, doi: 10.1109/IEMBS.2009.5332602.Google ScholarGoogle ScholarCross RefCross Ref
  9. F. Larradet, G. Barresi and L. S. Mattos, "Effects of galvanic skin response feedback on user experience in gaze-controlled gaming: A pilot study," 2017 39th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), 2017, pp. 2458-2461, doi: 10.1109/EMBC.2017.8037354.Google ScholarGoogle ScholarCross RefCross Ref
  10. Saundra G. Hart, Lowell E. Staveland, “ Development of NASA-TLX (NASA Load Index): Result of Empirical and Theoretical Research” Advance in Psychology, April 2008, vol. 52, pp. 139-183.Google ScholarGoogle Scholar
  11. X. Hou, Y. Liu, O. Sourina and W. Mueller-Wittig, "CogniMeter: EEG-based Emotion, Mental Workload and Stress Visual Monitoring," 2015 International Conference on Cyberworlds (CW), 2015, pp. 153-160, doi: 10.1109/CW.2015.58.Google ScholarGoogle ScholarDigital LibraryDigital Library
  12. W. L. Lim, O. Sourina, Y. Liu and L. Wang, "EEG-based mental workload recognition related to multitasking," 2015 10th International Conference on Information, Communications and Signal Processing (ICICS), 2015, pp. 1-4, doi: 10.1109/ICICS.2015.7459834.Google ScholarGoogle ScholarCross RefCross Ref
  13. Hsu B-W, Wang M-JJ, Chen C-Y, Chen F. Effective Indices for Monitoring Mental Workload While Performing Multiple Tasks. Perceptual and Motor Skills. 2015;121(1):94-117. doi:10.2466/22.PMS.121c12x5Google ScholarGoogle ScholarCross RefCross Ref
  14. J. N. Mak, R. H. M. Chan and S. W. H. Wong, "Evaluation of mental workload in visual-motor task: Spectral analysis of single-channel frontal EEG," IECON 2013 - 39th Annual Conference of the IEEE Industrial Electronics Society, 2013, pp. 8426-8430, doi: 10.1109/IECON.2013.6700546.Google ScholarGoogle ScholarCross RefCross Ref
  15. M. K. Johnson, J. A. Blanco, R. J. Gentili, K. J. Jaquess, H. Oh and B. D. Hatfield, "Probe-independent EEG assessment of mental workload in pilots," 2015 7th International IEEE/EMBS Conference on Neural Engineering (NER), 2015, pp. 581-584, doi: 10.1109/NER.2015.7146689.Google ScholarGoogle ScholarCross RefCross Ref
  16. Ahirwal, Mitul Kumar, Anil Kumar, and Girish Kumar Singh. "Prediction and Classification." Computational Intelligence and Biomedical Signal Processing. Springer, Cham, 2021. 83-111.Google ScholarGoogle Scholar
  17. Ahirwal M.K., Kumar A., Singh G.K. (2021) Biomedical Signals. In: Computational Intelligence and Biomedical Signal Processing. SpringerBriefs in Electrical and Computer Engineering. Springer, Cham. https://doi.org/10.1007/978-3-030-67098-6_1Google ScholarGoogle ScholarCross RefCross Ref
  18. Ahirwal M.K., Kose M.R. (2021) Development of Emotional Decision-Making Model Using EEG Signals. In: Bhateja V., Peng SL., Satapathy S.C., Zhang YD. (eds) Evolution in Computational Intelligence. Advances in Intelligent Systems and Computing, vol 1176.Google ScholarGoogle ScholarCross RefCross Ref
  19. Ahirwal, M. K., & Kose, M. R. (2020). Audio-visual stimulation based emotion classification by correlated EEG channels. Health and Technology, 10(1), 7-23.Google ScholarGoogle Scholar
  20. Jinjing Ke, Jing Du, Xiaowei Luo, “ The effect of noise content and level on cognitive performance measured by electroencephalography (EEG)” Automation in Construction, vol. 31, Oct. 2021, pp. 103836.Google ScholarGoogle ScholarCross RefCross Ref
  21. Mohamad Sawan, Muhamad T. Salam, Jerome Le Lan, Amal Kasab, “ Wireless Recording System: From noninvasive EEG-NIRS to Invasive EEG devices” IEEE Transaction on Biomedical Circuits and Systems, vol. 7, April 2013, pp. 186-195.Google ScholarGoogle ScholarCross RefCross Ref
  22. Jakko Malmivuo, Sari Ahokas, Toni Valkky, “High-resolution EEG recording system using smart electrodes “ IEEE 14th Biennial Baltic Electronic Conference, Oct. 2014.Google ScholarGoogle Scholar
  23. P. B. Colditz, “ Digital Signal Processing of EEG Signals” IEEE Engineering in Medicine and Biology Magazine, vol. 21, Sep./Oct. 2001.Google ScholarGoogle Scholar
  24. Ahirwal M.K. (2021) Analysis and Identification of EEG Features for Mental Stress. In: Bhateja V., Peng SL., Satapathy S.C., Zhang YD. (eds) Evolution in Computational Intelligence. Advances in Intelligent Systems and Computing, vol 1176.Google ScholarGoogle ScholarCross RefCross Ref
  25. H. Qu, Y. Liu, L. Pang and Y. Shan, "Mental workload classification based on visual and operational EEG signals," CSAA/IET International Conference on Aircraft Utility Systems (AUS 2020), 2020, pp. 831-836, doi: 10.1049/icp.2021.0170.Google ScholarGoogle ScholarCross RefCross Ref
  26. J. Song, D. Zhuang, G. Song and X. Wanyan, "Pilot mental workload measurement and evaluation under dual task," 2011 4th International Conference on Biomedical Engineering and Informatics (BMEI), 2011, pp. 809-812, doi: 10.1109/BMEI.2011.6098376.Google ScholarGoogle ScholarCross RefCross Ref
  27. I. Kakkos , "Mental Workload Drives Different Reorganizations of Functional Cortical Connectivity Between 2D and 3D Simulated Flight Experiments," in IEEE Transactions on Neural Systems and Rehabilitation Engineering, vol. 27, no. 9, pp. 1704-1713, Sept. 2019, doi: 10.1109/TNSRE.2019.2930082.Google ScholarGoogle ScholarCross RefCross Ref
  28. M. Bueno, E. Dogan, F. Hadj Selem, E. Monacelli, S. Boverie, A. Guillaume, “ How different mental workload levels affect the take-over after automated driving” IEEE 19th Conference of intelligent Transportation System (2016).Google ScholarGoogle ScholarDigital LibraryDigital Library
  29. J. Zhang, Z. Yin and R. Wang, "Recognition of Mental Workload Levels Under Complex Human–Machine Collaboration by Using Physiological Features and Adaptive Support Vector Machines," in IEEE Transactions on Human-Machine Systems, vol. 45, no. 2, pp. 200-214, April 2015, doi: 10.1109/THMS.2014.2366914.Google ScholarGoogle ScholarCross RefCross Ref
  30. Chawla, Nitesh V., "SMOTE: synthetic minority over-sampling technique." Journal of artificial intelligence research 16 (2002): 321-357.Google ScholarGoogle ScholarDigital LibraryDigital Library
  31. U. Singh and M. K. Ahirwal, "Mental Workload Classification for Multitasking Test using Electroencephalogram Signal," 2021 IEEE International Conference on Technology, Research, and Innovation for Betterment of Society (TRIBES), 2021, pp. 1-6, doi: 10.1109/TRIBES52498.2021.9751676.Google ScholarGoogle ScholarCross RefCross Ref
  32. Vishal Panday, Dhirendra Kumar Choudhary, Vinita Verma, Greeshma Sharma, Ram Singh, Sushil Chandra, “Mental Workload Estimation Using EEG”, 2020 Fifth International Conference on Research in Computional Intelligence and Communicaton Network (ICRCICN).Google ScholarGoogle Scholar
  33. Chakladar, Debashis Das, "EEG-based mental workload estimation using deep BLSTM-LSTM network and evolutionary algorithm." Biomedical Signal Processing and Control 60 (2020): 101989.Google ScholarGoogle ScholarCross RefCross Ref
  34. Zhang, J., Cui, X., Li, J. Imbalanced classification of mental workload using a cost-sensitive majority weighted minority oversampling strategy. Cogn Tech Work 19, 633–653 (2017). https://doi.org/10.1007/s10111-017-0447-x.Google ScholarGoogle ScholarDigital LibraryDigital Library
  35. G. Zhu, F. Zong, H. Zhang, B. Wei and F. Liu, "Cognitive Load During Multitasking Can Be Accurately Assessed Based on Single Channel Electroencephalography Using Graph Methods," in IEEE Access, vol. 9, pp. 33102-33109, 2021, doi: 10.1109/ACCESS.2021.3058271.Google ScholarGoogle ScholarCross RefCross Ref
  36. V. Pandey, D. K. Choudhary, V. Verma, G. Sharma, R. Singh and S. Chandra, "Mental Workload Estimation Using EEG," 2020 Fifth International Conference on Research in Computational Intelligence and Communication Networks (ICRCICN), 2020, pp. 83-86, doi: 10.1109/ICRCICN50933.2020.9296150.Google ScholarGoogle ScholarCross RefCross Ref
  37. P. Huynh, G. Warner and H. Lin, "Effects of EMD and Feature Extraction on EEG Analysis," Journal of Advances in Information Technology, Vol. 11, No. 1, pp. 26-34, February 2020. doi: 10.12720/jait.11.1.26-34Google ScholarGoogle ScholarCross RefCross Ref

Index Terms

  1. Improved Classification of Mental Workload Using One Dimensional Convolutional Neural Network and SMOTE Technique

    Recommendations

    Comments

    Login options

    Check if you have access through your login credentials or your institution to get full access on this article.

    Sign in
    • Published in

      cover image ACM Other conferences
      ICGSP '23: Proceedings of the 2023 7th International Conference on Graphics and Signal Processing
      June 2023
      83 pages
      ISBN:9798400700460
      DOI:10.1145/3606283

      Copyright © 2023 ACM

      Publication rights licensed to ACM. ACM acknowledges that this contribution was authored or co-authored by an employee, contractor or affiliate of a national government. As such, the Government retains a nonexclusive, royalty-free right to publish or reproduce this article, or to allow others to do so, for Government purposes only.

      Publisher

      Association for Computing Machinery

      New York, NY, United States

      Publication History

      • Published: 11 August 2023

      Permissions

      Request permissions about this article.

      Request Permissions

      Check for updates

      Qualifiers

      • research-article
      • Research
      • Refereed limited
    • Article Metrics

      • Downloads (Last 12 months)43
      • Downloads (Last 6 weeks)3

      Other Metrics

    PDF Format

    View or Download as a PDF file.

    PDF

    eReader

    View online with eReader.

    eReader

    HTML Format

    View this article in HTML Format .

    View HTML Format