About cookies on this site Our websites require some cookies to function properly (required). In addition, other cookies may be used with your consent to analyze site usage, improve the user experience and for advertising. For more information, please review your options. By visiting our website, you agree to our processing of information as described in IBM’sprivacy statement. To provide a smooth navigation, your cookie preferences will be shared across the IBM web domains listed here.
Publication
J Transp Eng Part A Syst
Paper
Joint prediction of remaining useful life and failure type of train wheelsets: Multitask learning approach
Abstract
Train wheel failures account for disruptions of train operations and even a large portion of train derailments. Remaining useful life (RUL) of a wheelset measures how soon the next failure will arrive, and the failure type reveals how severe the failure will be. RUL prediction is a regression task, whereas failure type is a classification task. In this paper, the authors propose a multitask learning approach to jointly accomplish these two tasks by using a common input space to achieve more desirable results. A convex optimization formulation is developed to integrate least-squares loss and negative maximum likelihood of logistic regression as well as model the joint sparsity as the L2/L1 norm of the model parameters to couple feature selection across tasks. The experiment results show that the multitask learning method outperforms both the single-task learning method and Random Forest.