arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2101.03497cs.LGmath.OC
收藏

训练轮对剩余使用寿命与故障类型联合预测:一种多任务学习方法

Joint Prediction of Remaining Useful Life and Failure Type of Train Wheelsets: A Multi-task Learning Approach

University at Buffalo, The State University of New York(纽约州立大学布法罗分校);IBM T J Watson Research Center(IBM T J 沃森研究中心)

Weixin Wang

摘要:

训练车轮的故障导致训练运行中断,甚至导致大部分训练脱轨。轮对的剩余使用寿命(RUL)衡量下一次故障的发生时间,而故障类型则揭示了故障的严重程度。RUL预测是一个回归任务,而故障类型是一个分类任务。在本文中,我们提出了一种多任务学习方法,通过使用一个共同的输入空间来共同完成这两个任务,以达到更理想的结果。我们开发了一个凸优化公式,以整合最小二乘损失和负最大似然逻辑回归,并建模的联合稀疏的L2/L1范数的模型参数耦合跨任务的特征选择。实验结果表明,我们的方法比单任务学习方法的预测精度提高了3%。

英文摘要:

The failures of train wheels account for disruptions of train operations and even a large portion of train derailments. Remaining useful life (RUL) of a wheelset measures the 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, we propose a multi-task learning approach to jointly accomplish these two tasks by using a common input space to achieve more desirable results. We develop a convex optimization formulation to integrate both least square loss and the negative maximum likelihood of logistic regression, and model the joint sparsity as the L2/L1 norm of the model parameters to couple feature selection across tasks. The experiment results show that our method outperforms the single task learning method by 3% in prediction accuracy.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