Multitask Pointwise Mutual Information Learning for Bearing Remaining Useful Life Cross-Domain Imbalanced Regression
In modern industry, the Industrial Internet of Things (IIoT) has enabled system health analytics and monitoring through continuous data collection and networked connectivity. Bearing remaining useful life (RUL) prediction is one pivotal analytical task in preventing industrial system failures and optimizing maintenance schedules. Existing prediction methods using data face two critical engineering challenges: (1) performance degrades when deployed to unseen operational domains, and (2) imbalanced sensor data distributions cause biased predictions. Two challenges can even co-exist, which further limits the effectiveness of prediction methods using data. This study proposes a multi-task pointwise mutual information learning (MPML) based prediction model development method to tackle bearing RUL prediction under this compound challenge. MPML offers several key innovations through the following developments. First, an auxiliary task-assisted multi-task model learning scheme is devised to obtain task-wise generalizability for learning invariant latent features, and theoretical analysis is provided to explain the invariant feature learning mechanism. Furthermore, pointwise mutual information (PMI) modeling with a statistical explanation is proposed to impose adaptive biases, rectifying penalties for RUL misprediction in minority groups. Consequently, MPML addresses the unseen domain prediction through the multi-task learning scheme and effectively handles the data imbalance with the novel PMI-assisted loss. Extensive computational experiments are conducted to demonstrate the superiority of MPML, achieving a 33.12 square error compared to state-of-the-art methods.
Authors
- Zhenling Mo (ORCID: https://orcid.org/0000-0002-4412-4684)
- Zijun Zhang (ORCID: https://orcid.org/0000-0002-2717-5033)
- Tao Hu (ORCID: https://orcid.org/0000-0002-9007-4577)
Institutions
- City University of Hong Kong (HK)
Publication Details
- Journal
- IEEE Internet of Things Journal
- Published
- 2025-05-14
- DOI
- https://doi.org/10.1109/jiot.2025.3569977
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00