UCB-Based Architecture Search for Remaining Useful Life Prediction

Pengli Mao,Yan Lin,Baochang Zhang,Lin Li
2023-01-01
Abstract:Remaining useful life (RUL) is a significant challenge in prognostics and health management. With the development of data science and deep learning, neural network technology has been applied to this problem. In this study, we present a self-optimized mechanism for neural architecture search. Based on upper confidence bound (UCB) algorithm, we construct a reinforcement learning method for search strategy. UCB explores the combinatorial parameter space of a multi-head convolutional neural network with a recurrent neural network to find a suitable architecture. We conducted experiments on the C-MAPSS dataset, and our proposed model achieved better results in measured by RMSE and Scoring function. Compared with other data-based approaches, our method had better performance in terms of both precision and efficiency.
What problem does this paper attempt to address?