Next Article in Journal
Optical Transmission Properties of Si3N4 Add-Drop Micro-Ring Resonator Induced by a Fabry–Perot Resonance Effect
Previous Article in Journal
Dynamic Hand Gesture Recognition in In-Vehicle Environment Based on FMCW Radar and Transformer
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

A Muscle Fatigue Classification Model Based on LSTM and Improved Wavelet Packet Threshold

1
Institute of Intelligent Machines, Hefei Institutes of Physical Science, Chinese Academy of Sciences, Hefei 230031, China
2
University of Science and Technology of China, Hefei 230026, China
*
Author to whom correspondence should be addressed.
Sensors 2021, 21(19), 6369; https://doi.org/10.3390/s21196369
Submission received: 7 July 2021 / Revised: 10 September 2021 / Accepted: 16 September 2021 / Published: 24 September 2021
(This article belongs to the Section Intelligent Sensors)

Abstract

:
Previous studies have used the anaerobic threshold (AT) to non-invasively predict muscle fatigue. This study proposes a novel method for the automatic classification of muscle fatigue based on surface electromyography (sEMG). The sEMG data were acquired from 20 participants during an incremental test on a cycle ergometer using sEMG sensors placed on the vastus rectus femoris (RF), vastus lateralis (VL), vastus medialis (VM), and gastrocnemius (GA) muscles of the left leg. The ventilation volume (VE), oxygen uptake (VO2), and carbon dioxide production (VCO2) data of each participant were collected during the test. Then, we extracted the time-domain and frequency-domain features of the sEMG signal denoised by the improved wavelet packet threshold denoising algorithm. In this study, we propose a new muscle fatigue recognition model based on the long short-term memory (LSTM) network. The LSTM network was trained to classify muscle fatigue using sEMG signal features. The results showed that the improved wavelet packet threshold function has better performance in denoising sEMG signals than hard threshold and soft threshold functions. The classification performance of the muscle fatigue recognition model proposed in this paper is better than that of CNN (convolutional neural network), SVM (support vector machine), and the classification models proposed by other scholars. The best performance of the LSTM network was achieved with 70% training, 10% validation, and 20% testing rates. Generally, the proposed model can be used to monitor muscle fatigue.

1. Introduction

The neuromuscular system consists of the nervous system and the muscular system. The main function of the human muscular system is to provide the energy needed by the human body to perform various actions. Exercise-induced muscle fatigue is a physiological phenomenon in which the maximum voluntary contraction (MVC) capacity and output power of the muscle decreases. The cause of exercise-induced muscle fatigue is the accumulation of metabolites (lactic acid, hydrogen ion, inorganic phosphate) in the human blood during exercise [1]. The risk of sports injuries increases with muscle fatigue. Therefore, the precise positioning of fatigued muscles is the basis for relieving and curing muscle fatigue and has important sports medicine significance. Electromyography (EMG) is an electrical signal generated by skeletal muscles when they contract spontaneously [2]. EMG signal analysis can provide continuous measurement of the muscle’s state during the continuous fatigue contraction process, which is different from the subjective typical assessment that usually indicates that the subject can no longer perform the test [3]. The surface electromyography signal (sEMG) is a comprehensive effect of superficial muscle EMG and nerve trunk electrical activity on the skin surface. sEMG can reflect neuromuscular activity to a certain extent, and it has the advantages of non-invasiveness and simple operation [4]. The analysis of the sEMG signal is performed by a trained neurologist. When there are not enough experts to meet the needs of fatigue warning, it is very important to use deep learning technology to realize neuromuscular fatigue detection and classification on the basis of EMG signal processing.
However, the sEMG signals are affected by many confounding factors, including the following: the shape of the volume conductor; thickness of the subcutaneous tissue layers; distribution and size of motor unit areas in muscles; changes in transmembrane action potential; conductivities of the tissues; electrode position; detection system; etc. [5]. Wavelet and wavelet packet transform are a multiresolution decomposition method that can be used to analyze signals and images denoising. Both wavelet transform and wavelet packet transform are multiresolution decomposition methods, which can characterize local features in the time domain and frequency domain, and they are usually used to analyze signals and image denoising [6]. Li et al. analyzed and compared several different wavelet threshold denoising algorithms, and they proposed a new wavelet threshold denoising algorithm to effectively eliminate noise interference under strong noise background and preserve signal details [7]. Traditional threshold functions include hard threshold and soft threshold functions, which have some deficiencies in signal de-noising. To solve these shortcomings, Zhang et al. proposed an improved wavelet threshold method to denoise MRI images [8]. The results show that the improved wavelet threshold method in denoising MRI images has better performance than the traditional wavelet threshold function.
When the local muscle is fatigued, the sEMG signal of the muscle will change in the time domain and the frequency domain [9]. Many scholars have studied muscle fatigue recognition models that can identify muscle fatigue states by detecting changes in sEMG signals. Latasa et al. detected the EMG signals of cyclists in an incremental continuous cycling test and used a multi-segment linear regression algorithm to study the aerobic-anaerobic threshold transition [3]. Martinez et al. recorded the subjects’ sEMG signals in the fatigue state of the vastus lateralis in the incremental cycling test and analyzed the correlation between muscle fiber conduction velocity (MFCV), instantaneous mean frequency (iMNF), normalized root mean square (RMS), and muscle fatigue state [10]. Subasi and Kiymik used independent component analysis (ICA) to process the surface EMG signal of the biceps brachii, and they used artificial neural network (ANN) to classify muscle fatigue [11]. M-wave normalization of the sEMG signal can describe the activity amplitudes in the state of muscle fatigue during repeated sprinting [12]. Wu et al. proposed a BFA–Gaussian support vector machine (GSVCM) model to improve the accuracy of muscle fatigue recognition [13]. Hussain and Mamun utilized different wavelet functions (WFs) to analyze the EMG signal of the right rectus femoris muscle to identify the fatigue state of the right rectus femoris muscle during walking [14]. Consequently, they used these functions for accurate automated muscle fatigue classification and systematically processing the EMG signal based on deep learning and machine learning methods.
This paper researches a new muscle fatigue state recognition model. First of all, we present a new wavelet packet threshold function denoising method to denoise sEMG signals. The signal-to-noise ratio (SNR) and root mean square error (RMSE) parameters of this method, hard threshold, and soft threshold denoising are compared. Secondly, we describe the time-domain and frequency-domain feature extraction applied in the sEMG signals classification process. We divided the sEMG signals into two groups: non-fatigue and fatigue, based on the anaerobic threshold (AT). Finally, the long and short-term memory (LSTM) network was used to identify muscle fatigue and compared with other classification algorithms in terms of classification performance.

