首页/文章/ 详情

IEEE || ABIGX:一个可扩展的故障检测和分类的统一框架(下)

1月前浏览578

从对抗攻击视角重构故障重建,东南大学团队提出ABIGX统一框架,将经典CP/RBC诊断方法拓展至任意深度学习模型,理论证明其等价性,破解"故障类别涂抹"难题,在TEP、晶圆图及工业图像上验证,为智能运维黑箱决策提供可信归因工具。      

本期给大家推荐这篇东南大学虞文武教授团队的论文。本文提出ABIGX(Adversarial Fault Reconstruction-Based Integrated Gradient eXplanation,基于对抗故障重构的积分梯度解释)框架,统一解决可解释故障检测与分类问题。核心创新在于:1)从对抗攻击视角重构故障重建,将传统CP(Contribution Plot,贡献图)/RBC(Reconstruction-Based Contribution,基于重构的贡献)方法扩展到非线性模型;2)提出分类SPE(Squared Prediction Error,平方预测误差)指标,使AFR(Adversarial Fault Reconstruction,对抗故障重构)适用于故障分类任务;3)结合集成梯度实现模型无关的变量归因。理论证明CP和RBC是ABIGX的线性特例,并分析"故障类别涂抹"问题,证明ABIGX优于传统梯度方法。实验涵盖多种模型,在TEP过程、晶圆图和MVTec-AD数据集上验证其优越的解释准确性与泛化能力。      
由于文章篇幅过长,小编将分两次为大家翻译介绍这篇论文,第一篇推文(点击即达)系统系统综述了故障检测与分类的研究背景,指出传统CP、RBC方法受限于线性模型且无法处理分类任务,而现有深度学习解释器缺乏故障诊断领域的理论基础。基于此,作者从对抗攻击视角重新诠释故障重建,提出AFR方法及分类SPE指标,进而构建ABIGX统一框架——通过AFR生成语义有效的反事实基线,结合集成梯度实现任意可微FDC模型的变量归因,为后续理论分析与实验验证奠定基础。      
本篇推文进一步涵盖理论证明与实验验证。理论层面,作者提出"故障类别涂抹"问题,证明ABIGX比显著性图和IG更能缓解该问题;同时证明CP和RBC是ABIGX在线性PCA检测器下的特例,实现经典方法与现代框架的统一。实验层面,在TEP化工过程、晶圆图及MVTec-AD工业图像上,ABIGX对PCA、AE、MLP、CNN、ViT等模型均展现出更优的解释准确性与一致性,验证了框架的泛化能力与实用价值。      

文章质量很高。本文配套思维导图&论文精华速览,已同步至小红书:(ID:故障诊断与python学习),需要的小伙伴可以自取💡~      

论文链接:通过点击最左下角的阅读原文进行在线阅读及下载。  

论文基本信息

论文题目: ABIGX: A Unified Framework for eXplainable Fault Detection and Classification

论文期刊:IEEE Transactions on Pattern Analysis and Machine Intelligence

Doi:https://doi.org/10.1109/TPΑΜΙ.2026.3689327

作者:  Yue Zhuo a , Duxin Chen a , Wenwu Yu a , Zhiqiang Ge a , Xiaoyu Jiang b , Jinchuan Qian c , Junhua Zheng c , Zhihuan Song d  
论文时间: 2026年1月  
机构:   

a  School of Mathematics, Southeast University, Nanjing 210096, China

b Hangzhou International Innovation Institute, Beihang University, Hangzhou 311115, China

c School of Automation and Electrical Engineering, Zhejiang University of Science and Technology, Hangzhou 310023, China

d School of Automation, Guangdong University  of Petrochemical Technology, Maoming, 525000, Guangdong, China

作者: 虞文武,东南大学首席教授、数学学院院长,入选教育部长江学者、青年长江、国家"万人计划"青年拔尖人才、国家优青,2014-2024连续十一次入选全球高被引科学家。主要从事系统科学与人工智能交叉研究,发表IEEE汇刊文章100余篇,Google和SCI引用超3万次,H指数75。主持科技部"新一代人工智能"重大项目等多项国家级课题,获国家自然科学二等奖1项、省部级一等奖3项;任Complex Engineering Systems主编及多个IEEE汇刊编委。(来源:Research)

摘要

本文提出了ABIGX,一种用于可解释故障检测与分类的统一框架。ABIGX建立在已有故障诊断方法的基础原理之上,包括贡献图和基于重构的贡献,同时将其适用范围扩展至通用FDC模型,并提升了故障解释效果。ABIGX的核心在于对抗故障重构方法,该方法从对抗攻击的视角重新审视故障重构,引入了一种适用于故障检测与分类任务的新型故障指标。在故障检测方面,我们通过理论证明CP和RBC是ABIGX的线性特例,从而建立了ABIGX与传统故障诊断方法之间的理论联系。对于故障分类,我们解决了故障类别模糊化问题——这是可能掩盖准确解释的固有缺陷。我们证明了ABIGX能够有效缓解该问题,其性能优于现有的基于梯度的解释方法。实验通过定量指标和直观图示对FDC(Fault Detection and Classification,故障检测与分类)的解释效果进行了评估。结果验证了AFR的通用性与准确性,并表明ABIGX在多种FDC模型上提供了更为全面和精确的解释,相较于现有方法具有显著提升。实现代码已公开发布于:https://github.com/qianjc3391/ABIGX。
关键词:特征归因;综合梯度;可解释的人工智能;反事实解释

目录

1 引言

2 预备知识及相关工作

   2.1 FDC

   2.2 故障诊断

   2.3 基于梯度的XAI方法

   2.4 对抗攻击

   2.5 可解释FDC的挑战

3 故障重建:对抗视角

   3.1 对抗故障重建

   3.2 故障分类的SPE指标

   3.3 AFR算法

   3.4 AFR与对抗攻击的对比

4 ABIGX:统一框架

   4.1 定义

   4.2 算法

   4.3 讨论

5 ABIGX在故障分类中的应用

 

   5.1 故障分类设定

   5.2 故障类别涂抹

6 ABIGX在故障诊断中的应用

   6.1 RBC与ABIGX-OneVar

   6.2 CP与ABIGX

7 实验

   7.1 评估指标与对比方法

   7.2 田纳西伊斯曼过程

   7.3 晶圆图

   7.4 MVTec-AD工业图像

   7.5 计算成本   

   7.6 实现细节

8 结论



注:小编能力有限,如有翻译不恰之处,请多多指正~
     若想进一步拜读完整版,请下载原论文进行细读。

5 AIBGX在故障分类中的应用

本节介绍故障类别模糊化问题——这是故障分类模型中固有的问题,会导致误导性的变量贡献,并使准确识别故障真正原因变得困难。针对该问题,我们对故障分类中变量贡献的正确性进行理论分析。我们表明,显著性图直接受到故障类别模糊化的影响,而IG(Integrated Gradients,积分梯度)提供了一定程度的缓解。然而,我们的分析证明ABIGX对该问题的缓解效果最佳,能够提供最准确的解释。这些发现为ABIGX在故障分类模型上的卓越可解释性奠定了坚实的理论基础。

     

5.1 故障分类设定

   

首先,我们构建一个故障分类数据集和模型的玩具示例。不失一般性,假设不同故障类型发生在不同变量上:

示例 1(玩具故障分类数据集)。假设所有正常变量独立同分布地采样自高斯分布          ,所有故障具有相同的幅度f,并且受损变量服从分布          。为简单起见,我们假设故障类型y在方向          上有一个受损变量。那么具有M个变量和N种故障类型的故障分类数据集定义为 (N<M):

其次,我们为该玩具示例构建线性分类模型,以实现所有类别的最佳分离: 

