TY - JOUR
T1 - Multi-task Pointwise Mutual Information Learning for Bearing Remaining Useful Life Cross-Domain Imbalanced Regression
AU - Hu, Tao
AU - Mo, Zhenling
AU - Zhang, Zijun
N1 - Publisher Copyright:
© 2014 IEEE.
PY - 2025/8/1
Y1 - 2025/8/1
N2 - 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. © 2025 IEEE.
AB - 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. © 2025 IEEE.
KW - Data Imbalance
KW - Deep Learning
KW - Domain Generalization
KW - Industrial Internet of Things (IIoT)
KW - Remaining Useful Life
UR - http://www.scopus.com/inward/record.url?scp=105005073297&partnerID=8YFLogxK
UR - https://www.scopus.com/record/pubmetrics.uri?eid=2-s2.0-105005073297&origin=recordpage
U2 - 10.1109/JIOT.2025.3569977
DO - 10.1109/JIOT.2025.3569977
M3 - RGC 21 - Publication in refereed journal
AN - SCOPUS:105005073297
SN - 2327-4662
VL - 12
SP - 30415
EP - 30425
JO - IEEE Internet of Things Journal
JF - IEEE Internet of Things Journal
IS - 15
ER -