INTRODUCTION
INTRODUCTION
ECG signals are commonly adopted to extract HRV [9]. HRV is defined
as the variation across intervals between consecutive regular RR intervals,1 and it
is measured by determining the length between two successive heartbeat peaks
from an ECG reading. Conventionally, HRV has been accepted as a term to
describe variations of both instantaneous heart rate and RR intervals [12].
Based on ECG data analysis (or HRV features, various machine learning
(ML) and deep learning (DL) algorithms have been developed in recent years for
stress prediction [20], [21], [22], [23], [24], [25], [26], [27] (see more details in
Sec. II). Among the publicly available datasets for stress detection, SWELL−KW
developed in [13] and [14] one of the two most popular ones. However, none of the
existing ML and DL studies based on the SWELL−KW dataset for multi-class
stress classification have achieved ultra-high accuracy, especially for multi-class
stress level classification [15], [16]. Therefore, there exists a research gap on
developing novel ML models which are able to achieve ultra-high accurate
prediction.
While the accuracy achieved with full features is nearly 100%, we have
also introduced a feature reduction algorithm based on analysis of variance
(ANOVA) F-test and demonstrate that it is possible to achieve an accuracy score of
96.5% with less than half of the features that are available in the SWELL−KW
dataset. Such a feature extraction reduces the computational load during the model
training phase.
In a nutshell, the novelty and the main contributions of this study are
summarized as follows:
• We have developed a novel 1D CNN model to detect multi-class stress status
with outstanding performance, achieving 99.9% accuracy with a Precision, F1-
score, and Recall score of 1.0 respectively and a Matthews correlation coefficient
(MCC) score of 99.9%. We believe this is the first study that achieves such a high
score of accuracy for multi-class stress classification.
• Furthermore, we reveal that not all 34 HRV features are necessary to accurately
classify multi-class stress. We have performed feature optimization to select an
optimized feature set to train a 1D CNN classifier, achieving a performance score
that beats the existing classification models based on the SWELL-KW dataset.
• Our model with selected top-ranked HRV features does not require resource-
intensive computation and it achieves also excellent accuracy without sacrificing
critical information.