Next Article in Journal
Investigation of Scale Conversion for Inductive Power Transfer in Series-Series Configuration
Previous Article in Journal
Prediction-Based Error Correction for GPU Reliability with Low Overhead
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Driver Fatigue Detection via Differential Evolution Extreme Learning Machine Technique

1
College of Electronic Information, Hangzhou Dianzi University, Hangzhou 310018, China
2
Discipline of Engineering and Energy, Murdoch University, Perth, WA 6150, Australia
3
Centre for Water-Energy-Waste, Murdoch University, Perth, WA 6150, Australia
4
Discipline of IT, Media, and Communications, Murdoch University, Perth, WA 6150, Australia
5
College of Sciences, China Jiliang University, Hangzhou 310018, China
*
Author to whom correspondence should be addressed.
Electronics 2020, 9(11), 1850; https://doi.org/10.3390/electronics9111850
Submission received: 30 September 2020 / Revised: 29 October 2020 / Accepted: 30 October 2020 / Published: 5 November 2020
(This article belongs to the Section Industrial Electronics)

Abstract

:
Fatigue driving (FD) is one of the main causes of traffic accidents. Traditionally, machine learning technologies such as back propagation neural network (BPNN) and support vector machine (SVM) are popularly used for fatigue driving detection. However, the BPNN exhibits slow convergence speed and many adjustable parameters, while it is difficult to train large-scale samples in the SVM. In this paper, we develop extreme learning machine (ELM)-based FD detection method to avoid the above disadvantages. Further, since the randomness of the weight and biases between the input layer and the hidden layer of the ELM will influence its generalization performance, we further apply a differential evolution ELM (DE-ELM) method to the analysis of the driver’s respiration and heartbeat signals, which can effectively judge the driver fatigue state. Moreover, not only will the Doppler radar and smart bracelet be used to obtain the driver respiration and heartbeat signals, but also the sample database required for the experiment will be established through extensive signal collections. Experimental results show that the DE-ELM has a better performance on driver’s fatigue level detection than the traditional ELM and SVM.

1. Introduction

With the rapid development of automobile technology, car ownership has increased rapidly over the past decades. However, the frequent occurrence of road traffic accidents has brought social problems that seriously threaten the safety of human life and property. According to the data of World Health Organization [1], more than 1.2 million people have died in traffic accidents each year, and millions were injured or maimed. Due to the increase in the number of traffic accidents, the severity of this problem has drawn considerable attention from society and governments [2]. Therefore, how to prevent traffic accidents has become one of the most important aspects in the world. According to relevant research, traffic accidents caused by fatigue driving (FD) account for 20–30% of all traffic accidents, which indicates that FD is a major cause of traffic accidents [3]. Drivers normally tend to be distracted, have less activity and slower brain response under fatigue, which will increase the likelihood of traffic accidents [4]. FD detection (FDD) techniques have broad development prospects in the prevention of traffic accidents, and has gradually attracted intensive attention among researchers, automotive industry as well as government organizations.
Researchers have found that drivers’ physiological or driving behaviors can change drastically over a period of time before the accidents caused by FD happen [5]. Based on these changes, many researchers have done some related researches and various methods have been proposed to detect FD, which can be classified into three categories: (1) detecting driver’s physiological information, e.g., electroencephalogram (EEG) [6], electrocardiogram (ECG) [7], electrooculogram (EOG) [8], electromyography (EMG) [9], respiratory signal and heart rate variability [10]; (2) detecting driver’s physical behavior, such as eye state [11], eye blink [12] and gaze direction [13], yawning [14] and head movements [15]; (3) detecting vehicle status and parameter, including lane position, steering wheel angle (SWA), driving speed, acceleration and braking [16,17,18,19,20].
The driver’s physiological signals are highly reliable, with small error and little external interference in responding to the level of fatigue, because it can truly reflect the conditions inside the driver’s body. Dong et al. obtained EEG signals through electrodes embedded in the human head, and analyzed these signals to determine whether the driver is in a fatigue state [21]. Wang et al. analyzed the fusion entropy combining EOG and EEG signals to detect the fatigue state of the driver [22]. Changes in respiration and heartbeat signals have been reported to be associated with driver fatigue and the corresponding detection systems have been developed [23]. However, in order to obtain the driver’s physiological signal, the existing detection methods and devices are normally invasive or contactable [24]. Electrodes need to be attached to drivers, which is unacceptable for many users [25].
Driver behavior based-detection technology has an important advantage that the detection is performed in a non-contact manner without causing interference to the driver, compared with the physiological signal detection method. The main work is to use the camera to obtain the driver’s facial condition and head movements, and use computer vision technology to determine the fatigue state [26]. Cheng et al. collected the videos of 21 participants’ faces and extract many features including number of yawns, blink rate, statistics of blink duration, closing speed, reopening speed and so on, for establishing an FD assessment model [27]. However, these methods are only applicable to the FD late stage, particularly when the driver’s facial changes are obvious and the driving behavior changes have reached a very dangerous stage. In addition, these technologies are also sensitive to external factors, such as driver wearing sunglasses or changes in brightness.
Vehicle parameter-based detection technology also exhibits the advantage of non-contact and is more suitable for real-time systems. However, this technology also has certain limitations. (i) In lane departure detection, road image data need to be captured and processed in real time, which results in the high cost of hardware and computer support development. (ii) Lane shift detection relies purely on external factors such as road markings, weather and light conditions. Although steering wheel detection seems to be a reliable alternative due to its low price and accurate detection without relying on external conditions [28], steering wheel detection is only applicable under very limited conditions, because it is closely related to vehicle type, driver experience, and driver’s condition.
After obtaining a large amount of data through the above methods, they need to be well processed for better determining the driver’s fatigue level. Data processing methods are implemented via mathematical models and machine learning. The machine learning-based implementation method is to train a large amount of driving data obtained from the laboratory and the road, which is called a data-driven algorithm [29]. At present, some researchers have made relevant explorations. Support vector machine (SVM) is a classic kernel learning method which aims to find the best hyperplane to maximize the margin [30]. Hu and Zheng extracted the eyelid-related parameters from the electrooculogram (EOG) as the data input, and utilized SVM to classify the driver state into alert, sleepy and very sleepy [31]. The disadvantage of SVM lies in that it cannot directly support multi-classification and it has difficulties in training large-scale data. The back propagation neural network (BPNN) is a multilayer feedforward neural network trained using the error back propagation algorithm [32], which has been widely used in driver fatigue detection. Ying et al. obtained the relevant parameters of eyes and mouth from the driver’s entire face image as data input and then adopted the BPNN to judge the fatigue state [33]. However, BPNN has significant drawbacks, such as slow convergence speed, many adjustable parameters, and falling into local minimums, which leads to training failure.
Recently, in the field of classifications, in addition to some typical machine learning algorithms, i.e., enhanced k-NN [34], random forests [35], extreme learning machine (ELM) as a new type of single-layer feedforward network (SLFN) learning algorithm has attracted a great deal of attention among researchers. Different from BPNN, ELM has the advantages of faster learning speed, easier modeling, and better generalization performance [36,37,38], which has a great potential to be applied in FDD. However, the random selection of the input weights and hidden thresholds might greatly affect the accuracy of ELM classification. Considering the fact that the differential evolution (DE) algorithm has the characteristics of simple structure, easy implementation, fast convergence and strong robustness [39], the DE algorithm can be used to optimize the initial weights and biases in traditional ELM, which can increase the sensitivity of neurons, obtain the optimal network model, and thereby improve the classification accuracy. The differential evolution extreme learning machine (DE-ELM) algorithm was successfully applied to analog circuit fault diagnosis [40].
Considering the above problems of FDD and motived by ELM and DE, in this paper, we use Doppler radar and smart bracelets to collect the driver’s respiration and heartbeat signals, and then develop an FDD method based on DE-ELM. The rest of this paper is organized as follows. In Section 2, the designed human respiratory and heartbeat signal detection platform and signal acquisition are given. Section 3 introduces the process of data collection, the sample database establishment and classification. In Section 4, DE-ELM-based FDD approach is developed in order to select the best weights and biases of the ELM. In Section 5, experimental results are presented involving traditional ELM, DE-ELM and SVM classification accuracy of fatigue driving samples. Section 6 draws the conclusion.

