Task-incremental broad learning system for multi-component intelligent fault diagnosis of machinery

摘要

Broad learning system (BLS) is widely used in intelligent fault diagnosis (IFD) since its high computation efficiency and incremental learning ability. However, its applicability is limited to the single-task learning scenario in which only one diagnosis task needs to be learned. Generally, a machinery system contains multiple critical components that need to be diagnosed. The fault data of different components will be collected at different times for model training. It is essentially a task-incremental learning scenario in which the diagnosis model needs to learn a series of diagnosis task for different components at different times. Existing BLS methods cannot meet this requirement due to the catastrophic forgetting issue. Therefore, this paper proposes a task-incremental broad learning system (TiBLS) for multi-component IFD. The TiBLS is developed as a multi-head configuration with a series of BLS blocks to learn different diagnosis tasks sequentially. Then, the catastrophic forgetting is prevented via parameter isolation. Finally, the structure-incremental learning ability is developed for the TiBLS to enhance the diagnosis performance of each task without retraining. In this way, the TiBLS will gain more and more functions to diagnose different components over time. The experiment validation is implemented on a simulated machinery system including three critical components. The diagnosis accuracies of the three tasks are 94.75%, 93.02%, and 92.00%, respectively. The training times of the three tasks are 22.2 s, 34.5 s, and 7.6 s, respectively. The satisfying results demonstrate that the TiBLS is an effective and efficient method for multi-component IFD.