NAISS
SUPR
NAISS Projects
SUPR
Physics-Informed Graph Neural Networks for Multi-Sensor Fault Diagnosis
Dnr:

NAISS 2026/4-1472

Type:

NAISS Small

Principal Investigator:

Pu Sun

Affiliation:

Kungliga Tekniska högskolan

Start Date:

2026-09-01

End Date:

2027-03-01

Primary Classification:

20307: Production Engineering, Human Work Science and Ergonomics

Allocation

Abstract

This project extends prior work on hybrid fault diagnosis for production machinery, in which physically grounded domain knowledge (order-domain, time-domain, and frequency-domain features) was integrated with a 1D-CNN-BiLSTM network to improve diagnostic robustness under variable operating speeds (published in IOP Conf. Ser.: Mater. Sci. Eng., vol. 1342, article 012065). Building on that published hybrid framework, this project develops a heterogeneous graph neural network (GNN) architecture in which multi-sensor vibration measurement channels and process-context information (e.g., conveyor speed) are represented as nodes in a heterogeneous graph. Attention-based message passing (GATConv, implemented in PyTorch Geometric) is used to learn cross-sensor relationships, with a soft physics-informed bias constraining attention weights toward physically meaningful sensor relationships. Perturbation-based faithfulness validation is used to assess whether the learned attention weights correspond to physically interpretable relationships, extending the interpretability analysis of the earlier hybrid framework. The project is validated using vibration data collected from a pharmaceutical carton labelling machine under multiple operating speeds and fault conditions. Training and evaluation of the graph-based models require GPU resources for iterative architecture development, cross-validation, and hyperparameter search. This work forms part of a PhD thesis on integrated production data systems for fault diagnosis. Main supervisor: Andreas Archenti: archenti@kth.se KTH Department of Production Engineering