2. Experimental Platform

The experimental platform for the collection of respiratory and heartbeat signals in this paper is shown in Figure 1. The platform mainly includes three parts: simulated driving system, physiological signal detection system and video signal acquisition system. The simulated driving system consists of two parts: a simulated driving device and a simulated driving environment reproduction system. The simulated driver device includes steering wheel, manual gear, clutch, throttle, brake, etc., while the simulated driving environment reproduction system uses a computer monitor to display the virtual driving environment. The driving simulation software adopts Yijiaxing driving simulation system, and the scenes are taken from on-site video images, such as from urban areas, residential areas, highways, and snowy environments. The front of the screen shows the virtual driving environment, traffic conditions, traffic lights, traffic signs, weather conditions, etc. The information of the car dashboard is displayed at the bottom of the screen. Additionally, physiological signal detection system is used to collect respiratory and heartbeat signals of drivers, video signal acquisition system is used to collect facial information of drivers, and video information serves as an important component for experts to judge fatigue level.
Based on the experimental platform, a set of data collected from the platform have been tested. The corresponding test results are described as follows: (i) the waveform shown in Figure 2 is the signal collected by the radar module without testers. As is seen from the waveform, there is no signal input other than a small amount of noise signal. (ii) The waveform shown in Figure 3 is the signal collected by the radar module with the tester breathing normally. As observed from the waveform, the waveform changes periodically and the signal should be composed of the respiratory signal, the heartbeat signal, and the noise superposition in each cycle. (iii) The waveform shown in Figure 4 is the signal collected by the tester when he holds his breath. It can be seen that the amplitude of signal change is very small, which indicates that the heartbeat signal collected by the radar module is very weak as well as mixed with noise. Therefore, during signal collection, a smart bracelet is added to detect the driver’s heartbeat signal in real time which will be recorded every 2 min.
Through the above tests, it is noted that the Doppler radar module is used to detect human respiratory signals. As for the heartbeat signal, several experiments have shown that the heartbeat signal is very weak and is basically covered by noise, such that the Doppler radar module cannot detect the heartbeat signal. Instead, the smart bracelet is used for heart rate collection. The experiment recruited 7 drivers as experimental test subjects, including 6 men and 1 woman, aged between 22 and 30 years old. It requires good health, normal hearing and vision, and no red–green color blindness. Before the experiment, the tester is required to ensure sufficient sleep time. After the debugging of the whole experimental platform, the test personnel will conduct simulated driving, and each test period will last for 3 h. Throughout the three-hour test, the tester will experience different fatigue states. The Doppler radar module and the smart bracelet collect respiratory and heartbeat signals, and synchronizes video signals to record facial features. At the end of each test, a data set containing respiration and heartbeat signals and corresponding video signals will be obtained through each test. The entire data set will be used for the next expert’s review before classification into different fatigue levels.

