This repository contains the implementation of MFGPCox, a unified Bayesian framework for jointly modeling:
- Time-to-event data (failure times)
- Condition-monitoring signals from multiple sensors
- Multiple Failure modes (categorical outcomes)
The model integrates:
- Convolved Multi-output Gaussian Process (CMGP) for modeling sensor signals
- Cox proportional hazards model for survival analysis
- Multinomial distribution for failure mode modeling
within a hierarchical Bayesian framework, enabling accurate prediction and uncertainty quantification.
-
case_study/
Files for the case study, including CMGP hyperparameter optimization code and outputs, ELBO optimization code, prediction code, evaluation code, benchmark comparison files, and associated data/results. -
numerical_study/
Files for the numerical study, including data generation, CMGP hyperparameter optimization code and outputs, ELBO optimization code, prediction code, evaluation code, benchmark comparison files, and associated data/results. -
utils/
Shared Python utility modules used across the case study and numerical study workflows. -
requirements.txt
Python dependency list for the main code environment.
This repository accompanies the following paper:
📌 https://arxiv.org/abs/2506.17036
(Accepted at Technometrics, 2026)
If you use this code or find it helpful, please cite:
@article{aghaee2025bayesian,
title={Bayesian Joint Model of Multi-Sensor and Failure Event Data for Multi-Mode Failure Prediction},
author={Aghaee Dabaghan Fard, Sina and Kim, Minhee and Deep, Akash and Lee, Jaesung},
journal={arXiv e-prints},
pages={arXiv--2506},
year={2025}
}