定义6.(线性分类器)设        为线性分类器的权重,其行向量        (       对应于各故障(或正常)类别的线性决策边界。基于Fisher线性判别分析[35],最优行向量通过最大化类间均值距离并最小化类内方差进行训练:

            

其中        为故障类别y的均值向量,      为类别均值的均值,      为各类协方差之和。 

基于示例1中的数据集和定义6中的线性分类器,我们可以证明最优模型权重:

定理2. 设        为故障类别y中变量i的最优模型权重。对于正常类别y = 0,最优       为:

对于故障类别y,最优        为:

证明. 证明详见附录A.1.1。

     

5.2 故障类别涂抹

   

在可解释人工智能领域中,XFDC(eXplainable Fault Detection and Classification,可解释故障检测与分类)旨在帮助人们理解导致故障的潜在因素。给定一个故障样本,XFDC指示各变量对监测结果(检测分数或分类置信度)的贡献程度。贡献度较高的变量被视为异常的潜在根本原因(如故障传感器)。通过识别这些变量贡献,XFDC能够辅助用户解释模型预测的故障,并主动修复存在缺陷的部件。

5.2.1 模型权重 

本节正式介绍故障类别模糊化问题——这是故障分类模型中固有的挑战,并分析其对基于梯度的解释方法的影响。故障类别模糊化产生于分类器的优化目标导致无关变量上出现非零权重,进而扭曲显著性图和IG(Integrated Gradients,积分梯度)等解释器的归因分数。

定义 7.(模型权重中的故障类涂抹)给定故障类型为y的受释故障样本x(如示例 1 中所定义),以及具有最优权重          的故障分类器,变量贡献在模型权重中被涂抹到其他不相关的变量上:

其中         。这些在不相关的故障变量上的非零权重会导致基于梯度的解释器高估它们的贡献,从而产生误导性的归因。

为了定量评估故障类涂抹的程度,我们定义了故障类涂抹 (FCS) 指标,该指标测量了错误分配给不相关变量的总归因比例:

直观地说,故障类涂抹的发生是因为分类器不仅必须将故障样本与正常样本区分开来,还必须区分不同的故障类型。这导致了跨多个类的共享权重,其中与其他故障类型相关联的变量仍然获得非零权重(例如,        ),尽管它们与当前的故障类y无关。另一方面,不对应任何故障类的变量(即,如公式5所示的其余M-N个变量)通常具有零权重,并且不会引发涂抹。

这种现象对模型解释来说是个问题:当解释方法直接依赖于模型梯度或权重(例如,显著图或IG)时,变量贡献可能反映的是类可分性偏差,而不是与故障相关的根本变量。在接下来的小节中,我们将从理论上分析这种涂抹如何影响不同的解释方法,以及ABIGX如何减轻它。

5.2.2 显著图 

显著图简单地计算模型输出关于输入变量的梯度的绝对值。线性模型的梯度就是它的权重,因此显著图贡献直接受到了故障类涂抹问题的影响。

定理 3.(显著图中的故障类涂抹)给定所解释的故障类型y及其真实变量 y,显著图的贡献被涂抹到其他不相关的变量上:

显著图中 FCS 的程度为:

5.2.3 IG 

这些在不相关变量上的涂抹权重在故障分类模型中是不可避免的。一个好的解释方法应该有效地减轻涂抹权重对变量贡献的影响。接下来,我们讨论IG方法中的故障类涂抹。

定理 4.(IG中的故障类涂抹)给定所解释的故障类型y及其真实变量y,以及正常样本期望的基线        ,IG贡献的期望为:

IG中故障类涂抹 (FCS) 的程度为:

        证明. 证明详见附录 A.1.2。      

5.2.4 ABIGX 

随后,我们分析了ABIGX中的故障类涂抹,表明ABIGX以更低的FCS程度优于IG。

定理 5.(ABIGX 中的故障类涂抹)ABIGX中的故障类涂抹程度低于IG:


        证明. 证明详见附录 A.1.3。      

5.2.5 与故障涂抹的比较 

先前在CP和RBC研究中讨论的故障涂抹 [7], [36] 是由所有输入共享的相同权重引起的,因为检测模型只输出一个标量(例如,SPE故障指数)。尽管每个故障类的模型权重都是唯一的,但我们表明分类器仍然会对不相关的变量分配权重(由于区分不同故障类型的任务)。这些非零的不相关权重涂抹了变量的贡献。总而言之,故障类涂抹论证了故障分类中故障涂抹问题的现象和原因。

6 预备知识及相关工作

本节介绍了ABIGX如何统一地重新构架两种基本的故障诊断方法CP和RBC。对于基于PCA的故障检测,我们指出CP和RBC都是ABIGX的特例。图 5 直观地展示了通过两个特定的经AFR重构的样本             和            ,CP和RBC与ABIGX的等价性。

在下文中,基于 PCA 的检测器用其残差子空间投影矩阵表示:            

6.1 RBC与ABIGX-OneVar

第4节讨论了ABIGX源于RBC的思想。此外,我们证明了RBC是具有AFR-OneVar的ABIGX的线性特例。

引理 1. (故障重构性质) 使得      最小化的故障重构中的      值为:

     

定理 6. (RBC与ABIGX的等价性)RBC等价于使用单变量AFR的ABIGX (即ABIGX-OneVar):

     

证明. 证明过程见附录A.2.1。

         

6.2 故障诊断CP与ABIGX

故障检测模型的解释——通常称为故障诊断——已得到广泛研究[7]、[6]、[9]、[23]。一旦检测到故障,故障诊断旨在通过量化各输入变量对故障指标(如SPE)的贡献程度,识别其根源变量。下文简要回顾三种常用于基于 PCA 模型的经典故障诊断方法。

     

我们证明CP也是ABIGX在      范数AFR下的线性特例。首先,我们计算PCA在      范数距离约束下的AFR重构样本:

引理2.(PCA的AFR)在      范数距离约束下,基于PCA的故障检测的AFR重构样本      为:

     

这意味着重构向量恰好是残差子空间投影。 证明.证明见附录A.2.2。 随后,我们可以证明CP与ABIGX的等同性。

定理7.(CP与ABIGX的等同性)CP与      范数AFR下的ABIGX等同:

     

证明. 根据引理2中的      ,ABIGX为(省略      的下标):

由式(35),我们有      ,因此得到ABIGX贡献:

     

这与      等同。

此外,受引理2的启发,我们讨论了基于梯度的攻击的可行性,这也是AFR和ABIGX的默认算法:

注 1. (针对PCA的基于梯度的攻击) 基于梯度的攻击总是沿着残差子空间投影的方向重构故障。

证明. PGD每一步的对抗扰动为:

     

其中L是一个标量。 这与直觉一致,因为残差投影指向SPE下降最快的方向。

7 实验

       

7.1 评估指标与对比方法

我们引入两类定量指标,用于对解释方法进行公平评估:

正确性[37]衡量变量贡献与根源变量之间的差异。更优的解释应更接近真实值。具体而言,Correctness-AUC将贡献视为二分类预测分数,通过改变贡献分数的负类阈值,计算受试者工作特征曲线下的面积[38]。正确性也可通过累加真实值上的归因分数来度量,称为Correctness-SUM。

一致性[39]检验贡献是否聚焦于模型真正关注的区域。更优的解释方法应与模型行为具有更高的一致性。Consistency-ADD指标逐步将解释故障样本的变量值添加至正常样本。通过滑动贡献阈值,贡献最大的变量首先被添加,贡献最小的最后被添加。更优的解释方法应使模型预测分数上升更快,这可通过阈值

相关的预测分数曲线下面积进行量化。相反,Consistency-DEL首先删除最重要的变量,直至所有特征均被正常样本替换。类似地,更优的方法应使模型预测分数下降更快,这可通过阈值相关的AUC进行度量。

对于基线方法,我们引入了几种先进的基于梯度的XAI方法,包括:

• 显著性图:输入变量梯度的绝对值,最常用的特征归因方法。

• DeepLift[40]:一种梯度反向传播解释方法,能够妥善处理IG(Integrated Gradients,积分梯度)可能产生误导结果的情形。

• 原始IG[16]:提出的原始IG方法,使用全零向量作为基线。

• 期望IG[41]:IG的一种变体,使用来自数据分布的信息性基线(即本实验中的正常样本),通过对IG归因求期望来缓解基线选择的影响。

• 重要方向梯度积分[42]:IG的一种变体,仅直接关注与梯度对齐的重要方向,在本实验中基于原始IG构建。

• 引导IG[43]:IG的一种变体,基于梯度构建积分路径,使用全零向量作为基线。

• AFR+ 引导IG:我们使用AFR重构样本作为基线,但将ABIGX的默认直线路径积分替换为引导IG的路径构建算法,作为消融设置以验证ABIGX中AFR基线的有效性。

为确保公平比较,我们将所有基于IG的方法(原始IG、期望IG、引导IG和IDGI)的主要超参数——步长,设置为与ABIGX相同。所有实验在INTEL CORE i7-12700K CPU上运行TEP数据集的PCA、AE和MLP模型,在单块NVIDIA GeForce RTX 3080 GPU上运行Wafermap数据集的CNN模型和MVTec-AD数据集的ViT模型。

7.2 田纳西伊斯曼过程

7.2.1 数据集描述

为使TEP设置自成体系,并阐明各故障的物理含义,我们在附录中提供了TE过程流程图及文献[18]中的官方变量和故障描述(图B.1、表B.1)。这些材料明确了故障注入位置(物流、单元)、涉及的执行器和传感器,以及预期的一阶响应,对于解释诊断性能和特征级归因至关重要。

我们的解释正确性指标依赖于为每种故障精心筛选的根源变量 集 合,该集 合源自经典TE过程文献并汇总于附录表B.2。除列出变量索引外,该表还提供了逐故障的推理依据,基于(i)注入位置、(ii)最直接受影响的测量变量和执行器、以及(iii)过程结构和控制回路,将各故障与其根源变量相关联。我们在报告基于根源变量的正确性分数时,将其作为真实值参考。这一故障-变量关联亦与代表性TEP诊断工作[9]、[44]、[45]中采用的故障语义相一致,从而使我们的故障-变量映射成为与文献对齐的、透明的经典TEP故障定义表征,用于解释评估。

           

图6  为NN分类器对TEP故障6的解释对比,图中实根变量为红色。故障6由根变量XMEAS ( 1 )和XMV ( 3 )引起,故障14由XMEAS ( 9 ),XMV ( 10 )和XMEAS ( 21 )引起,详见表B.2 

为清晰起见,图6直观展示了两种TEP故障类型(故障6和故障14)的解释对比。该图分别展示了故障检测和分类模型的传感器变量贡献,其中每个图的四个子图展示了四种不同技术解释的变量贡献。对于TEP数据集,我们采用了文献[46]中的绘图风格。水平轴表示变量贡献值(含方向)。变量按重要性在垂直轴上排序,其中展示了贡献最大的四个变量。值得注意的是,图中还通过彩色分布展示了变量(特征)值与其贡献之间的相关性。

7.2.2 可解释故障检测

               

故障诊断与ABIGX的等价性:第6节从理论上证明了两种故障诊断方法与ABIGX之间的等价性。在TEP故障的PCA检测器上,我们计算了它们之间的经验误差,即所有变量贡献的均值。表2报告了结果,表明差异足够小(尤其在使用MILP求解AFR重构样本时)。

解释结果:首先,图6a绘制了TEP数据集上AE模型不同解释方法的直观对比,表明ABIGX将最高贡献分配给了根源变量。同时,其他方法未能提供完整的解释。

表3报告了TEP上AE检测解释的四项评估指标,涉及14种故障类型。总体而言,ABIGX方法优于其他解释方法,包括故障诊断方法(CP和LLBBC [47])以及基于IG的深度学习解释器(DeepLift不适用于故障检测)。通过比较期望IG与原始IG和IDGI,可以发现全零基线对故障检测解释产生了不利影响。ABIGX-OneVar提供了最接近真实值根源变量的解释。独立归因变量的显著性图在正确性指标上也表现良好。另一方面,ABIGX提供了最匹配模型行为的解释。我们认为,考虑变量交互的解释更接近模型实际处理样本的方式。

基于消融研究(AFR+引导IG),在TEP数据集上使用AFR重构基线结合引导IG的自适应路径,相较于原始引导IG均有持续提升,同时相对于ABIGX具有竞争力的性能表现。

7.2.3 可解释故障分类

               

与故障检测类似,图6b绘制了基于NN的TEP故障6分类器解释。ABIGX的优势体现在变量XMV(3)上。ABIGX为该变量分配了更高的贡献,而显著性图甚至忽略了该变量,其原因已在第4.3.2节中分析。

表4报告了故障分类的解释指标,各方法性能与检测任务类似。ABIGX总体上仍表现更优,即使在正确性指标上(AUC除外)。ABIGX-OneVar仍然强调正确性,但模型一致性较差,这与显著性图类似。DeepLift与原始IG性能相近。尽管IDGI和引导IG相较于原始IG方法表现出性能提升,但由于其基线缺乏信息量,即使在期望IG设置下,在故障分类任务中仍不具竞争力。

分类SPE指标的验证:通过ABIGX-AdvAFR进行消融研究。与第3.2节提出的ABIGX分类SPE不同,ABIGX-AdvAFR采用原始对抗攻击目标(即最小化正常类别的Softmax输出置信度)进行故障重构。ABIGX总体上优于ABIGX-AdvAFR,验证了本文提出的新型分类SPE指标在故障重构中的有效性。

                 

7.2.4 故障级解释

为清晰展示TEP数据集上的归因对比,附录图B.2和B.3以及表B.4和B.3呈现了故障检测和分类模型在所有TEP故障类型上的详细实验结果。这些图可视化了ABIGX在数据分布中对排名靠前变量的归因,而表格则通过报告故障级别的根源变量排名来比较各方法的归因性能。具体而言,根源变量出现在归因前三位中的比例及其平均排名表明,ABIGX在大多数故障类型上与真实值故障变量的一致性最高。

为探究模型精度与ABIGX解释质量之间的相关性,图B.4详细分析了ABIGX在所有TEP故障类型上的归因性能,并按不同FDC模型的正确预测和错误预测样本分别展示。结果凸显了两个关键发现:(1)预测精度较高的故障类型往往表现出更准确的ABIGX归因;(2)在每个故障类型内部,正确预测样本的归因正确性通常高于误分类样本。这些发现表明,ABIGX解释与模型置信度密切相关——当模型成功检测故障时,其能够忠实地归因有意义的贡献;而当模型失效时,则揭示出归因的不确定性。

               

7.3 晶圆图  

7.3.1 数据级描述

晶圆图分析在半导体日常制造运营中至关重要。晶圆图提供了视觉细节,对于识别晶圆故障模式发生的制造阶段具有关键作用。我们研究了晶圆图故障模式分类的解释,其目标是确定晶圆图中哪些像素是故障根源。

我们在WM-811K数据集[48]上训练了一个基于CNN(Convolutional Neural Network,卷积神经网络)的故障分类器,选取了四种故障类型(中心型、边缘局部型、边缘环型和划痕型)。故障和正常晶圆图如图7所示,其中真实值根源像素由红色圆圈标注。绿色像素表示正常芯片,黄色像素表示缺陷芯片。当不存在特定缺陷芯片模式时,晶圆图为正常(如图7左上角所示)。


                   
                       

图7  CNN分类器对四种具有不同故障模式的采样晶圆图的解释,其中真实根像素以红色为圆心

7.3.2 可解释故障分类

图7和表5中呈现的实验结果提供了多种解释方法在用于识别晶圆图故障模式的CNN分类器上的对比分析。ABIGX在所有故障类型中均表现出高度聚焦且清晰的解释效果,与标注的真实值区域高度吻合。这种清晰性表明ABIGX能够有效识别与模型分类决策最相关的区域。其他方法如显著性图、原始IG和IDGI则产生更为分散且聚焦性较差的解释。可以发现,原始IG和IDGI中使用的黑色基线导致了较差的解释效果。DeepLift和期望IG表现中等,其解释清晰度优于显著性图和原始IG,但缺乏ABIGX所具有的精确性。

       

表5  解释方法的评价指标(在晶圆图CNN分类器上)

   
   

表5量化了各方法在四项关键指标上的性能。ABIGX在正确性(AUC(Area Under Curve,曲线下面积):0.839)和一致性(ADD:0.859,DEL:0.040)上均取得了最高分数,表明其在准确识别关键区域的同时,生成的解释具有高度一致性。相比之下,显著性图和原始IG总体表现最低。DeepLift和期望IG在正确性方面表现良好,但在一致性上逊于ABIGX。引导IG(Guided IG,引导积分梯度)在正确性和一致性之间取得了平衡,其分数具有竞争力但未能超越ABIGX。使用AFR(Adversarial Fault Reconstruction,对抗故障重构)重构基线结合引导IG的自适应路径(AFR+引导IG)始终优于原始引导IG,且与ABIGX具有竞争力,证实了AFR基线改进在不同路径选择下均持续有效。

晶圆图故障解释实验验证了ABIGX的泛化能力,其不仅适用于表格型传感器数据,还适用于CNN模型训练的图像等非结构化数据。

       

7.3.3 ABIGX详细解释

       
     

附录图B.5展示了ABIGX在七种故障类型更多晶圆图上的解释结果。每行展示每类故障的四个代表性样本,左侧为原始晶圆图,右侧为ABIGX归因热图。每对图像上方的Consistency-ADD(一致性-添加)分数量化了跨样本解释的一致性。ABIGX对于具有明确空间模式的故障(如"中心型"、"边缘环型"和"随机型")实现了高一致性,表明其在突出与视觉故障特征对齐的模型相关区域方面的有效性。

然而,ABIGX在处理更复杂或模糊的故障时表现出局限性。例如,"近满型"始终产生较低的Consistency-ADD分数(0.375),表明模型仅当晶圆大部分区域受影响时才能对该故障类型进行高置信度预测。"划痕型"、"局部型"和"边缘局部型"等故障类型也表现出不一致的解释质量,凸显了ABIGX在处理混合或不规则模式方面的挑战。这些观察指出了未来改进的必要性,以提高实际质量检测任务中的鲁棒性和泛化能力。

7.4 MVTec-AD工业图像    

7.4.1 数据集描述

MVTec-AD(MVTec Anomaly Detection,MVTec异常检测)数据集[49]是工业视觉检测中异常检测的广泛使用的基准数据集。该数据集包含15类物体和纹理类别的高分辨率图像,每类均含有正常样本和缺陷样本,缺陷类型涵盖划痕、凹痕和颜色偏差等多种表面异常。与传统故障检测中使用的表格型数据集(如TEP)不同,MVTec-AD专注于细粒度空间缺陷,使其适用于评估基于视觉的异常检测模型。其多样化的缺陷类型和真实的成像条件为可解释性带来了独特挑战,为评估ABIGX的泛化能力提供了有价值的测试平台。

       

7.4.2 模型描述

在我们的工作中,我们采用WinClip [33]作为代表性的基于ViT的异常检测器,以评估ABIGX的泛化能力。WinClip是一种最先进的视觉Transformer模型,专为工业图像中的零样本异常检测而设计。通过利用CLIP [50]嵌入和多模态对齐,其在MVTec-AD数据集的多样化类别上取得了优异性能。我们使用预训练的ViT骨干网络,并对WinClip实现进行少量修改以确保可微性,从而实现与基于梯度的解释方法的兼容性。该实验设置展示了ABIGX对更复杂、大规模和现代基于Transformer模型的适用性。

             

7.4.3 可解释异常检测

图8比较了不同方法在来自五个类别(地毯、木材、皮革、瓷砖和网格)的输入工业图像上的ViT归因图。与其他基线方法相比,ABIGX提供了显著更聚焦且更准确的归因,其一致性地突出了真实异常区域(如红色圆圈所示),同时抑制了无关噪声。相比之下,注意力图和原始IG等替代方法表现出分散或定位性较差的归因,往往延伸至真实值故障区域之外。

                   
                     
                         

图8  WinClip异常检测器在MVTec - AD数据集的5个样本上的解释结果。在红色(放大以查看小区域内更精细的细节)中圈出了真值异常区

定量而言,表5验证了ABIGX的优越解释性能。在正确性方面,ABIGX取得了最高的AUC和SUM(总和),表明其显著性区域与其他方法相比与实际故障区域的对齐程度更高。值得注意的是,这证明了ABIGX的归因图不仅具有视觉可解释性,而且在定量意义上也具有意义。相较于期望IG和显著性图的性能提升,进一步验证了ABIGX自适应基线策略的优势,尤其在传统IG方法难以处理任意基线的复杂高维图像域中。

在一致性方面,ABIGX在各项评估指标上表现出强鲁棒性。值得注意的是,WinClip的预测对微小扰动高度敏感,仅添加少量像素即可显著增加异常分数,导致所有方法的ADD(一致性-添加)分数均较高。然而,ABIGX取得了相对较低的DEL(一致性-删除)分数,表明其识别的高排名特征对模型预测至关重要。

总体而言,该实验证实了ABIGX与WinClip等大规模基于Transformer的模型兼容,并在工业图像的多样化异常类型上生成忠实且简洁的解释。ABIGX对不同模态和模型架构的泛化能力进一步巩固了其作为广泛适用且统一的表格型和视觉FDC任务可解释性框架的价值。

7.4.4 ABIGX详细解释

                 

附录图B.6展示了ABIGX在更多具有细微且视觉上不明显缺陷的MVTec-AD故障样本上的解释。这些具有挑战性的案例对于评估解释方法的必要性具有重要意义。对于FDC模型赋予高故障置信度的样本,ABIGX突出了与真实缺陷区域高度吻合的局部区域,表明模型正关注于语义相关的特征。相反,对于故障置信度较低的样本,ABIGX产生较弱或错位的归因,反映了模型的不确定性或未能识别异常。该分析表明:(1)对于难以观测的缺陷,解释对于验证模型是否基于有意义的证据做出决策至关重要;(2)解释作为诊断工具,有助于验证正确决策并解释失效案例,以进行进一步的模型诊断。   

7.5 计算成本      

尽管FDC领域的研究兴趣日益增长,但为基于深度神经网络的FDC模型生成准确且具备理论基础的解释仍然是一项关键挑战。虽然传统的故障诊断方法和深度学习解释器均已得到应用,但仍存在若干局限性:

附录表B.15展示了每样本的平均解释时间。尽管计算开销有所增加,ABIGX仍保持实际可用的高效性:轻量级模型(如PCA、AE、MLP和小型CNN)的平均运行时间约为1至2秒,即使对于ViT(Vision Transformers,视觉Transformer)等更复杂的架构,也能保持在数十秒以内。

传统方法如CP(Contribution Plot,贡献图)和RBC(Reconstruction-Based Contribution,基于重构的贡献)计算开销极低,对于PCA模型解释时间近乎为零。这种高效性源于其不依赖迭代优化或梯度计算。然而,这一速度是以非线性场景下的泛化能力和可解释性为代价的。

ABIGX的主要计算成本来自梯度计算,这对于AFR(Adversarial Fault Reconstruction,对抗故障重构)过程和IG(Integrated Gradients,积分梯度)计算均不可或缺。因此,总运行时间主要取决于模型的正向和反向传播次数,由AFR和IG中使用的步长参数决定。实践中,我们通常将AFR的步数设置为与IG相等,这使得ABIGX所需的梯度评估次数约为标准IG方法的两倍,约为简单基于梯度方法的2×步数倍。

在实践中,ABIGX可通过小批量(mini-batching)大幅加速,即在GPU显存限制下的批量大小(如我们设置中ViT为8)内,对多个样本并行执行AFR基线搜索和IG积分。

7.6 实现细节      

我们在附录中总结了模型架构和关键ABIGX超参数,以确保可复现性。具体而言,表B.5和B.14报告了实验中训练的网络结构;表B.6列出了所有四种被解释模型的主要ABIGX超参数。

7.6.1 步长、步数与收敛性

           

为避免AFR基线欠优化影响归因结果,我们在所有具有合理运行时间的实验中采用保守的(较大的)迭代预算,以确保优化不足不会混淆归因质量。四种被解释模型的步长和步数敏感性分析详见附录表B.7至B.10。

为诊断AFR基线搜索过程中的收敛性,我们监测迭代过程中的表征SPE值和模型预测置信度。具体而言,当SPE曲线达到平台期时(即连续迭代中SPE的相对下降变得可忽略不计),我们将该运行标记为已收敛。根据表B.16中的聚合收敛率,在默认步长和步数下,大多数样本表现出急剧的SPE下降后趋于饱和,表明跨模型的收敛行为稳定。我们还跟踪搜索过程中的模型正常性(图B.9),其通常与基于SPE的收敛诊断一致地趋于稳定。仅有少量运行仍陷入局部最优(未达到足够高的正常性置信度)或达到最大迭代预算上限。

7.6.2 故障分类器表示层

           

在本文的故障分类设置中,ABIGX需要选择内部表征h(x)以定义分类SPE并驱动AFR基线搜索。我们选择倒数第二层作为默认表征,因其提供了稳定且语义有意义的特征空间,同时避免了最终对数几率层可能出现的饱和和标签特定坍缩。我们通过在TEP和晶圆图分类器上的敏感性研究验证了这一选择:在候选层中,倒数第二层表征始终产生最佳或接近最佳的解释正确性和一致性(附录图B.7和表B.13)。因此,除非另有说明,所有分类器实验均采用倒数第二层作为表征。

7.6.3 失效案例

           

我们观察到ABIGX的两种反复出现的失效模式。第一种情况是AFR基线优化失败(最终基线的正常性较低)。在某些样本中,基于PGD的AFR搜索虽然降低了表征空间距离,但仍收敛至正常性置信度相对较低的基线,表明优化陷入了较差的局部最优。当最终基线仍非正常时,所得IG路径不再代表有意义的"故障到正常"过渡,归因正确性(如GT-AUC)随之下降(图B.8)。作为实际保障措施,在这些情况下我们可以回退至期望IG,即从输入空间正常样本池中抽取基线,而非依赖搜索得到的AFR基线,从而提高鲁棒性。

第二种情况是低置信度模型预测(本质上模糊的样本)。这类失效源于模型本身对被解释样本赋予较低的故障置信度。在此情况下,归因可能与人工标注的真实值对齐不佳,但这并非因为解释器不正确,而是因为底层决策本身不确定(图B.6)。这种行为是符合预期的:ABIGX旨在解释模型的决策机制,当模型置信度较低(或接近决策边界)时,任何忠实的解释都将反映这种模糊性,而非强制与外部标注对齐。

8 结论

本文提出了一种统一框架ABIGX,用于解释通用FDC模型。ABIGX源于CP和RBC的核心要素,这两种方法在PCA模型故障诊断中已取得显著成功。本文提出的新型故障重构方法AFR是ABIGX的核心组成部分。AFR从对抗攻击的视角重新构建现有方法,并通过新型分类SPE指标将FR推广至故障分类任务。对于可解释故障分类,我们从故障类别模糊化的角度分析了各方法的性能。与显著性图和IG相比,我们证明了ABIGX能够以更低的故障类别模糊化程度提供更准确的变量贡献。对于可解释故障检测,我们从理论上证明了线性CP和RBC与ABIGX的等价性。通过定量指标,在广泛模型和数据集上的实验验证了ABIGX在故障诊断和深度学习解释两个领域中相较于其他先进解释器的总体优越性。 

编辑:Leo

校核:李正平、陈凯歌、曹希铭、赵学功、白亮、任超、海洋、陈宇航、Kira、Tina、王金、赵诚、肖鑫鑫、张优

该文资料搜集自网络,仅用作学术分享,不做商业用途,若侵权,后台联系小编进行删除



来源:故障诊断与python学习
ACTSystem非线性半导体通用航空ANSApython芯片海洋UM理论电机材料数字孪生物流控制人工智能数控
著作权归作者所有,欢迎分享,未经许可,不得转载
首次发布时间:2026-09-08
最近编辑:1月前
故障诊断与python学习
硕士 签名征集中
获赞 87粉丝 153文章 340课程 0
点赞
收藏
作者推荐

Information Fusion|重庆大学攻克域偏移难题,提出零样本复合故障诊断新框架

本期聚焦旋转机械零样本复合故障诊断新进展。零样本故障诊断长期受域偏移与特征混淆两大技术瓶颈制约,单一原型表征不足、生成样本稳定性差更是领域普遍短板。重庆大学秦毅教授团队近日提出三原型融合(triple-prototype fsion, TPF)框架,为旋转机械复合故障诊断提供了全新技术路径。该框架融合仿真生成、实测特征、先验知识三类异构原型,从表征源头弥合域间分布差异;通过多分布融合对比学习优化正样本构造,强化特征判别力以破解特征混淆;配套模型 - 样本双维度稳定扩散模型,全面提升生成样本质量。实验证实,该方法无需复合故障训练样本即可实现高精度诊断,性能优于8种主流基准方法,相关研究内容已开源于https://github.com/QinYi-team/TPF。论文基本信息论文题目:Beyond single prototype: A multi-prototype fusion framework with contrastive learning and diffusion model for zero-shot fault diagnosis 论文期刊:Information Fusion论文日期:2026年论文链接:https://doi.org/10.1016/j.inffus.2026.104399作者:Lv Wang (a), Junyu Qi (b), Yi Qin *(a)机构:a. State Key Laboratory of Mechanical Transmission for Advanced Equipment, Chongqing University, Chongqing 400044, China; b. Reutlingen Research Institute (RRI), Reutlingen 72762, Germany.通讯作者简介:秦毅,重庆大学教授,于重庆大学机械电子工程系获工学博士学位。长期致力于智能故障诊断与装备健康管理、动力学建模与数字孪生、智能无人系统以及智能制造等领域的创新研究。秦毅教授在国内外权威学术期刊及高水平会议上发表论文230余篇,研究成果涵盖《IEEE Transactions on Industrial Informatics》《Mechanical Systems and Signal Processing》《Reliability Engineering and System Safety》等知名期刊,学术影响力广泛。他长期担任国家重点研发计划、长江学者奖励计划、国家自然科学基金等多项国家级与省部级科研项目评审专家,深度参与国际学术交流与期刊建设。在学术组织方面,秦毅教授是中国机械工程学会、中国振动工程学会的高级会员,并担任IEEE可靠性学会重庆分会副主席,在智能故障诊断与动态测试技术领域积累了深厚的学术声望。(来源:https://faculty.cqu.edu.cn/YiQin)目录1 摘要2 引言3 预备知识3.1 扩散模型3.2 马尔可夫链蒙特卡罗(MCMC)方法3.3 对比学习4 三原型融合模型4.1 问题定义4.2 整体框架4.3 模型 - 样本稳定扩散模型4.4 多分布融合对比学习4.5 原型融合模块5 实验验证5.1 BCF 数据集实验5.2 东北电力大学(NEEPU)数据集实验5.3 补充实验6 结论1 摘要随着工业装备复杂程度不断提升,旋转机械复合故障的发生频次日益增多。然而故障数据的获取难度极大。零样本学习领域近年取得的研究进展,为训练阶段无对应样本时识别未知复合故障提供了可行技术手段。但现有方法普遍存在域偏移与特征混淆问题,导致诊断精度大幅下降。针对上述问题,本文提出一种三原型融合(triple-prototype fusion,TPF)框架,用于零样本故障诊断。首先,构建原型融合框架以解决域偏移问题。该框架融合仿真故障生成原型、实测故障提取特征原型与先验知识构建理论原型,缓解仅依靠单一原型信息不足带来的域偏移现象。其次,设计多分布融合对比学习机制处理特征混淆问题。该机制在正样本生成过程中引入多元分布先验信息,提升构造正样本的区分度。此外,本文提出模型 — 样本稳定扩散模型,用以增强样本生成稳定性。该模型分别从模型训练、样本去噪两个维度引入稳定约束机制,全方位提升生成样本的稳定程度。通过两组实验案例验证了三原型融合框架的有效性。该框架无需任何复合故障训练样本,即可完成未知复合故障诊断。与 8 种经典模型及前沿零样本学习算法开展对比测试,本文所提 TPF 框架具备更优诊断性能。关键词:复合故障诊断;扩散模型;对比学习;原型融合;零样本学习2 引言在现代工业中,旋转机械是大多数生产过程中不可或缺的组成部分,其稳定性和可靠性直接影响生产效率与安全。轴承等关键部件在长期运行中不可避免地会发生故障,这些故障可能导致设备停机,并造成重大经济损失和安全隐患。因此,及时、准确的故障诊断尤为重要。除单一故障外,旋转机械中还会出现复合故障。与单一故障相比,复合故障具有相互耦合的特点,诊断难度较大。因此,研究高精度的复合故障诊断算法十分必要。现有的复合故障诊断方法大致分为两类:基于信号处理的方法和基于人工智能的方法。基于信号处理的方法主要将复合故障分解为单一故障进行诊断,常用的方法包括小波变换和变分模态分解。尽管这些方法在某些场景下有效,但其特征提取需要大量专家知识,限制了其在复杂工业环境中的泛化性和可扩展性。基于人工智能的方法又可进一步分为浅层学习算法和深度学习算法。浅层学习技术通常利用传统机器学习方法进行复合故障诊断。基于深度学习的方法依赖大量数据进行训练,然而在实际应用中,获取大量数据(尤其是复合故障数据)通常不切实际。为解决目标数据收集困难的问题,提出了零样本学习方法。零样本学习方法仅使用可见类别的数据和辅助信息即可识别未见类别,在故障诊断领域也引起了广泛关注。尽管现有零样本诊断技术取得了可接受的性能,但仍存在若干挑战,包括:1) 零样本学习方法需要定义相应的原型作为基准。然而,大多数零样本诊断方法仅考虑设计的理论原型或生成原型。由于训练数据和测试数据具有不同的故障类型,仅基于训练数据设计的原型将与测试数据的实际原型存在显著偏差,导致诊断性能急剧下降。这一问题称为域偏移问题,如图1(a)左侧所示。2) 当前大多数零样本诊断研究缺乏对初始特征的预处理,或仅使用简单方法进行处理。如图1(b)左侧所示,由于缺乏训练,未见测试类型的特征往往呈现显著重叠,对后续诊断性能产生不利影响。因此,开发一种提高初始故障特征可分性的技术对于实现更高诊断精度至关重要。图1 零样本诊断中的现有问题与解决方案:(a)域偏移问题;(b)特征混淆问题除零样本诊断方法的局限性外,传统的信号扩散模型仅依赖前一样本进行去噪,容易导致去噪过程的不稳定性。为解决上述三个问题,本文提出了一种三原型融合(TPF)模型用于零样本复合故障诊断。首先,构建了三原型融合框架以缓解零样本学习中的域偏移问题。该三原型框架的设计灵感来源于多源信息融合的互补性。它构建并整合了三种不同类型的原型:生成原型、特征原型和理论原型。生成原型完全由模型生成,基于可见数据输入扩散生成模型以生成未见域样本,然后进行聚类,不依赖任何未见域的真实观测数据,反映了模型自身的固有构建能力。特征原型直接从真实未见域故障数据中提取,通过无监督聚类从实际数据中获得,能够反映未见类别的实际特征。理论原型基于先验知识或故障机理构建,代表理论层面定义的理想参考。理论原型是零样本诊断中原型构建的常用方法,通过先验知识建立可见域和未见域的共同属性,从而仅利用可见域数据即可构建两个域的原型。生成原型和特征原型是本文提出的创新构建方法。最终原型从三个不同角度构建:数据的先验知识、模型学习的特征以及实际未见域特征。通过三者之间的互补性,增强了原型的完整性,有效缓解了未见域的偏移问题。此外,本文还开发了一种多分布融合对比学习机制。该机制有效提高了初始特征的可分性,从而提升了后续诊断性能,如图1(a)右侧所示。在该机制中,提出了一种新的正样本构建方法,引入多种分布的先验知识生成具有额外内在特征的样本,提高了提取特征的可分性。在该机制中,使用马尔可夫链蒙特卡洛(MCMC)方法对先验分布知识进行采样,并使用所提出的扩散模型生成样本。此外,还构建了模型-样本稳定扩散模型(model-sample stabiliztion diffusion model, MSSDM)以提高扩散模型在生成样本时的稳定性。在MSSDM中,引入了指数移动平均(exponential moving average, EMA)机制以增强模型训练的稳定性,并建立了回溯机制以增强去噪过程的稳定性。本文的主要贡献可总结如下:提出了一种三原型融合框架以解决零样本学习中的域偏移问题。在该框架中,构建的原型在了解任何故障或模型生成信息之前整合了样本的实际特征信息,提供了更全面的原型以缓解域偏移问题。为增强特征的可分性,开发了一种多分布融合对比学习机制。在生成正样本阶段注入多分布信息,引导生成的样本具有额外的基本特征,提高了提取特征的可分性。提出了一种模型-样本稳定扩散模型(MSSDM)以增强基于扩散的框架中生成样本的稳定性。该框架集成了两种关键稳定机制:引入指数移动平均机制以增强模型训练的稳定性;设计回溯机制以增强样本去噪过程的稳定性。通过上述两种机制,样本生成的稳定性得以增强。3 预备知识3.1 扩散模型扩散模型是一种基于概率扩散过程的生成模型。该模型利用扩散的逆过程,通过逐步去噪生成高质量数据样本。扩散模型主要包括两个过程:扩散过程与逆过程。扩散过程在每个时间步 中逐步向样本添加噪声。在此过程中,当前样本仅与前一时刻相关,可视为一个马尔可夫过程。每一步的转移概率 由下式给出: 其中 是第 步获得的数据, 表示高斯分布的方差超参数, 表示单位矩阵。第 步的数据可表示如下: 其中 表示第一步的高斯噪声。数据可通过重参数化技巧表示如下: 其中 , 。该处理技巧解决了随机采样过程中的求导难题。反向扩散过程为逐步去噪过程:从标准正态分布噪声出发,逐级去除噪声,最终还原生成目标数据。各时间步的样本更新公式可表示为: 其中 为神经网络训练得到的噪声预测网络, 代表第 步的标准差, 为高斯噪声。模型依照上述步骤从时间步 迭代至 0;每一步均对当前样本更新并持续降噪,直至输出最终生成样本。扩散模型通常选用U-Net 作为基础骨干网络。U-Net 具备多尺度特征提取能力与跳跃连接结构,能够辅助模型更好地复原样本细节。本文将模型预测噪声与真实噪声之间的均方误差作为损失函数,具体表达式如下: 3.2 马尔可夫链蒙特卡罗(MCMC)方法 马尔可夫链蒙特卡罗(MCMC)方法是从复杂分布中采样的有效工具。MCMC 的基本原理是构造一条马尔可夫链,使采样数据的状态分布最终收敛至目标分布。在 MCMC 方法中,目标平稳分布与状态转移矩阵满足细致平衡条件,表达式如下: 其中 为状态分布, 代表状态转移矩阵, 为接受概率。上述采样流程中,若接受概率 数值偏低,会造成大量样本被拒绝。为此引入梅特罗波利斯-黑斯廷斯(Metropolis-Hastings)采样算法解决该问题。对上式两侧同乘相同系数进行缩放后,在依旧满足细致平衡条件的前提下,接受概率能够得到提升。因此改进后的接受概率表达式可改写为: 通过上述过程,定义了选定的状态转移矩阵、目标平稳分布以及所需的样本数量。通过采样即可获得对应于所需分布的样本集。3.3 对比学习对比学习是一种旨在增强数据可分性的无监督技术。其目标是最大化不同样本之间的一致性,同时最小化与其他实例的一致性。目前,对比学习在故障诊断领域已引起广泛关注。尽管该方法展现出了相当的性能,但在对比样本的构建方面缺乏优化,而这对后续特征提取具有显著影响。为弥补这一空白,本文设计了一种多分布融合对比学习机制,以在样本构建阶段增强判别性特征提取。4 三原型融合模型在本节中,首先介绍零样本诊断的相关定义与符号。随后,描述所提出的TPF方法的诊断流程。接下来,对该框架中提出的三个关键模块进行详细说明:模型-样本稳定扩散模型、多分布融合对比学习机制以及三元原型融合模块。4.1 问题定义零样本复合故障诊断任务可定义如下:该任务的目标是通过可见单一故障识别未见复合故障。可见单一故障类别可表示为 。未见复合故障类别可表示为 。其中, 和 分别表示可见单一故障与未见故障的样本, 和 分别为可见类别与未见类别的标签。在零样本学习中,属性 和 通常用于辅助识别未见类别的类型。参数 和 分别表示可见样本与未见样本的数量。参数 和 分别表示属性与样本的维度。零样本复合故障诊断的目标可形式化表示为: 其中 是未见故障的预测标签, 表示由 和 组成的属性。4.2 整体框架TPF模型的整体框架如图2所示。首先,生成正样本与生成原型。正样本的构建是对比学习的重要组成部分。在TPF模型中,正样本由所提出的多分布引导的MSSDM生成。从可见故障样本中提取单分布特征并进行融合。使用MCMC采样生成所需数量的多分布表示,随后将这些表示嵌入MSSDM中以合成所需的正样本。具体生成过程详见第4.3节。此外,生成的原型由MSSDM、特征提取器和聚类算法共同构建。 图2 所提出的TPF模型的整体框架第二步是对比训练与特征对齐。对比学习框架中的三元组样本由锚点样本、正样本和负样本组成,可表示为 。TPF中某一类别的锚点样本采用原始样本,而对应的负样本则来自另一类别的原始样本。例如,选择四种故障类型作为锚点样本,包括正常状态、内圈故障、外圈故障和滚动体故障;对应的负样本分别具有以下故障类型:内圈故障、外圈故障、滚动体故障和正常状态。此外,锚点样本与对应的正样本属于同一类别。将锚点样本、正样本和负样本按不同故障顺序排列,形成训练数据集。采用宽混合扩张卷积神经网络(WHDCNN)作为特征提取器,提取的中间特征可表示为 ,其中 表示特征的维度。在本文框架中,最后一层提取的特征称为属性特征 。三元组损失用于缩小锚点特征 与正特征 之间的差异,同时扩大 与负特征 之间的差异。三元组损失函数定义如下: 其中 表示边际系数。均方误差损失用于将属性特征 与其对应的属性(即理论原型 )对齐。该过程可形式化表示为: 其中 表示属性特征,该特征源自特征提取器的最后一层。总体训练损失可表示为: 其中 表示控制三元组损失影响的超参数。然后,使用自适应矩估计优化器更新全连接层的参数 ,如下所示: 图2中的第三部分展示了所提出框架的测试阶段。首先,通过训练好的特征提取器获取未见故障的特征 。然后,通过聚类与匹配算法获取特征原型。接着,最终原型由三个不同层面的原型融合而成。不同原型的具体构建与融合过程详见第4.5节。最后,利用最近邻估计对最终原型与未见特征进行匹配。第 个未见样本的标签表示如下: 4.3 模型 - 样本稳定扩散模型4.3.1 特征提取器U-net广泛用于扩散模型中。为了更好地适应具有特定标签的一维数据生成,对网络结构进行了重新设计。在所设计的网络中引入了时间嵌入和标签嵌入。所设计的U-net网络结构如图3所示。该网络主要包括收缩过程和扩展过程。收缩过程包含三个模块,每个模块中执行两次卷积操作和一次最大池化操作。在扩展过程中,对数据执行与压缩过程类似的操作,即通过两次反卷积操作和一次上采样操作。在扩展过程中使用跳跃连接以丰富原始特征信息。 图3 所设计的U-Net结构4.3.2 模型-样本稳定机制为进一步提高扩散模型的稳定性,设计了一种模型稳定机制,具体过程如图4所示。 图4 所提出的MSSDM的训练过程该机制包含两项改进:模型稳定性与样本稳定性。引入指数移动平均( EMA )机制以稳定模型。 EMA 提供更平滑的参数估计,有助于减少单次训练迭代引起的随机波动并提高稳定性。 EMA 机制中的参数 可按如下方式更新: 其中 表示权重因子, 是第 轮时的模型参数。提出去噪稳定机制以稳定样本。传统的扩散模型大多依赖前一样本进行去噪,这容易导致样本不稳定,且难以在去噪过程中纠正错误。为解决这一问题,在去噪过程中的规律时间步引入前一样本的特征,以减少不同时间步之间的显著特征偏移。具体过程可表述如下: 其中 表示扩散模型的总时间步数, 表示控制初始样本比例的比重系数。4.4 多分布融合对比学习为提高提取特征的可分性,在所提出的框架中引入对比学习机制。在对比学习中,正样本与负样本的构建一直是一个挑战。为此,构建了一种多分布引导的正样本构建机制。该机制引入数据分布的先验知识来构造正样本。此外,如图5(a)和(b)所示,单一分布特征存在显著重叠,导致区分能力相对较低。相比之下,图5(c)显示融合分布的特征实现了较高的可分性。因此,引入多种分布特征的先验知识以生成更具判别性的样本。在图5中,NC、IF、OF和BF分别表示轴承的不同健康状态,其中NC表示正常状态,IF表示内圈故障,OF表示外圈故障,BF表示滚动体故障。在图5(a)和(b)中,x轴表示样本特征的均值或方差,y轴表示通过核密度估计方法估计的对应概率密度。在图5(c)中,x轴表示样本特征的均值,y轴表示样本特征的方差,z轴表示概率密度。多分布融合对比学习的具体过程如下所述。 图5 轴承复合故障数据集中单一故障的分布:(a)均值分布;(b)方差分布;(c)融合分布第一步是获取数据的分布特征 ,将所获得的分布作为目标分布 。第二步是对数据的特征分布进行采样,例如均值分布和方差分布。采用MCMC算法对目标分布进行采样,以正态分布 作为提议分布。在第 步,由提议分布生成的样本 的接受率可计算如下: 接下来,从均匀分布中生成一个随机数 ,用于决定是否接受样本 :其中 表示第 步的状态。通过连续生成和迭代,可以得到一系列服从目标分布的分布信息 。通常,前 个样本可以忽略,因为马尔可夫链尚未达到其平稳分布。第三步是分布信息嵌入。将数据的原始标签信息和分布信息融合在一起,输入到MSSDM模型中进行生成。4.5 原型融合模块本节介绍三种原型(生成原型、特征原型和理论原型)的构建与融合方法。三种原型构建的整体框架如图6所示。理论原型 通过对单一故障的数据与故障机制进行分析而构建。 图6 原型融合模块的整体框架每种类型的故障原型包含九个不同属性,涵盖故障类型、脉冲特征及脉冲幅值,详见表2。表2 为滚动轴承设计的九个属性 幅值相关属性源自信号的均方根(RMS)值。其计算公式定义如下: 其中 表示时域数据。频率相关属性通过对信号频谱的分析进行提取: 其中 表示频域数据。对于单一故障,可直接获取其数据及对应的属性。对于复合故障,其属性通过对单一故障属性进行逻辑运算推导得出。对于幅值相关属性,在融合不同单一故障的幅值相关属性时,取平均幅值的上限作为复合故障的幅值相关属性。故障类型和频率相关属性通过以下方式获得: 其中 和 分别表示可见故障 和未见故障 原型的前 个元素。这些元素对应于故障类型及相关频率。生成原型 的建立基于MSSDM。首先,将单一故障样本及其对应的属性标签输入MSSDM进行训练。然后,由训练好的MSSDM使用复合故障的属性标签生成复合故障样本 。接着,利用单一故障的实际样本 训练特征提取器。随后,通过训练好的特征提取器获取生成复合故障的特征 。最后,采用高斯混合模型(GMM)寻找生成复合故障样本的生成原型。高斯混合分布定义如下: 其中 表示分布的数量, 是分布权重, 是均值, 表示协方差矩阵, 表示第 个高斯分布的概率密度函数。GMM使用期望最大化算法迭代参数。数据 属于第 个高斯分布的概率 可计算如下: GMM中均值向量 的更新方法可表示如下: 在所提出的框架中,生成的复合故障特征的均值向量被用作生成原型: 特征原型 的构建过程与前述生成原型类似。首先,通过在实际单一故障样本上训练得到的特征提取器获取实际复合故障特征。随后,使用GMM算法对实际复合故障特征的均值矩阵 进行聚类。与生成原型不同的是,特征原型对无标签的未见类别特征进行聚类。因此,设计了一种最近邻匹配算法来获取特征原型的标签。由于理论原型的标签是已知的,特征原型的标签通过这两类原型之间的距离来定义。距离矩阵 可计算如下: 特征原型的标签取为其最近理论原型的标签。然后,所有特征原型的标签 均可获得。具体特征原型可表述为: 特征原型在测试阶段构建,不使用任何未见类别的标签。此外,训练阶段不涉及任何未见类别的数据。该过程完全符合标准的零样本学习范式。最后,将三种不同的原型融合在一起,融合原型可以表述如下: 其中 和 分别为生成原型和特征原型的权重系数。5 实验本节通过两个轴承复合故障诊断实验对TPF方法进行评估。第一个实验使用定制构建的轴承复合故障(BCF)测试平台,第二个实验采用东北电力大学(NEEPU)的轴承故障平台。为进一步证明所提出模型的优势,开展了全面的消融研究,包括不同模块和原型的消融实验。此外,还进行了参数敏感性分析以确定最优参数。每个实验重复五次,并报告平均诊断结果及相应的标准差。5.1 BCF实验5.1.1 实验描述定制的故障实验平台由以下部件组成:负载电机、转速-扭矩测量仪、行星齿轮减速器、轴承测试模块及驱动电机。实验设计在四种扭矩负载条件下评估七种不同的轴承健康状态。轴承故障分为七类:正常状态(NC);三种单一故障状态:外圈(OF)、内圈(IF)和滚动体(BF);以及三种复合故障状态:外圈与内圈(OI)、滚动体与外圈(BO)、滚动体与内圈(IB)。负载电机施加四种工况:0、2、4和6 N⋅m(负载0–3)。驱动电机转速和采样频率分别固定为500 rpm和12 kHz。平台结构如图7(a)所示,七种不同测试轴承的图像如图7(b)所示。 图7 BCF实验平台与测试轴承:(a) 具体结构;(b) 测试轴承理论原型的构建是零样本学习的基础之一。该原型基于第4.5节中建立的九个属性,并结合故障机理及BCF数据集中单一故障数据的分析得出,如图8所示。此外,在所提出的TPF框架中,模型仅使用单一故障数据进行训练,并在复合故障上进行评估,采用70%/30%的训练/测试划分。为扩充数据集,采用滑动窗口技术,每个样本包含3072个数据点。模型采用的学习率为0.00005,多分布融合对比学习权重 。 图8 BCF实验中设计的理论原型 5.1.2 结果与讨论为验证TPF框架在复合故障诊断中的有效性,将该模型与八种典型及当前最优方法进行了比较。此外,还将所提方法与FDAT的一种变体FDAT-I进行了比较,比较的更多细节详见第5.2.2节。所有对比方法均使用与所提出TPF模型相同的属性,以确保公平比较。DAP和IAP方法中的属性分类器均采用卷积神经网络。FDAT和SCE方法使用基于随机森林的分类器,与其原始实现一致。对于DEM方法,特征维度设置为256,与所提出的TPF模型一致。LIV方法保留其原始超参数配置进行识别。四种负载条件下各方法的诊断精度如表3所示。表3 BCF实验中九种不同方法的平均诊断精度 TPF方法凭借其强大的特征分离能力和域偏移缓解能力,在不同方法中获得了最高的平均诊断精度。所提出的ZSPE框架在四个任务中的精度分别达到66.33%、82.62%、97.11%和90.71%。总体而言,不同工况下的诊断精度是可接受的。尽管在负载0条件下诊断精度的提升相对较低,但在工况1–3中,所提出方法相比其他方法在诊断精度上表现出明显的改善。特别地,当仅使用单一故障样本进行训练时,在负载2和负载3条件下复合故障的平均诊断精度均超过90%。这一观察表明TPF方法在零样本故障诊断方面具有潜力。此外,为进一步进行定量分析,图9展示了TPF在四种负载条件下的混淆矩阵。可以看出,由于不同负载条件下数据分布的变化,OI和IB的诊断精度存在波动,而BO的诊断精度达到100%。总体而言,TPF框架仅使用单一故障数据进行训练即可实现复合故障的诊断。 图9 TPF模型在BCF数据集上四种不同负载下获得的混淆矩阵5.2 NEEPU实验5.2.1 实验细节为验证TPF框架的泛化能力和优势,在NEEPU数据集上进行了额外实验。与定制轴承复合故障实验类似,NEEPU数据集包含七种健康状态:正常状态(NC)、单一故障(OF、IF和BF)以及复合故障(OI、BO和IB)。在本实验中,通过磁力制动器对轴承施加不同负载,分别为0 N·m(负载0)、0.1 N·m(负载1)、0.2 N·m(负载2)和0.3 N·m(负载3)。训练集为每个单一故障类别分配700个样本,测试集为每个复合故障类别分配300个样本,与BCF实验一致。超参数(如学习率和权重参数)与BCF实验设置相同。此外,与BCF实验类似,理论原型基于NEEPU数据集的分析得出,如图10所示。 图10 NEEPU实验中设计的理论原型5.2.2 结果与讨论NEEPU实验的零样本故障诊断结果如表4所示。所提出的TPF框架同样与BCF实验中评估的八种典型及当前最优模型进行了基准比较。TPF框架在四种负载条件下的诊断精度分别为74.07%、90.58%、84.06%和82.96%。与BCF实验相比,不同条件下的诊断精度相对稳定,这可能归因于不同工况之间的差异较小。TPF的诊精精度高于其他零样本学习方法,总体诊断结果令人满意。表4 实验中九种不同方法的平均诊断精度 此外,所提出TPF在四种不同负载条件下的混淆矩阵如图11所示,提供了每种故障类型的详细诊断精度,进一步展示了TPF模型的诊断能力。 图11 所提出的TPF模型在NEEPU数据集上四种不同负载下获得的混淆矩阵值得注意的是,在某些负载条件下,几种经典方法相比部分较新方法展现出更高的诊断精度。这一观察结果可能与以下方面有关:首先,经典方法通常采用相对简单的训练流程,并已在各种实际场景中得到广泛验证,这可能有助于其在不同数据集上实现稳定的泛化性能。相比之下,较新的方法通常包含更复杂的特征预处理或联合优化策略。例如,FDAT和SCE涉及特征提取器和属性分类器的联合训练,而LIV将信号转换为图像表示并联合训练特征提取器和生成对抗模型。尽管这些设计旨在增强表示能力,但模型复杂度的增加也可能引入额外的优化挑战,从而在特定条件下影响泛化性能。其次,更复杂的架构通常对超参数配置更为敏感。当训练数据有限或超参数未经过广泛调优时,其潜力可能无法充分发挥,导致在某些场景下诊断性能较低。此外,为验证假设,对FDAT方法进行了架构改进。原始实现采用传统机器学习技术作为特征提取器,并要求特征提取器和属性分类器进行联合训练,这可能增加过拟合风险并导致性能较旧方法下降。为检验这一可能性,将原始特征提取和属性分类模块替换为基于CNN的结构,并将该变体记为FDAT-I。两个数据集上的实验结果列于表3和表4。对于BCF数据集,在四种工况下的诊断精度分别为71.93%、70.17%、67.57%和77.30%。对于NEEPU数据集,对应的精度分别为71.63%、82.33%、72.50%和74.33%。与原始FDAT方法相比,改进版本在不同条件下展现出一致的性能提升,并优于较早的方法。这为本文的假设——特征提取器和属性分类器的联合训练会降低零样本故障诊断的精度,提供了额外的实证支持。然而,改进模块结构可以缓解这一问题。5.3 拓展实验5.3.1 不同模块的消融实验为验证所设计的原型融合模块与多分布融合对比学习机制的有效性,在BCF和NEEPU数据集上进行了消融实验。不同负载下的两组实验结果如图12所示。横坐标标签中的WCWF-0表示在负载条件0下,TPF不含对比学习机制和原型融合模块。类似地,WC表示不含多分布融合对比学习机制,WF表示不含原型融合模块。无论是对比学习机制还是原型融合模块,单独缺失时均未能达到最高诊断精度。这一观察结果表明,改善判别性特征和校正域偏移的必要性。综合来看,所提出的TPF在多分布融合对比学习机制和原型融合模块的支持下,在所有上述实验中均能取得最佳性能。这表明所提出的MSSDM和原型融合模块在零样本诊断任务中是有效的。 图12 不同数据集中消融实验的平均诊断精度:(a) BCF数据集;(b) NEEPU数据集5.3.2 不同原型的消融实验为进一步评估每个原型的贡献,进行了多项消融研究。两个数据集在八种负载条件下的平均诊断精度如图13所示。符号WG、WF和WT分别表示TPF模型不含生成原型、特征原型和理论原型。例如,WG-0表示负载条件0下不含生成原型的TPF模型的诊断精度。结果表明,缺失任一原型均会导致诊断精度下降,证实了每个组件的独立贡献。值得注意的是,理论原型对模型性能的影响最为显著,在某些场景下,其消融导致精度下降近20%。这一显著影响可归因于理论原型在构建过程中融合了已知数据和先验知识,同时在训练过程中具有监督性参与。总体而言,三种原型的融合实现了最高的诊断精度,表明每种原型在模型中均发挥着关键且非冗余的作用。 图13 两个数据集中四种不同负载下原型消融实验的平均诊断精度:(a) BCF数据集;(b) NEEPU数据集5.3.3 计算成本分析为提供计算成本比较,基于三项指标进行了分析:参数量、浮点运算量和100个样本的测试时间。十种方法的计算成本对比如表5所示。表5 十种不同方法的计算成本 参数量和FLOPs主要针对深度学习和机器学习方法进行计算。在FLOPs计算中,除LIV方法使用尺寸为64×64的图像作为输入外,其余方法均以长度为3072的单个样本作为输入。测试时间指样本输入模型后的推理时间。对于FDAT和SCE方法,所使用的随机森林为非参数模型,其参数量取决于训练样本规模且具有一定随机性,基于100个训练样本提供了近似估计。在对比方法中,DAP、IAP、ESZSL、SJE和FDAT-I使用相似的卷积神经网络作为特征提取器,因此参数量和FLOPs相当。LIV使用了更复杂的特征提取器和生成对抗网络,导致参数量最高。所提出的TPF方法由于集成了扩散模型和三原型融合策略,具有相对更多的参数和更高的FLOPs。然而,其100个样本的平均测试时间仅为42.33毫秒,处于可接受范围内。总体而言,所提出TPF方法的计算成本是合理的。5.3.4 参数敏感性分析为研究不同系数 和 对诊断精度的影响,进行了超参数敏感性分析实验。 和 分别是生成原型和特征原型的权重系数。根据经验,将两个系数分别设置为 中的不同值,以观察其对精度的影响。首先,将两个系数的值设为相同。在NEEPU和BCF实验中的具体诊断精度如图14所示。在两个实验中,不同系数下的诊断精度略有波动。在NEEPU实验中,当系数设置为0.1时,在四种工况下均达到最高诊断精度。相反,在BCF实验中,当系数取0.15时获得最高诊断精度。 图14 (a) BCF和(b) NEEPU数据集上四种负载条件下等值融合系数的敏感性分析随后,为进一步分析敏感性,将 固定为第一步中每个数据集获得的最优值,并在相同范围 内变化 。结果如图15所示。 图15 两个数据集上四种负载条件下参数的敏感性分析实验:(a) BCF数据集;(b) NEEPU数据集结果表明,当 在测试范围内变化时,诊断精度保持相对稳定,但存在一定波动。总体观察表明, 过大可能导致性能轻微下降。在两个数据集上,当 分别设置为0.15和0.1时,达到最高的平均精度。因此,在NEEPU实验中, 和 均设为0.15,而在BCF实验中均设为0.1。6 结论为应对零样本故障诊断中的域偏移与特征融合挑战,本文开展了以下工作。首先,开发了一种三原型融合框架以缓解域偏移问题。该框架融合了完全由模型合成的生成原型、从故障数据中导出的特征原型以及基于先验知识构建的理论原型。所提出的TPF模型整合了来自先验知识、实际特征与训练模型三个维度的信息,有效缓解了单一原型信息不足所导致的域偏移问题。其次,针对特征融合问题,提出了一种多分布融合对比学习机制。该机制在生成正样本时引入了多种分布的先验信息,使构建的正样本更具判别性,从而引导提取更具判别力的特征。此外,还提出了一种模型-样本稳定扩散模型,以提高样本生成的稳定性。所提出的模型从模型训练和样本生成两个方面引入稳定机制以提升稳定性。在两个轴承复合故障实验中验证了所提出TPF的有效性。与八种典型模型及当前最优模型相比,所提出模型具有最高的诊断精度。编辑:Jin校核:李正平、陈凯歌、Leo、曹希铭、赵学功、白亮、任超、海洋、Tina、陈宇航、Kira、肖鑫鑫、赵诚、张优该文资料搜集自网络,仅用作学术分享,不做商业用途,若侵权,后台联系小编进行删除来源:故障诊断与python学习

未登录
还没有评论
课程
培训
服务
行家
VIP会员 学习计划 福利任务
下载APP
联系我们
帮助与反馈