Heat Equation Stein Variational Ensemble: Rethinking and Advancing Uncertainty-Aware Soft Sensor Modeling

Yiran Ma,Zhichao Chen,Zeyu Yang,Xinmin Zhang,Zhihuan Song
DOI: https://doi.org/10.1109/tii.2024.3438277
IF: 12.3
2024-01-01
IEEE Transactions on Industrial Informatics
Abstract:Data-driven soft sensors have been prevalent in industrial key performance indicator prediction. However, rigorously quantifying the uncertainty of predictions has been consistently neglected. This neglect can result in industrial practitioners being ignorant of the reliability of the predictions, potentially leading to dangerous misjudgments. To bridge this gap, this study interprets uncertainty quantification (UQ) from a Bayesian perspective, which conceptualizes model uncertainty as the parameter distribution, thus distinguishing it from data uncertainty. Based on this interpretation, a novel UQ method, heat equation stein variational ensemble (HESVE), is proposed. HESVE introduces an efficient nonparametric approach called Stein variational gradient descent (SVGD) to approximate parameter distributions more precisely. A heat equation-based method is also adopted for adaptive hyperparameter tuning of SVGD. In addition, our method incorporates the variable noise estimator to capture heteroscedastic data noise, which contributes to uncertainty decomposition. Experiments demonstrate the superiority of HESVE in multiple aspects compared to other state-of-the-art methods.
What problem does this paper attempt to address?