This is a PyTorch implementation of the BHGNN model from the paper [Bayesian Hierarchical Graph Neural Networks with Uncertainty Feedback for Trustworthy Fault Diagnosis of Industrial Processes]
- Python 3.9
- Pytorch 1.12.1+cuda11.3
- torch-geometric 2.1.0
- torch-scatter 2.0.9
- torch-sparse 0.6.15
- torch-geometric >= 1.6.0
- HGP-SL
- what
If you find BHGNN is useful for your research, please cite the following paper:
@ARTICLE{chenBHGNN2023, author={Chen, Dongyue and Xie, Zongxia and Liu, Ruonan and Yu, Wenlong and Hu, Qinghua and Li, Xianling and Ding, Steven X.}, journal={IEEE Transactions on Neural Networks and Learning Systems}, title={Bayesian Hierarchical Graph Neural Networks With Uncertainty Feedback for Trustworthy Fault Diagnosis of Industrial Processes}, year={2023}, volume={}, number={}, pages={1-14}, doi={10.1109/TNNLS.2023.3319468}