Information-based Gradient enhanced Causal Learning Graph Neural Network for fault diagnosis of complex industrial processes

Ruonan Liu,Yunfei Xie,Di Lin,Weidong Zhang,Steven X. Ding
DOI: https://doi.org/10.1016/j.ress.2024.110468
IF: 7.247
2024-09-04
Reliability Engineering & System Safety
Abstract:By representing the embedded components and their interactions in industrial systems as nodes and edges in a graph, Graph Neural Networks (GNNs) have achieved outstanding results due to their ability to model statistical correlations. However, these correlations may not capture the true causal relationships within the data, thereby impairing the model's performance in fault diagnosis. To address this issue, an Information-based Gradient enhanced Causal Learning Graph Neural Network (IGCL-GNN) is proposed for fault diagnosis of complex industrial processes. First, the information theory in graph representations is theoretically analyzed and the optimization objectives are derived separately for the causal and non-causal parts of the graph neural network, which decouple it into a multi-objective optimization problem. Then, to optimize such problem, a causal disentanglement layer is designed in the graph network that could effectively separate causal and non-causal information in graph representations. Thirdly, a novel gradient reactivation method is proposed to dynamically filter features from the disentangled layers, thereby capture the causal representations of graph data more accurately. For robust and efficient optimization, the multi-objective gradient descent algorithm is employed in this paper. Finally, comparative experiments were conducted on the three-phase flow facility (TFF) dataset, achieving a fault diagnosis accuracy of 98.42% for our proposed method.
engineering, industrial,operations research & management science
What problem does this paper attempt to address?