Firefighter Stress Monitoring: Model Quality and Explainability

Annu Int Conf IEEE Eng Med Biol Soc. 2022 Jul:2022:4653-4657. doi: 10.1109/EMBC48229.2022.9871470.

Abstract

A cognitive and physical stress co-classification effort started with acquisition of a training dataset and generation of machine learning models from 17 heart rate variability parameters. Accuracy was improved with multilayer perceptron models and tested on 85 firefighters in a cage maze. A specific platform acquired a dataset with better label accuracy providing a second model. Feature importance and model performance were assessed using the cage maze data. A SHAP analysis provided the basis for the model comparison and feature important assessment. Conclusions were drawn on best time windows, feature selection, and model hyperparameters.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Firefighters*
  • Heart Rate
  • Humans
  • Machine Learning
  • Neural Networks, Computer
  • Physical Exertion