Combining Deep Learning and Survival Analysis for Asset Health Management



Published Nov 13, 2020
Linxia Liao Hyung-il Ahn


We propose a method to integrate feature extraction and prediction as a single optimization task by stacking a three-layer model as a deep learning structure. The first layer of the deep structure is a Long Short Term Memory (LSTM) model which deals with the sequential input data from a group of assets. The output of the LSTM model is followed by meanpooling, and the result is fed to the second layer. The second layer is a neural network layer, which further learns the feature representation. The output of the second layer is connected to a survival model as the third layer for predicting asset health condition. The parameters of the three-layer model are optimized together via stochastic gradient decent. The proposed method was tested on a small dataset collected from a fleet of mining haul trucks. The model resulted in the “individualized” failure probability representation for assessing the health condition of each individual asset, which well separates the in-service and failed trucks. The proposed method was also tested on a large open source hard drive dataset, and it showed promising result.

Abstract 601 | PDF Downloads 496



deep learning, long short term memory, survival analysis, prognostics and health management

Ahn, H.-i., Leung, Y. T., & Hochstein, A. (2014). Building a data-driven vital sign indicator for an economically optimized component replacement policy. In Annual conference of the prognostics and health management (phm) society.
Babu, G. S., Zhao, P., & Li, X.-L. (2016). Deep convolutional neural network based regression approach for estimation of remaining useful life. In International conference on database systems for advanced applications (pp. 214–228).
Benkedjouh, T., Medjaher, K., Zerhouni, N., & Rechak, S. (2015). Health assessment and life prediction of cutting tools based on support vector regression. Journal of Intelligent Manufacturing, 26(2), 213–223.
Carter, C. K., & Kohn, R. (1994). On gibbs sampling for state space models. Biometrika, 81(3), 541–553.
Gers, F. A., Schmidhuber, J., & Cummins, F. (2000). Learning to forget: Continual prediction with lstm. Neural computation, 12(10), 2451–2471.
Hong, S., Zhou, Z., Zio, E., &Wang,W. (2014). An adaptive method for health trend prediction of rotating bearings. Digital Signal Processing, 35, 117–123.
Katzman, J., Shaham, U., Cloninger, A., Bates, J., Jiang, T., & Kluger, Y. (2016). Deep survival: A deep cox proportional hazards network. arXiv preprintarXiv:1606.00931.
Li, H., Pan, D., & Chen, C. P. (2014). Intelligent prognostics for battery health monitoring using the mean entropy and relevance vector machine. Systems, Man, and Cybernetics: Systems, IEEE Transactions on, 44(7), 851–862.
Li, K., & Wang, Q. (2015). Study on signal recognition and diagnosis for spacecraft based on deep learning method. In Prognostics and system health management conference (phm), 2015 (pp. 1–5).
Qiao, L. Q., & Xun, L. J. (2015). State of health estimation combining robust deep feature learning with support vector regression. In Control conference (ccc), 2015 34th chinese (pp. 6207–6212).
Tamilselvan, P., & Wang, P. (2013). Failure diagnosis using deep belief learning based health state classification. Reliability Engineering & System Safety, 115, 124–135.
Trappey, A. J., Trappey, C. V., Ma, L., & Chang, J. C. (2015). Integrating real-time monitoring and asset health prediction for power transformer intelligent maintenance and decision support. In Engineering asset management-systems, professional practices and certification (pp. 533–543). Springer.
Yan, W., & Yu, L. (2015). On accurate and reliable anomaly detection for gas turbine combustors: A deep learning approach. In Proceedings of the annual conference of the prognostics and health management society.
Technical Papers