3. Sample Library

In Section 2, the data are classified through the facial expert evaluation method. This method was first introduced to the driver’s fatigue assessment and becomes the most practical method for evaluating the fatigue state of drivers [41]. The specific operation procedure is as follows. Firstly, the video signals and the synchronously collected radar and smart bracelet signals are segmented every 2 min and stored randomly. Secondly, three facial experts are selected to score based on multiple indicators such as the rubbing eyes, scratching face, yawning, closing eyes, and adjusting postures, etc. The evaluation result is a continuous value between 0 and 3. The specific classification criteria are described in Table 1. If more than two facial experts judge the same on a certain fatigue level, then the corresponding fatigue level of the driver in the video signal is determined. If the evaluation levels of the three experts are different, then the fatigue level evaluation of this signal needs to be re-evaluated. Finally, three experts will discuss and determine the fatigue level. After the fatigue level is determined, the signals need to be labeled for subsequent neural network learning. The video evaluation results are corresponding to the synchronous radar signals and the smart bracelet signals, which are used as the criteria and basis for fatigue driving evaluation.
After classifying all signals according to the expert evaluation mechanism, it is necessary to conduct filtering processing and discrete Fourier transform (DFT) for each group of data. A zero-phase indefinite impulse response (IIR) filter is used for the filtering algorithm, which can completely eliminate the signal phase distortion and improve the real-time performance of detection at the cost of increasing the appropriate computation [42]. After the filtering is completed, DFT processing of the signal will continue to obtain the spectrum diagram. Then, both the frequency and amplitude of the signal are extracted. Figure 3 and Figure 5 show the respiratory signal and its amplitude–frequency characteristics, respectively.
Finally, the following characteristic values are determined as the training sample data: R C , R A , H R , where R C is the respiratory cycle, R A is the respiratory amplitude, and H R indicates the heart rate.
The sample library can be built as:
X = ( R C ( s ) R A ( s ) H R ( s ) ) ( i )
T = ( h ( s ) h ( s 1 ) h ( s 2 ) h ( s 3 ) ) ( i )
S = { ( X ( i ) T ( i ) ) } ,   i [ 1 ,   N ]
where
X: input data.
T: output label corresponding to X.
h(x): impact function which defined as:
h ( x ) = { 1 ,   x = 1 0 ,   x 1
where s is the fatigue state with the value as 1, 2, 3 or 4, which correspondingly represents sober state, first-level fatigue state, second-level fatigue state and third-level fatigue state, i is the sample index, and N is the total sample size. A total of 720 sets of respiration and heartbeat data were collected in this experiment.
After obtaining the complete data set S , we also need to divide it into training set and test set by using a subject-wise method. Data classification should follow the following three principles: (i) randomly assigned, (ii) training set sample size: test set sample size = 7:3, and (iii) the same number of samples per fatigue level.
In the following section, we will introduce the basic principles of ELM and DE-ELM in detail and further give the DE-ELM-based FDD approach.

4. Introduction of Classification Method

4.1. Extreme Learning Machine

ELM is an SLFN proposed by Guangbin Huang [43], which consists of the following parts: dimension of input feature vector n, total number of samples N, number of hidden layer neurons L, and dimension of outputs m, data set S = [ X T ] . It can be represented by following matrix:
Input data:
X = [ X ( 1 ) X ( 2 ) X ( N ) ] n × N
X ( i ) = ( x 1 x 2   x n ) ( i ) T
Output label:
T = [ T ( 1 ) T ( 2 ) T ( N ) ] T N × m
T ( i ) = ( t 1 t 2 t m ) ( i ) T
Hidden layer input weight matrix is W :
W = [ W 1 W 2 W L ] T L × n
W j = ( w j 1 w j 2   w j n ) T
Hidden layer bias is b :
b = ( b 1 b 2   b L ) T
Hidden layer output weight matrix is β :
β = [ β 1 β 2 β L ] T L × m
β k = ( β k 1 β k 2 β k m ) T
The activation function selected in this paper is sigmoid function which is defined as:
g ( x ) = 1 1 + e x
By using the activation function, the nonlinear characteristics can be added to make learning faster and more efficient [44]. Thus, we have
H j ( i ) = g ( W j X ( i ) + b j ) = 1 1 + e ( W j X ( i ) + b j )
The output O ( i ) can be expressed as:
H ( i ) β = O ( i ) ,   i = 1 , , N
The goal of the neural network learning is to minimize the output error
i = 1 N O ( i ) T ( i ) m × N = 0
That is, there exist β , W j and b j , such that:
H ( i ) β = T ( i ) , i = 1 , 2 , , N
For the entire training set, Equation (17) can be expressed in matrix form as:
H β = T
where H is the hidden layer output matrix:
H ( W , b , X ) = [ g ( W 1 X ( 1 ) + b 1 ) g ( W L X ( 1 ) + b L ) g ( W 1 X ( N ) + b 1 ) g ( W L X ( N ) + b L ) ] N × L
For fixed input weights and the hidden layer biases, to train an SLFN is simply equivalent to find a least-squares solution β ^ of the linear system H β = T :
| | H β ^ T | | = min β | | H β T | |
According to the minimum norm criterion, the solution is obtained by finding the least squares:
β ^ = ( H ) T
where ( H ) is the Moore–Penrose generalized inverse of the hidden layer output matrix H . In summary, when inputting training data and randomly initializing the input weight matrix, the output weight matrix can be obtained through Equation (22). The design of the ELM neural network model for fatigue driving detection is shown Figure 6. It is noted that the ELM possesses the advantages of high learning efficiency and strong generalization ability and thus is widely used in classification, regression, clustering, feature learning and other problems [45]. Since the input weights and the hidden layer biases of the ELM are randomly assigned, these weights and biases may not be the optimal choices relative to the input data. For practical applications, in order to enable the neural network to have better generalization performance, more hidden layer neurons may be needed, thereby increasing the complexity of the network. To compensate for these shortcomings, we will introduce differential evolution algorithms in the following section to optimize the weights and biases of the ELM, such that the optimal network structure can be obtained.

4.2. Differential Evolution ELM (DE-ELM)

Differential evolution, proposed by Storn and Price in 1995, is a simple yet powerful evolutionary algorithm (EA) [39]. The basic idea of the optimization algorithm is as follows: starting from a randomly generated initial population, a new individual is generated by summing the vector difference of any two individuals in the population with a third individual, and then comparing the new individual with the corresponding individual in the contemporary population. The corresponding fitness is better than the fitness of the current individual, so the new individual will replace the old individual in the next generation, otherwise the old individual will still be saved. Through continuous evolution, it will keep the good individuals, eliminate the bad individuals, and guide the search to the optimal solution. Compared with most of the available evolutionary algorithms, it exhibits the advantages of simple structure, fast convergence, few adjustable parameters, and strong robustness, etc.
Next, we show the detailed mathematical description of DE algorithm in the following.
Step 1: Initialization. We randomly generate N P individuals to form the primary population, where D is the dimension of the population. The i-th individual θ i ( g ) in the g-th iteration can be marked as:
θ i ( g ) = { θ i , j ( g ) } , i = 1 , 2 , , N P ; j = 1 , 2 D
The value of the j-th dimension of the i-th individual θ i , j ( g ) can be obtained by the following equation:
θ i , j ( g ) = θ m i n + rand ( 0 , 1 ) · ( θ m a x θ m i n )
where θ m a x and θ m i n represent the upper and lower bounds of each parameter: θ m i n θ i , j ( g )   θ m a x and rand ( 0 , 1 ) represents a random number uniformly distributed in the interval (0,1).
Step 2: Individual Evaluation. In this step, the entire population is evaluated, that is, the fitness function value of each individual in the population is calculated.
Step 3: Mutation Operation. DE achieves the mutation of individuals through a differential strategy, which is also an important difference from genetic algorithms. The differential strategy used in this paper is to randomly select two different individuals in the population, scale their vector differences, and perform vector synthesis with the individuals that need to be mutated, that is:
v i ( g + 1 ) = θ r 1 ( g ) + F · ( θ r 2 ( g ) θ r 3 ( g ) )
where r 1 , r 2 , r 3 are randomly chosen in the range [ 1 , N P ] , with r 1 r 2 r 3 i , θ r 2 ( g ) θ r 3 ( g ) is the differential variation, v i ( g + 1 ) is the new mutation individual, and the constant factor F is a scaling parameter, which is used to control the amplification of the differential variation.
In the mutation process, in order to ensure the validity of the solution, it must be determined whether the parameters of each individual are between the maximum and minimum values. If this condition is not met, they will be regenerated by using Equation (24).
Step 4: Crossover Operation. The crossover to differential evolution algorithm is introduced for the sake of increasing the diversity of generation. Crossover operation is described as follows:
u i , j ( g + 1 ) = { v i , j ( g + 1 ) rand ( 0 , 1 ) C R   or   j = j r a n d θ i , j ( g ) otherwise
where C R is the crossover probability and j r a n d is a random integer generated in the set { 1 , 2 , , D } .
Step 5: Selection Operation. The purpose of this step is to generate individuals of the population in g + 1 generation. Among the target individual θ i ( g ) and u i ( g + 1 ) obtained in the previous step, the one with better effect is selected as the individual θ i ( g + 1 ) of the g + 1 generation population according to the fitness function:
θ i ( g + 1 ) = { u i ( g + 1 ) i f   f ( u i ( g + 1 ) ) f ( θ i ( g ) ) θ i ( g ) otherwise
where f is the fitness function. The smaller fitness function value is selected as the individual of the g + 1 generation population, which is used to replace the previous individual. Meanwhile, g = g + 1 .
Step 6: Stop Test. Judge whether the termination condition is reached or the maximum evolutionary algebra is reached. If so, the evolution is terminated, and the optimal parameters obtained at this time are output as the solution. Otherwise, the program will jump to Step 2 for re-execution.
In order to reduce the number of hidden layer neurons and improve the generalization performance of the neural network, the global optimization capability of the DE algorithm is applied to the reasonable selection of the input weights and the hidden layer biases of the ELM.
Figure 7 shows the algorithm flow of the DE-ELM. Then, the optimization problem to be solved is to perform min f ( x ) , where f ( x ) is the fitness function. Suppose that the cost function ( E ) is root mean squared error (RMSE) [46]:
E = j = 1 N i = 1 L | | β i g ( w i x j + b j ) t j | | 2 2 N × m
The RMSE on the whole training dataset is used as the fitness function. In the following section, we will carry out experiments to compare the classification effect of ELM, DE-ELM and SVM on the fatigue driving dataset.
Remark 1: Before training, we need to determine the parameters of the DE algorithm and the corresponding parameter selection criteria is given as follows. (i) The population size N P refers to the number of individuals in the population. When the population is large, the entire population exhibits diversity, which makes a larger search space and greater possibility of searching for the optimal solution, but the convergence rate will be reduced. On the other hand, when the population is small, the convergence rate is fast, but sometimes the global optimal solution cannot be obtained. (ii) The scaling parameter F is used to control the amplification of the differential variation, which plays a moderating role in the local search and global search of the algorithm. When F has large value, differential variation will have big impact on the mutation individual seen from (25) resulting in large disturbances, which is beneficial to maintain population diversity and global search capabilities. However, the search efficiency will be lower and the accuracy of the global optimal solution obtained will be lower. Smaller value of F may lead to loss of population diversity and the algorithm is prone to fall into a local optimum causing early convergence. (iii) Crossover probability C R can determine whether members in a population perform crossover operations, which has an important impact on population diversity.

5. Results

In the experiments, after many trials and comparisons, the following parameters for the DE algorithm were determined: N P = 20 ,   F = 0.7 ,   C R = 0.8 ,   g m a x = 30 . The experimental results are listed in Table 2. As seen from Table 2, with the increase in the hidden layer nodes number, the training accuracy and the testing accuracy of both the ELM and the DE-ELM are well improved. For the DE-ELM, when the number of hidden layer nodes is 150, it has better performance than the ELM with 200 nodes, in terms of both the training accuracy and the testing accuracy. It can also be seen that the DE-ELM using fewer hidden layer nodes can achieve better classification results than the ELM with more hidden layer nodes. The DE-ELM not only reduces network complexity, but also achieves stronger generation ability. Moreover, when the number of hidden layer nodes increases to 150 and 200, the training and test accuracies of SVM cannot compete with those of ELM and DE-ELM.
In order to further verify the effects of three approaches on the test set, the classification results on each fatigue state are shown in Figure 8, Figure 9, Figure 10, Figure 11, Figure 12, Figure 13 and Figure 14, where the category labels of 1–4 represent the driver’s sober state, first-level fatigue state, second-level fatigue state and third-level fatigue state, respectively. The classification results of the ELM and the DE-ELM for 100, 150, and 200 hidden layer nodes are shown in Figure 8, Figure 9, Figure 10, Figure 11, Figure 12 and Figure 13, while the classification results of the SVM are shown in Figure 14. It can be seen that, for the test set samples, the DE-ELM prediction outputs match with the actual outputs much better than the ELM and the SVM. In order to clearly evaluate the classification accuracy of the fatigue state, Table 3 shows the classification accuracy of three approaches for each state on the test set when the number of hidden layer nodes is 200. It is worth noting that the recognition rate of the DE-ELM for various fatigue states all exceed 90%, which achieves the best classification performance. In detail, although the ELM obtains similar performance to the DE-ELM for the first and second level fatigue state, its classification accuracy for sober state and third-level fatigue state are not as good as DE-ELM. In addition, the classification accuracies of the DE-ELM are a lot better than the ones of the SVM for third-level fatigue states, except for the sober state in which similar accuracy is obtained. It clearly demonstrates that the developed DE-ELM method in this paper exhibits the most excellent classification performance of fatigue driving dataset compared to its ELM and SVM counterparts.
Please note that the criterion for determining the level of fatigue driving in this article is based on the evaluation method of facial video experts. This method only subjectively recognizes and judges the facial expression movement characteristics of the tester. It may not accurately and objectively determine the driver’s fatigue state, which also leads to a lower recognition rate of fatigue level in this experiment. The more input feature values and training samples of a classification model we have, the higher the classification accuracy we will obtain. Due to the limited conditions of this experiment, fewer input feature values and insufficient training samples have limited the recognition rate of fatigue level in this paper. Further studies on determining the fatigue level and the selection of input feature values will be carried out to obtain a higher recognition rate for fatigue driving.

6. Conclusions

In this paper, the ELM-based FDD algorithm has been developed to determine the driver’s fatigue status. Considering that the input weights and hidden layer biases are randomly obtained in the ELM leading to the degraded generalization performance, the weights and biases are optimized via the DE algorithm, such that the sensitivity of neurons is increased and the classification accuracy can be improved. The driver’s respiratory and heartbeat signals are collected by Doppler radar and the smart bracelet in this work, which has little impact on the driver’s normal operation. Experimental studies have demonstrated that the DE-ELM has significantly improved the accuracy of driver’s fatigue state detections compared to the traditional ELM and SVM approaches.

Author Contributions

Conceptualization, L.C.; data curation, G.W.; formal analysis, Z.Z.; funding acquisition, L.C.; investigation, L.C., A.Y. and X.Z. (Xuefeng Zheng); methodology, H.W.; project administration, L.C.; resources, A.Y.; software, X.Z. (Xiaojie Zhi) and X.Z. (Xuefeng Zheng); supervision, H.W.; validation, Z.Z.; writing—original draft, X.Z. (Xiaojie Zhi); writing—review and editing, H.W. All authors have read and agreed to the published version of the manuscript.

Funding

This work is supported by National Natural Science Foundation of China (No. 61771178).

Conflicts of Interest

The authors declare no conflict of interest.

References

  1. WHO. Global Status Report on Road Safety 2015; WHO: Geneva, Switzerland, 2015. [Google Scholar]
  2. Williamson, A.; Lombardi, D.A.; Folkard, S.; Stutts, J.; Courtney, T.K.; Connor, J.L. The link between fatigue and safety. Accid. Anal. Prev. 2011, 43, 498–515. [Google Scholar] [CrossRef] [PubMed]
  3. Rau, P.S. Drowsy Driver Detection and Warning System for Commercial Vehicle Drivers: Field Operational Test Design, Data Analyses, and Progress; National Highway Traffic Safety Administration: Washington, DC, USA, 2005; p. 5.
  4. Yu, X.; Li, X. A real-time driving fatigue detection system based on portable EEG. Mod. Inf. Technol. 2018, 2, 45–47. [Google Scholar]
  5. Kaplan, S.; Guvensan, M.A.; Yavuz, A.G.; Karalurt, Y. Driver behavior analysis for safe driving: A survey. IEEE Trans Intell. Transp. Syst. 2015, 16, 3017–3032. [Google Scholar] [CrossRef]
  6. Balandong, R.P.; Ahmad, R.F.; Saad, M.N.M.; Malik, A.S. A review on EEG-based automatic sleepiness detection systems for driver. IEEE Access 2018, 6, 22908–22919. [Google Scholar] [CrossRef]
  7. Jeong, J.W.; Kim, H.K.; Lee, Y.J.; Jung, W.W.; Harashima, F.; Lee, M.H. The implementation of the autonomous guided vehicle driving system for durability test. In Proceedings of the IEEE Intelligent Transportation Systems, Dearborn, MI, USA, 1–3 October 2000; pp. 101–106. [Google Scholar]
  8. Zhang, Y.F.; Gao, X.-Y.; Zhu, J.-Y.; Zheng, W.-L.; Lu, B.-L. A novel approach to driving fatigue detection using forehead EOG. In Proceedings of the 7th International IEEE/EMBS Conference on Neural Engineering (NER), Montpellier, France, 22–24 April 2015; pp. 707–710. [Google Scholar]
  9. Katsis, C.D.; Ntouvas, N.E.; Bafas, C.G.; Fotiadis, D. Assessment of muscle fatigue during driving using surface EMG. In Proceedings of the IASTED International Conference on Biomedical Engineering, Innsbruck, Austria, 16–18 February 2004; pp. 259–262. [Google Scholar]
  10. Patel, M.; Lal, S.K.L.; Kavanagh, D.; Rossiter, P. Applying neural network analysis on heart rate variability data to assess driver fatigue. Expert Syst. Appl. 2011, 38, 7235–7242. [Google Scholar] [CrossRef]
  11. Mandal, B.; Li, L.; Wang, G.S.; Lin, J. Towards detection of bus driver fatigue based on robust visual analysis of eye state. IEEE Trans. Intell. Transp. Syst. 2017, 18, 545–557. [Google Scholar] [CrossRef]
  12. Ahmad, R.; Borole, J.N. Drowsy driver identification using eye blink detection. Int. J. Comput. Sci. Inf. Technol. 2015, 6, 270–274. [Google Scholar]
  13. Choi, I.H.; Kim, Y.G. Head pose and gaze direction tracking for detecting a drowsy driver. In Proceedings of the IEEE International Conference on Big Data and Smart Computing, Bangkok, Thailand, 15–17 January 2014; pp. 241–244. [Google Scholar]
  14. Omidyeganeh, M. Yawning Detection Using Embedded Smart Cameras. IEEE Trans. Instrum. Meas. 2016, 65, 570–582. [Google Scholar] [CrossRef]
  15. Mittal, A.; Kumar, K.; Dhamija, S.; Kaur, M. Head movement-based driver drowsiness detection: A review of state-of-art techniques. In Proceedings of the IEEE International Conference on Engineering and Technology, Coimbatore, India, 17–18 March 2016; pp. 903–908. [Google Scholar]
  16. Liang, Y.; Reyes, M.; Lee, J. Real-time detection of driver cognitive distraction using support vector machine. IEEE Trans. Intell. Transp. Syst. 2007, 8, 340–350. [Google Scholar] [CrossRef]
  17. Liu, C.; Hosking, S.; Lenne, M. Predicting driver drowsiness using vehicle measures: Recent insights and future challenges. J. Safety Res. 2009, 40, 239–245. [Google Scholar] [CrossRef]
  18. Forsman, P.; Vila, B.; Short, R.; Mott, C.; van Dongen, H. Efficient driver drowsiness detection at moderate levels of drowsiness. Accid. Anal. Prevent. 2013, 50, 341–350. [Google Scholar] [CrossRef]
  19. Wang, J.; Zhu, S.; Gong, Y. Driving safety monitoring using semisupervised learning on time series data. IEEE Trans. Intell. Transp. Syst. 2010, 11, 728–737. [Google Scholar] [CrossRef]
  20. Wu, B.-F.; Chen, Y.-H.; Yeh, C.-H.; Li, Y.-F. Reasoning-based framework for driving safety monitoring using driving event recognition. IEEE Trans. Intell. Transp. Syst. 2013, 14, 1231–1241. [Google Scholar] [CrossRef] [Green Version]
  21. Dong, N.; Li, Y.; Gao, Z.; Ip, W.H.; Yung, K.L. A WPCA-Based Method for Detecting Fatigue Driving From EEG-Based Internet of Vehicles System. IEEE Access 2019, 7, 124702–124711. [Google Scholar] [CrossRef]
  22. Wang, H.; Wu, C.; Li, T.; He, Y.; Chen, P.; Bezerianos, A. Driving Fatigue Classification Based on Fusion Entropy Analysis Combining EOG and EEG. IEEE Access 2019, 7, 61975–61986. [Google Scholar] [CrossRef]
  23. Yilmaz, T.; Foster, R.; Hao, Y. Detecting vital signs with wearable wireless sensors. Sensors (Basel) 2010, 10, 10837–10862. [Google Scholar] [CrossRef]
  24. Leng, L.B.; Giin, L.B.; Chung, W.-Y. Wearable driver drowsiness detection system based on biomedical and motion sensors. In Proceedings of the IEEE SENSORS, Busan, Korea, 1–4 November 2015; pp. 1–4. [Google Scholar]
  25. Zhao, C.; Zhao, M.; Liu, J. Electroencephalogram and electrocardiograph assessment of mental fatigue in a driving simulator. Accident Anal. Prev. 2012, 45, 83–90. [Google Scholar] [CrossRef]
  26. Gu, W.H.; Zhu, Y.; Chen, X.D.; He, L.F.; Zheng, B.B. Hierarchical CNN-based real-time fatigue detection system by visual-based technologies using MSP model. IET Image Process. 2018, 12, 2319–2329. [Google Scholar] [CrossRef]
  27. Cheng, Q.; Wang, W.; Jiang, X.; Hou, S.; Qin, Y. Assessment of Driver Mental Fatigue Using Facial Landmarks. IEEE Access 2019, 7, 150423–150434. [Google Scholar] [CrossRef]
  28. Li, Z.; Li, S.E.; Li, R.; Cheng, B.; Shi, J. Online detection of driver fatigue using steering wheel angles for real driving conditions. Sensors 2017, 17, 495. [Google Scholar] [CrossRef] [Green Version]
  29. Sikander, G.; Anwar, S. Driver Fatigue Detection Systems: A Review. IEEE Trans. Intell. Transp. Syst. 2019, 20, 2339–2352. [Google Scholar] [CrossRef]
  30. Cristianini, N.; Shawetaylor, J. An introduction to support vector machines and other kernel-based learning methods. Kybernetes 2001, 30, 103–115. [Google Scholar]
  31. Hu, S.; Zheng, G. Driver drowsiness detection with eyelid related parameters by support vector machine. Expert Syst. Appl. 2009, 36, 7651–7658. [Google Scholar] [CrossRef]
  32. Rabik, M.M.; Balasubadra, K. Distributive Aware Probabilistic Back Propagation (DAPBP) Routing Scheme for Vehicular Networks. Wirel. Pers. Commun. 2017, 97, 483–493. [Google Scholar] [CrossRef]
  33. Ying, Y.; Jing, S.; Wei, Z. The Monitoring Method of Driver’s Fatigue Based on Neural Network. In Proceedings of the 2007 International Conference on Mechatronics and Automation, Harbin, China, 5–8 August 2007; pp. 3555–3559. [Google Scholar]
  34. Nguyen, B.P.; Tay, W.; Chui, C. Robust Biometric Recognition from Palm Depth Images for Gloved Hands. IEEE Trans. Hum. Mach. Syst. 2015, 45, 799–804. [Google Scholar] [CrossRef]
  35. Breiman, L. Random Forests. Mach. Learn. 2001, 45, 5–32. [Google Scholar] [CrossRef] [Green Version]
  36. Huang, G.; Zhu, Q.; Siew, C. Extreme learning machine: Theory and applications. Neurocomputing 2006, 70, 489–501. [Google Scholar] [CrossRef]
  37. Hu, Y.; Wang, H.; Cao, Z.; Zheng, J.; Ping, Z.; Chen, L.; Jin, X. Extreme-learning- machine-based FNTSM control strategy for electronic throttle. Neural Comput. Appl. 2019, 32, 14507–14518. [Google Scholar] [CrossRef]
  38. Zhang, J.; Wang, H.; Cao, Z.; Zheng, J.; Yu, M.; Yazdani, A.; Shahnia, F. Fast nonsingular terminal sliding mode control for permanent magnet linear motor via ELM. Neural Comput. Appl. 2019, 32, 1–11. [Google Scholar] [CrossRef]
  39. Storn, R.; Price, K. Differential evolution-A simple and efficient heuristic for global optimization over continuous spaces. J. Glob. Optim. 1997, 11, 341–359. [Google Scholar] [CrossRef]
  40. Zhang, L.; Qin, Q.; Shang, Y.; Chen, S.; Zhao, S. Application of DE-ELM in analog circuit fault diagnosis. In Proceedings of the 2016 Prognostics and System Health Management Conference (PHM-Chengdu), Chengdu, China, 19–21 October 2016; pp. 1–6. [Google Scholar]
  41. Wierwille, W.W.; Ellsworth, L.A. Evaluation of driver drowsiness by trained raters. Accid. Anal. Prev. 1994, 26, 571–581. [Google Scholar] [CrossRef]
  42. Chen, L.; Yang, X.; Wu, J. Non-contact physiological signal monitoring system based on Doppler radar. Ieice Electron. Express 2017, 14, 1–9. [Google Scholar] [CrossRef] [Green Version]
  43. Huang, G.; Zhu, Q.; Siew, C. Extreme learning machine: A new learning scheme of feedforward neural networks. In Proceedings of the International Joint Conference on Neural Networks, Budapest, Hungary, 25–29 July 2004; Volume 2, pp. 985–990. [Google Scholar]
  44. Cai, Z.; Huang, L. Finite-Time Stabilization of Delayed Memristive Neural Networks: Discontinuous State-Feedback and Adaptive Control Approach. IEEE Trans. Neural Netw. Learn. Syst. 2018, 29, 856–868. [Google Scholar] [CrossRef]
  45. Tang, J.; Deng, C.; Huang, G. Extreme Learning Machine for Multilayer Perceptron. IEEE Trans. Neural Netw. Learn. Syst. 2016, 27, 809–821. [Google Scholar] [CrossRef]
  46. Zhu, Q.; Qin, A.K.; Suganthan, P.N.; Huang, G. Evolutionary extreme learning machine. Pattern. Recogn. 2005, 38, 1759–1763. [Google Scholar] [CrossRef]
Figure 1. Physiological signal acquisition experimental platform.
Figure 1. Physiological signal acquisition experimental platform.
Electronics 09 01850 g001
Figure 2. Signal without testers.
Figure 2. Signal without testers.
Electronics 09 01850 g002
Figure 3. Tester’s signal when breathing normally.
Figure 3. Tester’s signal when breathing normally.
Electronics 09 01850 g003
Figure 4. Tester’s signal when holding their breath.
Figure 4. Tester’s signal when holding their breath.
Electronics 09 01850 g004
Figure 5. Amplitude–frequency characteristics of respiratory signals.
Figure 5. Amplitude–frequency characteristics of respiratory signals.
Electronics 09 01850 g005
Figure 6. Extreme learning machine (ELM) model based on fatigue driving detection.
Figure 6. Extreme learning machine (ELM) model based on fatigue driving detection.
Electronics 09 01850 g006
Figure 7. Algorithm flow of differential evolution ELM (DE-ELM) model.
Figure 7. Algorithm flow of differential evolution ELM (DE-ELM) model.
Electronics 09 01850 g007
Figure 8. ELM (100 nodes)-based fatigue state detection results.
Figure 8. ELM (100 nodes)-based fatigue state detection results.
Electronics 09 01850 g008
Figure 9. DE-ELM (100 nodes)-based fatigue state detection results.
Figure 9. DE-ELM (100 nodes)-based fatigue state detection results.
Electronics 09 01850 g009
Figure 10. ELM (150 nodes)-based fatigue state detection results.
Figure 10. ELM (150 nodes)-based fatigue state detection results.
Electronics 09 01850 g010
Figure 11. DE-ELM (150 nodes)-based fatigue state detection results.
Figure 11. DE-ELM (150 nodes)-based fatigue state detection results.
Electronics 09 01850 g011
Figure 12. ELM (200 nodes)-based fatigue state detection results.
Figure 12. ELM (200 nodes)-based fatigue state detection results.
Electronics 09 01850 g012
Figure 13. DE-ELM (200 nodes)-based fatigue state detection results.
Figure 13. DE-ELM (200 nodes)-based fatigue state detection results.
Electronics 09 01850 g013
Figure 14. SVM-based fatigue state detection results.
Figure 14. SVM-based fatigue state detection results.
Electronics 09 01850 g014
Table 1. Fatigue level and evaluation criteria.
Table 1. Fatigue level and evaluation criteria.
Fatigue GradeFatigue StateFatigue Grade Criteria
0SoberGood mental state, no fatigue performance.
IMild fatigueQuick blinking, rubbing eyes, yawing, inattention.
IIModerate fatigueDecreased eye rotation, frequent blinking, frequent yawning, reduced limb activity and marked inattention.
IIISevere fatigueEyelid fatigue, drowsiness, into a sleepy state and even subconscious operation behavior.
Table 2. Classification accuracy of ELM, DE-ELM and support vector machine (SVM).
Table 2. Classification accuracy of ELM, DE-ELM and support vector machine (SVM).
Classification ModelHidden Layer Nodes NumberTraining Accuracy (%)Testing Accuracy (%)
ELM10089.0987.96
15094.2590.28
20095.0491.20
DE-ELM10090.0889.35
15095.2491.67
20095.8393.06
SVM-90.8789.35
Table 3. Classification accuracy of ELM, DE-ELM (200 nodes) and SVM for each state.
Table 3. Classification accuracy of ELM, DE-ELM (200 nodes) and SVM for each state.
Level of Driver FatigueELM Accuracy (%)DE-ELM Accuracy (%)SVM Accuracy (%)
Sober state88.8994.4496.30
First-level fatigue state92.5992.5990.74
Second-level fatigue state90.7490.7479.63
Third-level fatigue state92.5994.4490.74
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Share and Cite

MDPI and ACS Style

Chen, L.; Zhi, X.; Wang, H.; Wang, G.; Zhou, Z.; Yazdani, A.; Zheng, X. Driver Fatigue Detection via Differential Evolution Extreme Learning Machine Technique. Electronics 2020, 9, 1850. https://doi.org/10.3390/electronics9111850

AMA Style

Chen L, Zhi X, Wang H, Wang G, Zhou Z, Yazdani A, Zheng X. Driver Fatigue Detection via Differential Evolution Extreme Learning Machine Technique. Electronics. 2020; 9(11):1850. https://doi.org/10.3390/electronics9111850

Chicago/Turabian Style

Chen, Long, Xiaojie Zhi, Hai Wang, Guanjin Wang, Zhenghua Zhou, Amirmehdi Yazdani, and Xuefeng Zheng. 2020. "Driver Fatigue Detection via Differential Evolution Extreme Learning Machine Technique" Electronics 9, no. 11: 1850. https://doi.org/10.3390/electronics9111850

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