A Multi-Task Joint Learning Model Based on Transformer and Customized Gate Control for Predicting Remaining Useful Life and Health Status of Tools.

ABSTRACT

Previous studies have primarily focused on predicting the remaining useful life (RUL) of tools as an independent process. However, the RUL of a tool is closely related to its wear stage. In light of this, a multi-task joint learning model based on a transformer encoder and customized gate control (TECGC) is proposed for simultaneous prediction of tool RUL and tool wear stages. Specifically, the transformer encoder is employed as the backbone of the TECGC model for extracting shared features from the original data. The customized gate control (CGC) is utilized to extract task-specific features relevant to tool RUL prediction and tool wear stage and shared features. Finally, by integrating these components, the tool RUL and the tool wear stage can be predicted simultaneously by the TECGC model. In addition, a dynamic adaptive multi-task learning loss function is proposed for the model's training to enhance its calculation efficiency. This approach avoids unsatisfactory prediction performance of the model caused by unreasonable selection of trade-off parameters of the loss function. The effectiveness of the TECGC model is evaluated using the PHM2010 dataset. The results demonstrate its capability to accurately predict tool RUL and tool wear stages.