2. Materials and Methods

2.1. Participants

Experimental participants: Twenty healthy males (n = 20; Age range: 22–33 years; Height: 1.79 ± 0.09 m; Weight: 65.5 ± 6.6 kg) were recruited to participate in this experiment. All participants were screened and free from cardiovascular, neuromuscular, and metabolic diseases. Before the test, all participants voluntarily participated in the test, fully understood the purpose, details, methods, and potential test risks of the experiment, and signed an informed consent form. They did not engage in vigorous exercise and did not consume caffeine, nicotine, and alcohol within 48 h before testing. The study was conducted in accordance with the Declaration of Helsinki and the National law of China.

2.2. Instrumentation

To ensure that the power is accurately regulated, the exercise test wase performed in the Lode Corival cpet cycle ergometer. The Ag–AgCl surface electrodes were arranged in bipolar configuration (20 mm center-to-center, 1 cm in diameter), and a Noraxon Ultium sEMG sensor were used to record the sEMG signal during the experiment. The sample frequency of the Noraxon Ultium sEMG sensors is 2000 Hz. We use Noraxon MR3 software to analyze the collected sEMG signals offline. The Ultima GX system was used to measure the participants’ ventilation (VE), oxygen uptake (VO2), and carbon dioxide production (VCO2) during the exercise test. We calibrated all the sensors and experimental equipment before the experiment.
The experimental data analysis was carried out on a workstation with an Intel CoreTM i7-9700 and 16 GB memory card. The preparation of the simulation programs was carried out on MATLAB 2020a.

2.3. Procedure

Before the experiment, in order to reduce the impedance, it is necessary to shave the excess hair of the skin of the participant’s legs and wipe it with medical alcohol. Then, we placed sEMG sensors on the appropriate position of the vastus rectus femoris (RF), vastus lateralis (VL), vastus medialis (VM), and gastrocnemius (GA) muscles according to the SENIAM (Surface EMG for Non-Invasive Assessment of Muscles) guidelines [15]. In order for participants not to be affected during the execution of the procedure, the sports bandages were used to fix the sEMG sensors. The position of the sEMG sensors on the left leg is shown in Figure 1.
Before the test, participants could use equipment and procedures proficiently. After a 3 min warm-up on a cycle ergometer (Lode Corival cpet), each of the participants performed an incremental protocol in a temperate environment (25–28 ℃), starting at a 100 W initial workload with increases of 25 W every 1 min. The participants were instructed to maintain a pedaling rate within the range of 70–75 r/min throughout the test [16,17]. During the test, we strongly encouraged each participant to provide a maximal effort. The test was terminated when the participant was unable to maintain a pedaling rate above 70 r/min due to volitional exhaustion. The participant performs the test as shown in Figure 2.
After the test, we use the V-slope method to calculate the AT based on analyzing the slopes of VO2 and VCO2 volume curves [18]. We divide the sEMG signals into fatigue and non-fatigued status at the time corresponding to the AT.

2.4. Wavelet Packet Threshold Denoising

The wavelet packet threshold denoising algorithm uses the multi-scale characteristics of wavelet packet analysis to decompose the signal by wavelet packet. The threshold function determines whether the wavelet signals of different layers are noise signals according to the threshold. The wavelet coefficients related to noise become corresponding appropriate values according to a different number of layers. In the threshold denoising method based on wavelet transform, the selection of wavelet basis, the number of wavelet decomposition layers, the threshold, and the threshold function have a great influence on the effect of wavelet threshold denoising [19]. According to previous research, the Daubechies (Db) wavelet family has the most suitable wavelet functions for sEMG signals denoising analysis [20,21]. Therefore, this paper uses the db45 wavelet function for the wavelet packet decomposition. This chapter will mainly describe the optimal wavelet packet decomposition layer algorithm and the improved wavelet threshold function.

2.4.1. Best Tree Wavelet Packet Analysis

