arXiv:2604.24818cs.LG2026-04

用细粒度离散化+可解释规则,让工业设备寿命预测更准更快。

Heterogeneous Variational Inference for Markov Degradation Hazard Models: Discretized Mixture with Interpretable Clusters

论文配图:Heterogeneous Variational Inference for Markov Degradation Hazard Models: Discretized Mixture with Interpretable Clusters
图 1 · 摘自论文原文
  • 8级分段+30维特征融合,强化退化信号并提升模型稳定性
  • 在280台泵上验证,ADVI比NUTS快84倍且无发散问题
  • 提出可解释性筛选规则,防止过拟合,适合工业部署

贝叶斯有限混合模型可识别离散风险集群(低风险与高风险设备),但存在三大瓶颈:(1) 粗粒度状态离散化导致退化信号不足;(2) 当数据支持的簇数少于探索数时,簇识别不稳定;(3) 马尔可夫链蒙特卡洛(MCMC)方法计算不可行,单模型需7小时以上。本文提出实用框架:(1) 采用8状态全局百分位离散化以增强退化事件;(2) 30维特征工程,融合统计趋势(22特征)、连续健康指标及文本嵌入(经PCA压缩至3维);(3) 可解释模型选择规则,结合最小簇占比、分离度与WAIC;(4) 使用全协方差结构的自动微分变分推断(ADVI),实现稳定快速估计。应用于280台工业泵设备、104,703条检测记录,结果表明:(1) 随机效应模型下,ADVI与NUTS估计几乎一致,速度提升15倍,验证了ADVI准确性;(2) 有限混合模型在可解释约束下识别最优簇数;(3) NUTS出现严重收敛问题和标签切换,而ADVI在84倍更短时间内提供稳定结果。贡献包括:(1) 首次证明细粒度状态离散化(8状态)对生存分析中混合模型稳定性至关重要;(2) 提出整合统计、连续与语义信号的综合特征工程策略;(3) 实践性可解释规则有效防止自动化模型选择中的过拟合;(4) 实证表明,ADVI在收敛性、稳定性与计算效率上优于NUTS。

原文摘要 · Abstract (English)

Bayesian finite mixture models can identify discrete risk clusters (low-risk vs. high-risk equipment), but face three critical bottlenecks: (1) insufficient degradation signals from coarse state discretization, (2) unstable cluster identification when data inherently supports fewer clusters than explored, and (3) computational infeasibility of Markov Chain Monte Carlo (MCMC) methods for production deployment (7+ hours per model). We propose a practical framework combining (1) 8-state global percentile discretization that amplifies degradation events, (2) 30-dimensional feature engineering integrating statistical trends (22 features), continuous health indicators, and text embeddings (PCA-compressed to 3 dimensions), (3) interpretable model selection rules enforcing minimum cluster share and separation alongside WAIC, and (4) Automatic Differentiation Variational Inference (ADVI) with full-rank covariance for stable, fast estimation. Applied to 280 industrial pump equipment with 104,703 inspection records, we demonstrate: (1) Random effect models (baseline) show ADVI and NUTS produce nearly identical estimates with 15$\times$ speedup, validating ADVI accuracy. (2) Finite mixture models identify optimal number of clusters with interpretability constraints. (3) NUTS exhibits severe convergence issues and label switching, while ADVI provides stable results in 84$\times$ less time. We contributed that (1) First demonstration that fine-grained state discretization (8-state) is essential for mixture model stability in survival analysis.(2) Comprehensive feature engineering strategy combining statistical, continuous, and semantic signals. (3) Practical interpretability rules preventing overfitting in automated model selection. (4) Empirical evidence that ADVI outperforms NUTS for finite mixture models in terms of convergence, stability, and computational efficiency.

生存分析变分推断工业预测可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。