arXiv:2412.18980cs.LG2024-12被引 5

对比多种不确定性模型,发现集成模型在机械故障诊断中更可靠且高效。

Evaluating deep learning models for fault diagnosis of a rotating machinery with epistemic and aleatoric uncertainty

  • 采用深度集成、贝叶斯网络等方法捕捉认知与随机不确定性。
  • 在未知故障和噪声场景下,集成模型对异常数据检测率超80%。
  • 适合部署于实际工业系统的故障诊断系统,尤其看重稳定性与速度。

不确定性感知的深度学习模型近年来在故障诊断中受到关注,因其能在分布外(OOD)数据出现时提升故障检测可靠性,应对未见故障(认知不确定性)或噪声干扰(随机不确定性)。本文首次对旋转机械故障诊断中前沿的不确定性感知深度学习架构进行了全面比较,考察了不同认知不确定性场景及多种随机不确定性类型的影响。所选模型包括基于丢弃采样的方法、贝叶斯神经网络和深度集成。此外,为区分分布内与分布外数据,本文提出一种新阈值并结合已有方法进行判别。实证结果表明:在认知不确定性存在时,所有模型平均可有效检测大部分分布外数据;而深度集成模型在各类阈值下表现最优。在随机不确定性下,低噪声水平会降低检测能力,但深度集成模型性能下降最轻微,仍优于其他模型。结合其推理时间更短的优势,深度集成成为首选方案。

原文摘要 · Abstract (English)

Uncertainty-aware deep learning (DL) models recently gained attention in fault diagnosis as a way to promote the reliable detection of faults when out-of-distribution (OOD) data arise from unseen faults (epistemic uncertainty) or the presence of noise (aleatoric uncertainty). In this paper, we present the first comprehensive comparative study of state-of-the-art uncertainty-aware DL architectures for fault diagnosis in rotating machinery, where different scenarios affected by epistemic uncertainty and different types of aleatoric uncertainty are investigated. The selected architectures include sampling by dropout, Bayesian neural networks, and deep ensembles. Moreover, to distinguish between in-distribution and OOD data in the different scenarios two uncertainty thresholds, one of which is introduced in this paper, are alternatively applied. Our empirical findings offer guidance to practitioners and researchers who have to deploy real-world uncertainty-aware fault diagnosis systems. In particular, they reveal that, in the presence of epistemic uncertainty, all DL models are capable of effectively detecting, on average, a substantial portion of OOD data across all the scenarios. However, deep ensemble models show superior performance, independently of the uncertainty threshold used for discrimination. In the presence of aleatoric uncertainty, the noise level plays an important role. Specifically, low noise levels hinder the models' ability to effectively detect OOD data. Even in this case, however, deep ensemble models exhibit a milder degradation in performance, dominating the others. These achievements, combined with their shorter inference time, make deep ensemble architectures the preferred choice.

故障诊断不确定性建模深度集成旋转机械

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