The wavelet packet decomposes the sEMG signals to obtain high-frequency coefficients and low-frequency coefficients. Both the high-frequency coefficient and the low-frequency coefficient are decomposed by analogy as the input signal of the next stage until the decomposition reaches the set number of layers. The complete binary tree is produced as shown in Figure 3. Before the decomposition of the wavelet packet, it is necessary to find the best tree of the wavelet packet. Starting with the root node, the best tree is calculated using Shannon entropy. The Shannon entropy calculation method is as follows.
E ( s i ) = i Z s i 2 l o g ( s i 2 )
where s i is the wavelet packet coefficient sequence. The following method is used to calculate the best tree. A node N1 is split into two nodes N2 and N3 if and only if the sum of the entropy of N2 and N3 is lower than the entropy of N1; otherwise, node N1 will not be decomposed [22]. This is a local criterion based on the information available at node N1, as shown in Figure 4. It is calculated by the Shannon entropy criterion in which the decomposition of four layers is most suitable in the sEMG signals’ wavelet packet decomposition.

2.4.2. Threshold Function

The threshold function is essential for sEMG signal denoising analysis. The hard threshold and soft threshold functions proposed by Donoho [23] are extensively used. The hard threshold and soft threshold functions are expressed as Equations (2) and (3), respectively.
The hard threshold function is calculated as follows:
w ^ j , k = { w j , k , | w j , k | λ 0 , | w j , k | < λ .
The soft threshold function is expressed as follows:
w ^ j , k = { sign ( w j , k ) ( | w j , k | λ ) , | w j , k | λ 0 , | w j , k | < λ .
Although the above-mentioned traditional signal denoising methods have some effects in practical applications, it is undeniable that they still have shortcomings. The disadvantage of the hard threshold function itself is discontinuous at threshold λ , and the wavelet coefficients ( w j , k ) larger than the threshold λ are not processed, and the w j , k smaller than the threshold λ are set to zero, which will cause oscillations in the reconstruction of w j , k . Although the soft threshold function is continuous and derivable at the threshold λ , the wavelet coefficients processed by Equation (3) deviate from the actual wavelet coefficients. In order to compensate for the deficiencies of the above-mentioned threshold function, we proposed the improved threshold function as follows:
w ^ j , k = { w j , k m w j , k 1 + l o g ( | w j , k | λ ) , | w j , k | λ ( 1 m ) s i g n ( w j , k ) | w j , k | k + 1 ( 1 l o g ( | w j , k | λ ) λ k , | w j , k | < λ .
When | w j , k | λ + , Equation (4) can be written as:
lim | w j , k | λ + ( w j , k m w j , k 1 + l o g ( | w j , k | λ ) ) = ( 1 m ) λ .
When | w j , k | λ , Equation (4) can be written as:
lim | w j , k | λ ( ( 1 m ) s i g n ( w j , k ) | w j , k | k + 1 ( 1 l o g ( | w j , k | λ ) λ k ) = ( 1 m ) λ .
In Equations (2)–(4), w j , k is the wavelet coefficient, and λ is the threshold. Parameters k and m in Equation (4) include the adjustment parameters ( k N , m ( 0 , 1 ) ). The improved threshold function is not only continuous at the threshold but also high-order differentiable. The denoising effect of the wavelet packet threshold denoising algorithm is closely related to the choice of threshold and threshold function [24]. In recent years, the following four threshold estimation criteria have been widely used: fixed threshold estimation (Sqtwolog), maximal minimum threshold estimation (Minimaxi), unbiased risk estimation (Rigsure), and heuristic threshold estimation (Heursure). In this paper, we select the Heursure estimation method to calculate the threshold λ. The algorithm flow is as follows:
  • Square each element in the vector W, and then sort in ascending order to obtain a new vector W ( W ( 1 ) , W ( 2 ) , …, W ( N ) ). The length of the vector W is the integer N.
  • The threshold is the square root of the i-th element of the vector W ; then, the risk algorithm is as follows [25]:
    r i s k ( i ) = N 2 k + j = 1 i W ( j ) + ( N i + 1 ) W ( N i + 1 ) N i = 1 , 2 , , N .
  • Calculate the i value corresponding to the minimum risk (i), as shown in Equation (8). The threshold λ calculation method is shown in Equation (9).
    i ^ = arg min i ( r i s k ( i ) )
    λ = W ( i ^ )
The procedure of the wavelet packet threshold denoising algorithm is as follows [26]:
  • Select the appropriate wavelet function, and perform the wavelet packet decomposition calculation according to the best tree of the wavelet packet;
  • A threshold value is selected for the wavelet packet coefficients of each decomposition scale. Select the appropriate threshold function to process the wavelet packet coefficients w j , k ;
  • The processed wavelet packet coefficients w ^ j , k are reconstructed by inverse wavelet packet transformation. The denoised signals are obtained.
The objective evaluation of the wavelet packet threshold denoising algorithm is described by the signal-to-noise ratio (SNR) and mean square error (MSE). The calculation methods of SNR and RMSE are shown in Equations (10) and (11).
S N R = 10 lg [ n x ^ 2 ( n ) n [ x ^ ( n ) x ( n ) ] 2 ]
R M S E = 1 n n [ x ^ ( n ) x ( n ) ] 2
In Equations (10) and (11), x ^ ( n ) is the original signal, x ( n ) is the signal after denoising, and n is the signal length.

2.5. Feature Extraction

The sEMG signal is a 1D time-series signal of the neuromuscular system that is recorded on the skin surface. The analysis of sEMG concentrated on two main fields: the frequency domain and the time domain [27]. In this study, we choose two time-domain features, Root Mean Square (RMS) and Integrated Electromyogram (IEMG), to describe the changes in the amplitude of the sEMG signals. The mathematical expressions of RMS and IEMG are expressed as:
IEMG = t t + T | x ( t ) | d t
RMS = 1 T t t + T x 2 ( t ) d t .
The frequency-domain feature is extracted from the Fourier transform of the signal. We choose median frequency (MF) and mean power frequency (MPF) because of their abilities to reflect the fatigue-caused frequency changes of sEMG [28,29,30]. The mathematical expressions of MF and MPF are expressed as:
f 1 MF P ( f ) d f = MF f 2 P ( f ) d f
MPF = f 1 f 2 f P ( f ) d f f 1 f 2 P ( f ) d f .
In Equations (14) and (15), f1 and f2 determine the bandwidth, and P(f) is the power spectral density (PSD) of the sEMG signals. P(f) is expressed as:
P ( f ) = | x ( f ) | 2 L .
In the above equations, L is the signal length, and x(f) is the sEMG signals in the frequency domain. The RMS, IEMG, MF, and MPF are extracted from the denoised sEMG signals by a moving window of 2 s. In order to construct a muscle fatigue feature data set, we divide the sEMG feature data into fatigue and non-fatigued sEMG data according to the time point of the AT. The V-Slope method was introduced to calculate the AT in the fatigue test.

2.6. Fatigue Recognition Model

The fatigue recognition model classifies the sEMG signal of muscle fatigue status and muscle non-fatigue status and was constructed based on LSTM networks. The proposed method is presented in Figure 5. The LSTM network is a modified version of recurrent neural network (RNN). The control system of the LSTM unit consists of input, output and forget gates. The internal state c t of the LSTM network records the historical information up to the current moment, and the three gates control the information transmission path [31]. The forget gate f t controls the amount of information that needs to be forgotten in the internal state c t 1 at the previous moment, and the activation state of the forget gate is computed as shown in Equation (17).
f t = σ ( W f x t + U f h t 1 + b f )
where x t is the current input vector of the LSTM unit, h t 1 is the output of the previous LSTM unit, σ is the logistic sigmoid function, W f is the weight vector, and b f is the biased vector. The input gate i t determines the amount of information stored in the current candidate state, and the calculation method is shown in Equations (18) and (19).
i t = σ ( W i x t + U i h t 1 + b i )
c t = f t c t 1 + i t tanh ( W c x t + U c h t 1 + b c )
The data and information of the current LSTM unit are conveyed to the output gate, and the output calculations are shown in Equations (20) and (21).
o t = σ ( W o x t + U o h t 1 + b o )
h t = o t tanh ( c t )
The LSTM network used in this study consists of an LSTM layer, fully connected layer, rectified linear unit (ReLU) layer, dropout layer, and softmax layer. The fully connected layer connects all neurons of the previous and next layers. The dropout layer can make a certain neuron activation value stop working with a certain probability, which can make the model more generalized and prevent overfitting. Finally, the activation function of the softmax layer was used to classify the muscle state. The LSTM network used the initial hyperparameter configuration shown in Table 1. We selected the Stochastic Momentum Gradient Descent (SGDM) algorithm to optimize the learnable parameters. During the LSTM network training process, these hyperparameters were reset many times until the optimal configuration was reached.
We constructed muscle fatigue recognition models based on SVM (support vector machine) and CNN (convolutional neural network) respectively and then compared the LSTM muscle fatigue recognition models. The kernel function has an important impact on the classification performance of the SVM. Numerous applications indicate that the Gaussian kernel function has good learning capability, so the SVM used in this study uses the Gaussian kernel function. The CNN network consists of 5 layers: an input layer, two convolutional layers, and two fully connected layers. The cross-entropy loss function is used to conduct back-propagation training based on the error. The initial learning rate is 0.1.

2.7. Evaluation of the Proposed Model

In order to ensure the generalization ability of the model, the muscle fatigue recognition model needs to be trained, verified, and tested on an independent data set [32]. The verification method selected in this article is Holdout. This method consists of dividing the data into three independents subsets: training, validation, and test. The premise is that the ratio of fatigue data to non-fatigued data in each subset is approximately equal to the ratio of fatigue data to non-fatigued data in the total sEMG data set. The training data set was used for LSTM network training. The validation data set was used to evaluate performance during training through accurate measurements and error. The test data set was used for a final evaluation of the predictions performed by the model. Three datasets were independent, as Figure 6 shows. The dataset division mode was divided into 3 types: (a) 70%, 10%, 20% (training, validation, testing); (b) 60%, 10%, 30% (training, validation, testing); (c) 50%, 10%, 40% (training, validation, testing).
We select four indicators of accuracy ( A c c ), sensitivity ( S n ), specificity ( S p ), and precision ( P r ) to evaluate the performance of the model. The calculation methods of the above indicators are shown in Equations (22)–(25), respectively.
A c c = T F + T N T F + T N + F F + F N
S n = T F T F + F N
S p = T N T N + F F
P r = T F T F + F F
The parameters TP, TN, FF, and FN in Equations (22)–(25) are true fatigue (TF), true non-fatigue (TN), false fatigue (FF), and false non-fatigue (FN).

3. Results

3.1. Evaluation of Enoising Algorithm Performance

In order to illustrate the effectiveness of the wavelet packet threshold algorithm in denoising sEMG signals, the traditional threshold function is compared with the improved threshold function proposed. Our sEMG data were obtained from the MIT-BIH Normal Sinus Rhythm Database (Accessed date: 2 July 2021. https://physionet.org/content/emgdb/1.0.0/). A total of 26,860 points were selected after 6 s of the first channel of the emg_healthy signal, and they were used in this study after adding Gaussian white noise.
The denoising effect comparison between the improved wavelet packet threshold function and the traditional threshold function is shown in Figure 7. It can be seen from the waveform before and after denoising that the reconstructed waveform after the denoising of the wavelet packet-improved threshold function is relatively smoother, and its amplitude is almost consistent with the original signal, which retains useful information. The denoising performance is shown in Table 2; it can be seen that the SNR of the sEMG signals after the improved threshold function denoising is nearly 49.34% and 56.98% higher than that of the sEMG signals after the hard and soft threshold functions, respectively. The RMSE values of the sEMG signals denoised by the improved threshold function are 51.01% and 16.34% lower than the RMSE of the hard threshold function and soft threshold function, respectively.
The sEMG signals denoised by the wavelet packet improved threshold algorithm is shown in Figure 8. Based on these signals, time and frequency-domain feature data would be extracted to construct a training–test dataset.

3.2. Fatigue Status Recognition

The graphs of training, validation, testing (70%, 10%, 20%) accuracies, and training testing loss values are given in Figure 9. The training and testing time was 1702 s.
The performance criteria results, which consist of accuracy, sensitivity, specificity, and precision, are shown in Table 3. As can be seen from Table 3, in the process of using the LSTM network to perform fatigue classification on the sEMG feature dataset, the best classification performance that was obtained with the training data, the verification data, and the test data accounted for 70%, 10%, and 20% of the entire data set, respectively. Moreover, the best classification accuracy of CNN and SVM that was also obtained with the training data, the verification data, and the test data accounted for 70%, 10%, and 20% of the entire data set, respectively. In Table 3, the worst result in the evaluation of muscle fatigue status recognition is the accuracy standard of 0.8569, which is obtained from the training–validation–testing rate of 50% 10% 40% and CNN. For 70%, 10%, 20% rates of training–validation–testing, the confusion matrix results of LSTM, SVM, and CNN are shown in Figure 10. The performance of the LSTM model with the sEMG signals denoised by the improved threshold function as input is better than the LSTM model with the sEMG signals denoised by the hard and soft threshold denoising as input, as shown in Table 4. For instance, the classification accuracy of the combination of the improved wavelet threshold function and the LSTM model is 6.47% and 3.85% higher than the hard threshold function and the soft threshold function, respectively.

4. Discussion

In this section, the proposed methods and common methods will be discussed in terms of denoising performance and classification performance. The original sEMG signal was almost submerged by noise after adding Gaussian noise. The sEMG signal denoised by the hard threshold and soft threshold function still retains considerable noises. By comparing the images (Figure 7 and Figure 8) and denoising performance (Table 2), it can be seen that the improved threshold function denoising method is more suitable for surface EMG signal denoising than the other two threshold functions. The signal denoised by the improved threshold function has better performance than the signal denoised by the hard threshold and soft threshold function in approximating the original signal.
The classification performancea of these methods are given in Table 4 Wu et al. [13] proposed a novel bacterial foraging algorithm (BFA)–Gaussian support vector classifier machine (GSVCM) model to improve the muscle fatigue classification accuracy. The RMS, IEMG, MPF, MF, and mean instantaneous frequency (MIF) features of the sEMG signals were extracted to evaluate the fatigue status during muscle contraction. With the training–testing rate of 80% 20%, the GSVCM model achieves the best accuracy of 93.94%. Khan et al. [33] constructed three separate random forest models to classify muscle fatigue status based on eight sEMG features, with the best accuracy 87%. A CNN-SVM algorithm was proposed to identify muscle fatigue status [33]. The RMS, IEMG, MPF, MF, and BSE features of the sEMG signals are extracted as the input dataset of the algorithm, and the best recognition accuracy is 86%.
This research proposes a new muscle fatigue recognition model based on LSTM, which is learned from scratch. In the preprocessing stage, the collected sEMG signal was processed by the wavelet packet threshold function denoising algorithm, and then the time-domain and frequency-domain features of the signal were extracted as the input of the LSTM algorithm. As shown in Table 5, the wavelet packet threshold function denoising algorithm improves the performance of the classification algorithm, and the improved wavelet threshold function denoising algorithm has the best effect. With the same dataset as input, the LSTM algorithm performs better than CNN, SVM, and BFA–GSVCM in the accuracy of muscle fatigue recognition. The model proposed in this paper was trained, validated, and tested based on three different dataset ratios. The best accuracy of the model was obtained at type (a) 70%, 10%, 20% training, validation, testing. This indicates that the classification performance of the proposed model will improve with the increase in sEMG training data. However, the computation time for training process of the new muscle fatigue classification model is relatively long.
In addition, the model can be applied to the automatic detection of leg muscle fatigue during exercises other than cyclic resistance exercises. This model can be used as an auxiliary tool for coaches to monitor the muscle state of cyclists and long-distance runners during training. In the future, this model can be used to detect the state of muscles in other parts of the human body during exercise.

5. Conclusions

In this study, we analyzed the shortcomings of the traditional threshold function in the denoising of sEMG signal and proposed an improved threshold function. We compared the classification performance of LSTM with CNN, SVM, and other classification algorithms. In conclusion, the muscle fatigue recognition model constructed based on the improved wavelet packet threshold function denoising algorithm and LSTM network has excellent performance in the denoising of sEMG signal and the classification of muscle fatigue. The experimental results prove that the improved wavelet packet threshold function denoising algorithm is significantly better than the hard threshold and soft threshold functions in denoising the EMG signal. When the dataset was divided into training–verification–test 70% 10% 20%, the LSTM network achieved the best performance, including accuracy, sensitivity, specificity, and precision. Compared with other classification algorithms, the LSTM network achieved the best performance, including accuracy, sensitivity, and precision. However, we plan to improve the accuracy of the model in terms of feature extraction and algorithm optimization.

Author Contributions

Individual contributions were distributed as follows: conceptualization (J.W., Y.S.), methodology (J.W., Y.S., S.S.), data collection (J.W.), software (J.W.), project administration (S.S.), review and editing (Y.S., S.S.). All authors have read and agreed to the published version of the manuscript.

Funding

This research was funded by Anhui Provincial Key Research and Development Plan (grant 202004a07020037).

Institutional Review Board Statement

All the experiment procedures covered in this study were evaluated and approved by the Ethic Committee of the Affiliated Hospital of Institute of Neurology, Anhui University of Chinese Medicine. The Affiliated Hospital of the Institute of Neurology of Anhui University of Traditional Chinese Medicine is a partner of this research funded project.

Informed Consent Statement

Written informed consent has been obtained from the participants involved in this study.

Data Availability Statement

Access the data using the Google Cloud command line tools (please refer to the gsutil documentation for guidance): gsutil -m -u YOUR_PROJECT_ID cp -r gs://emgdb-1.0.0.physionet.org DESTINATION.

Acknowledgments

We would like to thank Jincheng Chen and Shaoming Sun for their help in guiding the project.

Conflicts of Interest

The authors declare no conflict of interest.

Abbreviations

ATanaerobic threshold
CNNconvolutional neural network
FFfalse fatigue
FNfalse non-fatigue
GAgastrocnemius
IEMGintegrated electromyogram
LSTMlong short-term memory
MFmedian frequency
MPFmean power frequency
MSEmean square error
MVCmaximum voluntary contraction
PSDpower spectral density
ReLUrectified linear unit
RFrectus femoris
RMSroot mean square
RMSEroot mean square error
RNNrecurrent neural network
sEMGSurface electromyography
SNRsignal-to-noise ratio
SVMsupport vector machine
TFtrue fatigue
TNtrue non-fatigue
VCO2carbon dioxide production
VEventilation volume
VLvastus lateralis
VMvastus medialis
VO2oxygen uptake

References

  1. Biglandritchie, B.; Woods, J.J. Changes in Muscle Contractile Properties and Neural Control during Human Muscular Fatigue. Muscle Nerve 1984, 7, 691–699. [Google Scholar] [CrossRef]
  2. Stashuk, D. Emg Signal Decomposition: How Can It Be Accomplished and Used? J. Electromyogr. Kinesiol. 2001, 11, 151–173. [Google Scholar] [CrossRef]
  3. Latasa, I.; Cordova, A.; Quintana-Ortí, G.; Lavilla-Oiz, A.; Navallas, J.; Rodriguez-Falces, J. Evaluation of the Electromyography Test for the Analysis of the Aerobic-Anaerobic Transition in Elite Cyclists during Incremental Exercise. Appl. Sci. 2019, 9, 589. [Google Scholar] [CrossRef] [Green Version]
  4. González-Izal, M.; Malanda, A.; Gorostiaga, E.; Izquierdo, M. Electromyographic models to assess muscle fatigue. J. Electromyogr. Kinesiol. 2012, 22, 501–512. [Google Scholar] [CrossRef]
  5. Farina, D.; Merletti, R.; Enoka, R.M. The extraction of neural strategies from the surface EMG. J. Appl. Physiol. 2004, 96, 1486–1495. [Google Scholar] [CrossRef] [Green Version]
  6. Shi, Y.; Zhang, J.; Jiao, J.; Zhao, R.; Cao, H. Calibration Analysis of High-G MEMS Accelerometer Sensor Based on Wavelet and Wavelet Packet Denoising. Sensors 2021, 21, 1231. [Google Scholar] [CrossRef] [PubMed]
  7. Li, H.; Zhou, Y.; Tian, F.; Sun, T. Wavelet-based vibration signal de-noising algorithm with a new adaptive threshold function. Chin. J. Sci. Instrum. 2015, 36, 2200–2206. [Google Scholar]
  8. Zhang, Y.; Ding, W.; Pan, Z.; Qin, J. Improved wavelet threshold for image de-noising. Front. Neurosci. 2019, 13, 39. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  9. Petrofsky, J.S.; Glaser, R.M.; Phillips, C.A.; Lind, A.R.; Williams, C. Evaluation of amplitude and frequency components of the surface EMG as an index of muscle fatigue. Ergonomics 1982, 25, 213–223. [Google Scholar] [CrossRef] [PubMed]
  10. Martinez-Valdes, E.; Guzman-Venegas, R.A.; Silvestre, R.A.; Macdonald, J.H.; Haichelis, D. Electromyographic adjustments during continuous and intermittent incremental fatiguing cycling. Scand. J. Med. Sci. Sports 2016, 26, 1273–1282. [Google Scholar] [CrossRef]
  11. Subasi, A.; Kiymik, M.K. Muscle Fatigue Detection in EMG Using Time–Frequency Methods, ICA and Neural Networks. J. Med. Syst. 2010, 34, 777–785. [Google Scholar] [CrossRef]
  12. Girard, O.; Bishop, D.J.; Racinais, S. M-wave normalization of emg signal to investigate heat stress and fatigue. J. Sci. Med. Sport 2018, 21, 518–524. [Google Scholar] [CrossRef]
  13. Wu, Q.; Xi, C.; Ding, L.; Wei, C.; Ren, H.; Law, R.; Dong, H.; Li, X.L. Classification of EMG signals by BFA-optimized GSVCM for diagnosis of fatigue status. IEEE Trans. Autom. Sci. Eng. 2017, 14, 915–930. [Google Scholar] [CrossRef]
  14. Hussain, M.; Mamun, M. Effectiveness of the Wavelet Transform on the Surface EMG to Understand the Muscle Fatigue During Walk. Meas. Sci. Rev. 2012, 12, 28–33. [Google Scholar] [CrossRef] [Green Version]
  15. Hermens, H.J.; Freriks, B.; Merletti, R.; Stegeman, D.; Blok, J.; Rau, G.; Disselhorst-Klug, C.; Hägg, G.M. European Recommendations for Surface Electromyography: Results of the SENIAM Project Roessingh Research and Development; SENIAM: Enschede, The Netherlands, 1999. [Google Scholar]
  16. Camic, C.L.; Housh, T.J.; Johnson, G.O.; Hendrix, C.R.; Zuniga, J.M.; Mielke, M.; Schmidt, R.J. An EMG frequency-based test for estimating the neuromuscular fatigue threshold during cycle ergometry. Eur. J. Appl. Physiol. 2010, 108, 337–345. [Google Scholar] [CrossRef] [PubMed]
  17. Sbriccoli, P.; Sacchetti, M.; Felici, F.; Gizzi, L.; Lenti, M.; Scotto, A.; De Vito, G. Non-invasive assessment of muscle fiber conduction velocity during an incremental maximal cycling test. J. Electromyogr. Kinesiol. 2009, 19, e380–e386. [Google Scholar] [CrossRef] [PubMed]
  18. Beaver, W.L.; Wasserman, K.; Whipp, B.J. A new method for detecting anaerobic threshold by gas exchange. J. Appl. Physiol. 1986, 60, 2020–2027. [Google Scholar] [CrossRef]
  19. Huang, J.; Tian, G.; Xie, J.; Li, H.; Chen, X. Self-adaptive decomposition level de-noising method based on wavelet transform. Telkomnika Indones. J. Electr. Eng. 2012, 10, 1015–1020. [Google Scholar] [CrossRef]
  20. Hussain, M.S.; Reaz, M.; Mohd-Yasin, F.; Ibrahimy, M.I. Electromyography signal analysis using wavelet transform and higher order statistics to determine muscle contraction. Expert Syst. 2009, 26, 35–48. [Google Scholar] [CrossRef]
  21. Coifman, R.R.; Wickerhauser, M.V. Entropy-based algorithms for best basis selection. IEEE Trans. Inf. Theory 1992, 38, 713–718. [Google Scholar] [CrossRef] [Green Version]
  22. Donoho, D.L. De-noising by soft-thresholding. IEEE Trans. Inf. Theory 1995, 41, 613–627. [Google Scholar] [CrossRef] [Green Version]
  23. Yu, B.; Li, S.; Chen, C.; Xu, J.; Qiu, W.; Wu, X.; Chen, R. Prediction subcellular localization of Gram-negative bacterial proteins by support vector machine using wavelet denoising and Chou’s pseudo amino acid composition. Chemom. Intell. Lab. Syst. 2017, 167, 102–112. [Google Scholar] [CrossRef]
  24. Lu, Q.; Pang, L.; Huang, H.; Shen, C.; Cao, H.; Shi, Y.; Liu, J. High-G Calibration Denoising Method for High-G MEMS Accelerometer Based on EMD and Wavelet Threshold. Micromachines 2019, 10, 134. [Google Scholar] [CrossRef] [Green Version]
  25. Zhang, X. A Modified Artificial Bee Colony Algorithm for Image Denoising Using Parametric Wavelet Thresholding Method. Pattern Recognit. Image Anal. 2018, 28, 557–568. [Google Scholar] [CrossRef]
  26. Duchene, J.; Goubel, F. Surface Electromyogram During Voluntary Contraction—Processing Tools and Relation to Physiological Events. Crit. Rev. Biomed. Eng. 1993, 21, 313–397. [Google Scholar] [PubMed]
  27. Thongpanja, S.; Phinyomark, A.; Phukpattaranont, P.; Limsakul, C. A Feasibility Study of Fatigue and Muscle Contraction Indices Based on EMG Time-dependent Spectral Analysis. Procedia Eng. 2012, 32, 239–245. [Google Scholar] [CrossRef] [Green Version]
  28. De Luca, C.J. Use of the surface emg signal for performance evaluation of back muscles. Muscle Nerve 1993, 16, 210–216. [Google Scholar] [CrossRef]
  29. Gonzalez-Izal, M.; Malanda, A.; Navarro-Amezqueta, I.; Gorostiaga, E.M.; Mallor, F.; Ibanez, J.; Izquierdo, M. EMG spectral indices and muscle power fatigue during dynamic contractions. J. Electromyogr. Kinesiol. 2010, 20, 233–240. [Google Scholar] [CrossRef]
  30. Hochreiter, S.; Schmidhuber, J. Long short-term memory. Neural Comput. 1997, 9, 1735–1780. [Google Scholar] [CrossRef] [PubMed]
  31. Haykin, S. Neural Networks and Learning Machines; Pearson Schweiz Ag.: London, UK, 2008; Volume 3. [Google Scholar]
  32. Khan, T.; Lundgren, L.E.; Jrpe, E.; Olsson, M.C.; Viberg, P. A novel method for classification of running fatigue using change-point segmentation. Sensors 2019, 19, 4729. [Google Scholar] [CrossRef] [Green Version]
  33. Wang, J.; Sun, Y.; Sun, S. Recognition of muscle fatigue status based on improved wavelet threshold and cnn-svm. IEEE Access 2020, 8, 207914–207922. [Google Scholar] [CrossRef]
Figure 1. Description of muscle location on the left leg and sEMG electrode placement; GA, RF, VM, and VL muscles.
Figure 1. Description of muscle location on the left leg and sEMG electrode placement; GA, RF, VM, and VL muscles.
Sensors 21 06369 g001
Figure 2. Representative participant performs exercise test.
Figure 2. Representative participant performs exercise test.
Sensors 21 06369 g002
Figure 3. Schematic diagram of wavelet packet decomposition tree at level 3.
Figure 3. Schematic diagram of wavelet packet decomposition tree at level 3.
Sensors 21 06369 g003
Figure 4. Best tree decomposition principle.
Figure 4. Best tree decomposition principle.
Sensors 21 06369 g004
Figure 5. Representation of the proposed method.
Figure 5. Representation of the proposed method.
Sensors 21 06369 g005
Figure 6. Holdout method for training, validation, and testing.
Figure 6. Holdout method for training, validation, and testing.
Sensors 21 06369 g006
Figure 7. (a) Improved threshold function denoising for the EMG signal; (b) Soft threshold function denoising for the EMG signal; (c) Hard threshold function denoising for the EMG signal.
Figure 7. (a) Improved threshold function denoising for the EMG signal; (b) Soft threshold function denoising for the EMG signal; (c) Hard threshold function denoising for the EMG signal.
Sensors 21 06369 g007
Figure 8. (a) The sEMG signals of VL, RF, VM, and GA denoised by improved wavelet packet threshold function; (b) sEMG signal of RF after local amplification.
Figure 8. (a) The sEMG signals of VL, RF, VM, and GA denoised by improved wavelet packet threshold function; (b) sEMG signal of RF after local amplification.
Sensors 21 06369 g008
Figure 9. The graphs of accuracy and loss values for the LSTM training and validation.
Figure 9. The graphs of accuracy and loss values for the LSTM training and validation.
Sensors 21 06369 g009
Figure 10. Confusion matrix results for sEMG feature data and the rates of training–validation–testing.
Figure 10. Confusion matrix results for sEMG feature data and the rates of training–validation–testing.
Sensors 21 06369 g010
Table 1. LSTM configuration.
Table 1. LSTM configuration.
HyperparameterValue
LSTM units100
Layers5
Loss functionRTRL
OptimizerAdam
ActivationReLU
Batch size70
Initial learning rate0.001
Table 2. Denoising results of the three threshold functions.
Table 2. Denoising results of the three threshold functions.
Threshold FunctionSNRRMSE
Hard Threshold Function13.27415.892
Soft Threshold Function12.6279.307
Improved Threshold Function19.8237.786
Table 3. Performance criteria for training–validation–testing rate.
Table 3. Performance criteria for training–validation–testing rate.
Training–Validation–Testing Rates (%)MethodsAccSnSpPr
70–10–20LSTM0.95180.94080.96380.9665
60–10–30LSTM0.92410.93190.91530.9243
50–10–40LSTM0.89360.89600.89080.9010
70–10–20CNN0.92720.91240.90810.9279
60–10–30CNN0.90130.89280.88530.8976
50–10–40CNN0.85690.83980.84260.8578
70–10–20SVM0.90300.87330.89160.9042
60–10–30SVM0.88740.88050.89190.8882
50–10–40SVM0.86110.86730.85900.8717
Table 4. The classification performance of the proposed method.
Table 4. The classification performance of the proposed method.
MethodsNumber of FeatureDatasetAcc(%)
Random Forest Model [33]8Private87.00
BFA–GSVCM [13]5Public93.94
CNN–SVM Model [33]5Private86.69
Proposed Method4Private95.18
Table 5. Compare fatigue classification accuracy (in %) based on the combination of denoising threshold function and classification algorithm.
Table 5. Compare fatigue classification accuracy (in %) based on the combination of denoising threshold function and classification algorithm.
Threshold FunctionLSTMCNNSVM
Hard Threshold Function88.7186.7882.90
Soft Threshold Function91.3389.8487.63
Improved Threshold Function95.1892.7290.30
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Share and Cite

MDPI and ACS Style

Wang, J.; Sun, S.; Sun, Y. A Muscle Fatigue Classification Model Based on LSTM and Improved Wavelet Packet Threshold. Sensors 2021, 21, 6369. https://doi.org/10.3390/s21196369

AMA Style

Wang J, Sun S, Sun Y. A Muscle Fatigue Classification Model Based on LSTM and Improved Wavelet Packet Threshold. Sensors. 2021; 21(19):6369. https://doi.org/10.3390/s21196369

Chicago/Turabian Style

Wang, Junhong, Shaoming Sun, and Yining Sun. 2021. "A Muscle Fatigue Classification Model Based on LSTM and Improved Wavelet Packet Threshold" Sensors 21, no. 19: 6369. https://doi.org/10.3390/s21196369

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop