Enhanced CNN-LSTM Model for RUL Prediction
Enhanced CNN-LSTM Model for RUL Prediction
ABSTRACT
Remaining useful life (RUL) prediction is one of the key technologies of aircraft
prognosis and health management (PHM) which could provide better maintenance
decisions. In order to improve the accuracy of aircraft engine RUL prediction under
real flight conditions and better meet the needs of PHM system, we put forward an
improved CNN-LSTM model based on the convolutional block attention module
(CBAM). First, the features of aircraft engine operation data are extracted by multi-
layer CNN network, and then the attention mechanism is processed by CBAM in
channel and spatial dimensions to find key variables related to RUL. Finally, the hidden
relationship between features and service time is learned by LSTM and the predicted
RUL is output. Experiments were conducted using C-MPASS dataset. Experimental
results indicate that our prediction model has feasibility. Compared with other state-
of-the-art methods, the RMSE of our method decreased by 17.4%, and the score of the
prediction model was improved by 25.9%.
Subjects Algorithms and Analysis of Algorithms, Artificial Intelligence, Data Mining and
Machine Learning, Data Science
Keywords Aircraft engine, Remaining useful life, Convolutional block attention module,
Convolutional neural network, LSTM
Submitted 13 June 2022
Accepted 15 August 2022
Published 30 August 2022 INTRODUCTION
Corresponding author The aircraft engine is one of the core equipment of aircraft, which largely determines
Zhuojian Wang,
zhuojianw1974@[Link]
the flight performance and flight safety of aircraft. With the development of maintenance
theory, sensing and communication technology, aircraft prognosis and health management
Academic editor
Ka-Chun Wong (PHM) is increasingly a significant way to improve the reliability and safety of aircraft (Li,
Additional Information and Verhagen & Curran, 2020). PHM collects operational data in real time through sensors, and
Declarations can be found on realizes the functions of aircraft condition monitoring, fault prediction and maintenance
page 16
planning, among which the remaining useful life(RUL) prediction is a vital part of PHM
DOI 10.7717/peerj-cs.1084 system (Che et al., 2019).
Copyright At present, the RUL prediction methods can be roughly divided into two categories. They
2022 Li et al. are methods based on the physical model and methods based on monitoring data (Peng,
Distributed under Ye & Chen, 2019; Singh et al., 2020; Ren et al., 2018; Shankar kumar, Kumaraswamidhas &
Creative Commons CC-BY 4.0 Laha, 2021). The physical model-based method is usually for constructing and estimating
OPEN ACCESS the degradation model of equipment mathematically by summarizing the rules of failure
How to cite this article Li H, Wang Z, Li Z. 2022. An enhanced CNN-LSTM remaining useful life prediction model for aircraft engine
with attention mechanism. PeerJ Comput. Sci. 8:e1084 [Link]
mechanism (Lei et al., 2018). This method is more suitable for the crack propagation of
metal components, etc. Ye et al. (2018) studied a fatigue life prediction method based on
multi-scale crack propagation model, and the prediction ability of this method was verified
by nickel-based GH4169 alloy. However, for complex systems and equipment, the method
based on physical model is no longer applicable, mainly because in industrial practice,
we don’t fully grasp the failure mechanism of equipment, and the process of establishing
mathematical model is very complicated. Therefore, more studies on RUL prediction
mainly focus on data-based methods, which could be divided into three categories (Muneer
et al., 2021; Sun et al., 2021; Shankar kumar, Kumaraswamidhas & Laha, 2021). One is to
describe the degradation of equipment through stochastic process. Gao et al. (2019) studied
the characteristics of light-emitting diode driving power, and proposed a RUL prediction
method based on Wiener process. Lin et al. (2021) proposed a novel method to model
two-phase degenerate behavior of products based on the nonlinear Wiener process. The
second is to establish RUL prediction model by machine learning. Ordez et al. (2019)
proposed a RUL prediction model which combined support vector regression model
(SVM) and auto-regressive integrated moving average (ARIMA) model. Pan et al. (2020)
put forward a two-stage prediction method based on extreme learning machine (ELM).
Their method could quickly and accurately predict the RUL of rolling bearings. Kamat,
Sugandhi & Kumar (2021) used the unsupervised machine learning for anomaly trend
analysis of bearings, and the semi-supervised method for RUL prediction. The third
category is the deep learning prediction method that has emerged with the development
of artificial intelligence. Wen et al. (2021) established a deep learning architecture based
on bidirectional gated recurrent unit (BGRU), and proposed a domain-adaptive RUL
prediction model. Guo et al. (2017), Kim & Liu (2021) and Khumprom, Grewell & Yodo
(2020), respectively use three network models, deep belief network (DBN), bayesian deep
learning framework, and feedforward neural network(FNN) to establish different RUL
prediction models, and all of them achieved good prediction results.
Aircraft engine monitoring data has the features of large amount of data and high
dimensionality, and deep learning has better feature extraction ability in processing such
data (Mao et al., 2022; Zhao & Wang, 2021), so the prediction method of deep learning
is more suitable for aircraft engine RUL prediction. At present, some scholars have
also conducted relevant research. Hu et al. (2021) built several new DBRNNs with new
network configurations and designed a new customized loss function for RUL prediction
of aircraft engine. Chui, Gupta & Vasant (2021) proposed a RUL prediction algorithm
in combination with RNN and LSTM, used non-dominated sorting genetic algorithm II
(NSGA-II) to optimize it, and achieved better results. Li et al. (2021) proposed a new
Bayesian deep learning nework considering the epistemic uncertainty and aleatoric
uncertainty to solve the prognostic uncertainty problem. Ragab et al. (2021) addressed
the domain shift problem based on contrastive adversarial domain adaptation and got
better results in experiments. However, the above methods are all based on the research
under the stable working condition; without considering the influence of the working
condition change of the aircraft engine on the prediction accuracy of RUL under the real
flight condition, it still faces the following problems in the practical application process:
THEORETICAL METHODS
Convolutional Neural Network (CNN)
CNN is a special multi-layer perceptron (MLP), which has the traits of local connection
and weight sharing. Because of its advantages in image processing, it is widely used in
computer vision, image classification and other applications (Zhou, 2020). CNN can be
divided into one dimensional, two dimensional and three dimensional CNN (1-D CNN,
2-D CNN and 3-D CNN). We mainly use 2-D CNN, which usually includes convolution
layer, activation layer and pooling layer. Figure 1 shows a simple convolutional neural
network.
The crucial part of CNN is convolution layer. Convolution layer convolves input data
through local connection and weight sharing, and its function is to extract features by
shifting convolution filters on original data. Convolution filter is also called receptive field.
Convolution filter moves along all dimensions of input data, calculates the weight and the
dot product of input, and then adds bias to realize convolution operation. The specific
convolution operation is:
yn = wm ∗ xn + bm (1)
Input data
Y
X
where wm and bm ∈ Rh×w represent the weight and bias of the m-th filter, respectively,
and their sizes h × w are specified when designing the network, xn is the n-th region of the
input data, and yn is the convolution output of the filter.
As can be seen from Eq. (1), each convolution filter can generate a corresponding feature
map. When multiple filters are specified in a convolution layer, a corresponding number
of feature maps can be generated, and the output of the convolution layer can be obtained
by stacking all the feature maps in the channel dimension.
The role of activation function is to introduce nonlinear factors to enhance the feature
expression ability of the model. At present, the Rectified Linear Unit (ReLU) is the most
versatile and effective activation function (Dubey & Chakraborty, 2021), which can improve
the network sparsity and reduce the network over-fitting. The expression of ReLU is:
(
x if x > 0
ReLU (x) = (2)
0 if x ≤ 0
where Pmax and Pavg are the results of global max pooling and global average pooling of
input features, σ is sigmoid activation function, W0 and W1 are the weights of the first and
second layers of MLP, and their sizes are c/r × c and c × c/r respectively.
As shown in Fig. 4, the SAM uses the data map D0 as input which is the output of CAM,
and gets two feature maps with the size of h × w × 1 through max pooling and average
pooling operation in channel, and takes these two feature maps as input of convolution
Sigmoid
Channel attention
Sigmoid
layer containing a filter. And the output is activated by sigmoid to get spatial attention As .
Finally, D0 is multiplied with As to get the final feature.
The spatial attention can be expressed as:
where fconv is convolution operation, maxpool and avgpool are max pooling and average
pooling respectively.
Ct-1 Ct
╳ ┼
tanh
it
ft ╳ ╳
Ct ot
σ σ tanh σ
ht
ht-1
xt
the updated content of the previous state Ct −1 , and the output gate determines the hidden
state ht that the current cell state passes to the next cell through the output control ot .
The specific calculation process in LSTM network unit is as follows:
Ct = ft Ct −1 + it Ct
e (8)
ht = ot tanh(Ct ) (10)
s
le
c
Spatial attention
cy
Maxpooling
SequenceInput layer
Batch normalization
Channel attention
2D-Convolution
Relu
time
Input data
Batch normalization
RUL
Sigmoid
FCL
LSTM
PROPOSED METHODOLOGY
Our proposed RUL prediction model contains a variety of deep learning layers. Figure 6
shows the network structure of our model. First, input data is converted into the data
map as shown in Fig. 6, and then the CNN module will extract the features of the input
data firstly. In this process, we use three CNN blocks with the same structure. In addition,
we introduce the batch normalization layer into the CNN network, its function is to
independently normalize the small batch data of all observations in each channel, improve
the training speed and effect of CNN and reduce the influence of initialization parameters
on training process. Then, the CBAM module infers attention, and carries out adaptive
weighting processing. Finally, the obtained feature map is input into LSTM network to
mine the hidden relationship between features and service time. Finally, the predicted RUL
value is obtained through a fully connected layer and activated by sigmoid function.
The above is the design process of our model. For the training, verification and
deployment of the model, our process is shown in Fig. 7. After the design of our model
is completed, the training data set and test data set should be prepare after. Training data
is used as input data to train RUL prediction model and get the trained model. After that
the trained model will be verified with the test data, and the performance will be evaluated
by the specified evaluation method. If the trained model can meet the available standards,
we can deploy the model to the aircraft fleet. After each flight, we collect the real-time
monitoring data for the same preprocessing, and input it into the RUL prediction model.
And then according to the prediction results and the set maintenance threshold, we can
is performance
Test flight data Trained model evaluation meet the credibility
requirements ?
Yes
Figure 7 Realization process of aircraft engine RUL prediction with proposed method.
Full-size DOI: 10.7717/peerjcs.1084/fig-7
make a reasonable decision on engine maintenance, and improve the accuracy of aircraft
engine maintenance activities.
EXPERIMENTAL STUDY
Dataset description
The dataset used in our experiment is the latest Aircraft Engine Run-to-Failure data set
published by NASA Ames Research Center in 2020. The dataset was obtained through
C-MPASS platform which could simulate the working process of turbofan aircraft engine,
its structure is shown in Fig. 8. Compared with the previous run-to-failure trajectories,
it considered the performance degradation behavior of the aircraft engine in real flight
conditions, and recorded all the flight environment parameters, operation data and
degradation data (Chao et al., 2021).
In order to compare with other methods, we use DS02, the most widely used data set in
this data set, which contains the run-to-failure simulation data of nine engines, of which
six (unit =2,5,10,16,18,20) are the training set data and three (unit =11, 14, 15) are the
test set data. The recorded data of each engine contains four scenario descriptors (W ),
13 measured physical properties (Xs ), 18 virtual sensors data (Xv ) and 10 model health
parameters (θ ). Table 1 shows the description of monitoring parameters. The units are
divided into three flight classes (Flight class 1, Flight class 2, and Flight class 3) according to
the length of operation time (short-length flights, medium-length flights, and long-length
flights). Figure 9 shows the working conditions of different flight classes of engines in the
flight envelope, and the green area in the figure is the flight envelope of the engine.
Figure 10 shows the kernel density estimations of scenario descriptors (W ) of each
engine. From the distribution, we can see that the working environment of different
engines is different. Especially, compared with other engines, the working altitude of
unit14 is lower. Under this more realistic condition, the RUL of the aircraft engine is
harder to predict.
Experimental process
Data preprocessing
The preprocessing of the monitoring data includes slicing and normalization. Slicing
refers to selecting a part of the recorded data of the aircraft engine as the input data of
RUL prediction. This is because the monitoring data of aircraft engine in one service
cycle contains a large amount of repeated and redundant information. We divide it into
400 copies equally according to the total time in one service cycle, and the data of the
corresponding time is extracted and recombined into input data, which can ensure that the
input data can describe the RUL information of the aircraft engine, reduce the computing
resources and improve the learning speed.
In order to eliminate the training error caused by different measuring units of monitoring
data, we normalize the data monitored by different sensors. The normalization method
adopted in this article is Min-Max normalization, and the calculation formula is:
d − dmin
d=
b (11)
dmax − dmin
dmax , dmin are the maximum and minimum value of the variable, and d, b
d are the
original and normalized values of the variable.
Evaluation metric
In order to quantify the performance of our proposed RUL prediction model, we select the
RMSE and Score to measure the prediction accuracy of the model (Custode et al., 2022).
RMSE is the root mean square error, which is often used as a measure of error in deep
learning models, and can reflect the overall deviation between the predicted RUL and true
#10-4
30
4
20
Density
Density
2
10 Unit 2
Unit 5
Unit 10
0 0 Unit 11
1 2 3 4 0.2 0.4 0.6 0.8
Altitude/ft 4 Mach Number Unit 14
#10
0.15 Unit 15
Unit 16
0.06 Unit 18
0.1 Unit 20
Density
Density
0.04
0.05
0.02
0 0
20 40 60 80 100 400 420 440 460 480 500
Throttle Resolver Angle/% Temperature at fan inlet/°R
where Rpi is the ith engine predicted value of RUL , Rt i is the ith engine true value of RUL,
and n is the number of predicted RUL.
Score function is an evaluation function designed by the data set provider. The lower
the score, the better the prediction effect of the model. Its mathematical expression is:
n i i
X exp(− Rp − Rt ) − 1 if (Ri − Ri ≤ 0)
p t
13
Score = i=1
n . (13)
X Rip − Rit
) − 1 if (Rp − Rt > 0)
i i
exp(
10
i=1
60 60 Predicted
60 True
50 50
RUL/cycles
RUL/cycles
RUL/cycles
40 40
40
30 30
20 20
20
10 10
0 0 0
0 20 40 60 0 20 40 60 80 0 20 40 60 80
Time/cycles Time/cycles Time/cycles
spatial attention in our model. Therefore, we extracted the spatial attention matrix in the
5th, 15th, 25th, 35th, 45th and 55th cycles of unit 11, mapped it into the size of the input
data, and the result is displayed in the form of the hot map in Fig. 12.
In the figure, we are able to distinctly see that the weights of different variables are
obviously different. A larger weight means that the network will focus more on the
changes of the corresponding variables in RUL prediction, while a smaller weight means
the opposite. We found four variables with the largest weights, namely HPT _eff _mod,
HPT _flow_mod, LPT _eff _mod and LPT _flow_mod, and showed the changes of their
Time
Time
Time
0.4 0.4
0.2
0.2
0.2
Variables Variables Variables
cycle=35 cycle=45 cycle=55
0.6
0.8
0.8
0.5
0.6 0.6
Time
Time
Time
0.4
0.4 0.4
0.3
0.2
0.2
0.2
Variables Variables Variables
#10 -3
0 1
-5 0.5
HPT_flow_mod
HPT_eff_mod
-10 0
-15 -0.5
Unit 2
-20 -1 Unit 5
0 2 4 6 8 10 12 0 2 4 6 8 10 12 Unit 10
Time/s #10 5 Time/s #10 5 Unit 16
0 0
Unit 18
-0.005 Unit 20
-0.005
LPT_flow_mod
LPT_eff_mod
-0.01
-0.01
-0.015
-0.015
-0.02
-0.02
0 2 4 6 8 10 12 0 2 4 6 8 10 12
Time/s #10 5 Time/s #10 5
values with the service time in Fig. 13. The results show that in the whole life cycle, the
changes of the values of most variables are strongly correlated with the service time, so
their values also reflect the health of aircraft engines to some extent.
CONCLUSIONS
This article presents an enhanced RUL prediction method of aircraft engine based on
CBAM. It mainly combines CBAM attention mechanism in the CNN-LSTM network. The
Funding
This work was supported by the National Natural Science Foundation of China (No.
61873351). The funders had no role in study design, data collection and analysis, decision
to publish, or preparation of the manuscript.
Grant Disclosures
The following grant information was disclosed by the authors:
The National Natural Science Foundation of China: No. 61873351.
Competing Interests
The authors declare there are no competing interests.
Author Contributions
• Hao Li conceived and designed the experiments, performed the experiments, analyzed
the data, performed the computation work, authored or reviewed drafts of the article,
and approved the final draft.
• Zhuojian Wang conceived and designed the experiments, authored or reviewed drafts
of the article, and approved the final draft.
• Zhe Li performed the experiments, performed the computation work, prepared figures
and/or tables, and approved the final draft.
Supplemental Information
Supplemental information for this article can be found online at [Link]
peerj-cs.1084#supplemental-information.
REFERENCES
Chao MA, Kulkarni CS, Goebel KF, Fink O. 2021. Aircraft engine run-to-failure
dataset under real flight conditions for prognostics and diagnostics. Data 6:5
DOI 10.3390/data6010005.
Chaudhari S, Polatkan G, Ramanath R, Mithal V. 2021. An attentive survey of attention
models. ACM Transactions on Intelligent Systems and Technology (TIST) 12:1–32.
Che C, Wang H, Fu Q, Ni X. 2019. Combining multiple deep learning algorithms for
prognostic and health management of aircraft. Aerospace Science and Technology
94:105423 DOI 10.1016/[Link].2019.105423.
Chui KT, Gupta BB, Vasant P. 2021. A genetic algorithm optimized RNN-LSTM model
for remaining useful life prediction of turbofan engine. Electronics 10(3):285
DOI 10.3390/electronics10030285.
Custode LL, Mo H, Ferigo A, Iacca G. 2022. Evolutionary optimization of spik-
ing neural P systems for remaining useful life prediction. Algorithms 15:98
DOI 10.3390/a15030098.
Dubey SR, Chakraborty S. 2021. Average biased ReLU based CNN descriptor for
improved face retrieval. Multimedia Tools and Applications 80:23181–23206.
Gao Z, Yin X, Zhang B, Chen M, Li B. 2019. A Wiener processbased remaining life
prediction method for light-emitting diode driving power in rail vehicle carriage.
Advances in Mechanical Engineering 11:1–8.
Guo L, Li N, Jia F, Lei Y, Lin J. 2017. A recurrent neural network based health indicator
for remaining useful life prediction of bearings. Neurocomputing 240:98–109
DOI 10.1016/[Link].2017.02.045.
Hu K, Cheng Y, Wu J, Zhu H, Shao X. 2021. Deep bidirectional recurrent neural
networks ensemble for remaining useful life prediction of aircraft engine. IEEE
Transactions on Cybernetics 12:1–13.
Jang B, Kim M, Harerimana G, Kang S-u, Kim JW. 2020. Bi-LSTM model to increase
accuracy in text classification: combining Word2vec CNN and attention mechanism.
Applied Sciences 10(17):5841 DOI 10.3390/app10175841.
Kamat P, Sugandhi R, Kumar S. 2021. Deep learning-based anomaly-onset aware
remaining useful life estimation of bearings. PeerJ Computer Science 7:e795
DOI 10.7717/peerj-cs.795.