本期聚焦旋转机械零样本复合故障诊断新进展。零样本故障诊断长期受域偏移与特征混淆两大技术瓶颈制约,单一原型表征不足、生成样本稳定性差更是领域普遍短板。重庆大学秦毅教授团队近日提出三原型融合(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.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 结论
随着工业装备复杂程度不断提升,旋转机械复合故障的发生频次日益增多。然而故障数据的获取难度极大。零样本学习领域近年取得的研究进展,为训练阶段无对应样本时识别未知复合故障提供了可行技术手段。但现有方法普遍存在域偏移与特征混淆问题,导致诊断精度大幅下降。
针对上述问题,本文提出一种三原型融合(triple-prototype fusion,TPF)框架,用于零样本故障诊断。首先,构建原型融合框架以解决域偏移问题。该框架融合仿真故障生成原型、实测故障提取特征原型与先验知识构建理论原型,缓解仅依靠单一原型信息不足带来的域偏移现象。其次,设计多分布融合对比学习机制处理特征混淆问题。该机制在正样本生成过程中引入多元分布先验信息,提升构造正样本的区分度。此外,本文提出模型 — 样本稳定扩散模型,用以增强样本生成稳定性。该模型分别从模型训练、样本去噪两个维度引入稳定约束机制,全方位提升生成样本的稳定程度。通过两组实验案例验证了三原型融合框架的有效性。该框架无需任何复合故障训练样本,即可完成未知复合故障诊断。与 8 种经典模型及前沿零样本学习算法开展对比测试,本文所提 TPF 框架具备更优诊断性能。
关键词:复合故障诊断;扩散模型;对比学习;原型融合;零样本学习
在现代工业中,旋转机械是大多数生产过程中不可或缺的组成部分,其稳定性和可靠性直接影响生产效率与安全。轴承等关键部件在长期运行中不可避免地会发生故障,这些故障可能导致设备停机,并造成重大经济损失和安全隐患。因此,及时、准确的故障诊断尤为重要。除单一故障外,旋转机械中还会出现复合故障。与单一故障相比,复合故障具有相互耦合的特点,诊断难度较大。因此,研究高精度的复合故障诊断算法十分必要。
现有的复合故障诊断方法大致分为两类:基于信号处理的方法和基于人工智能的方法。基于信号处理的方法主要将复合故障分解为单一故障进行诊断,常用的方法包括小波变换和变分模态分解。尽管这些方法在某些场景下有效,但其特征提取需要大量专家知识,限制了其在复杂工业环境中的泛化性和可扩展性。
基于人工智能的方法又可进一步分为浅层学习算法和深度学习算法。浅层学习技术通常利用传统机器学习方法进行复合故障诊断。基于深度学习的方法依赖大量数据进行训练,然而在实际应用中,获取大量数据(尤其是复合故障数据)通常不切实际。
为解决目标数据收集困难的问题,提出了零样本学习方法。零样本学习方法仅使用可见类别的数据和辅助信息即可识别未见类别,在故障诊断领域也引起了广泛关注。尽管现有零样本诊断技术取得了可接受的性能,但仍存在若干挑战,包括:
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.1 扩散模型
扩散模型是一种基于概率扩散过程的生成模型。该模型利用扩散的逆过程,通过逐步去噪生成高质量数据样本。扩散模型主要包括两个过程:扩散过程与逆过程。扩散过程在每个时间步 中逐步向样本添加噪声。在此过程中,当前样本仅与前一时刻相关,可视为一个马尔可夫过程。每一步的转移概率 由下式给出:
其中 是第 步获得的数据, 表示高斯分布的方差超参数, 表示单位矩阵。第 步的数据可表示如下:
其中 表示第一步的高斯噪声。数据可通过重参数化技巧表示如下:
其中
反向扩散过程为逐步去噪过程:从标准正态分布噪声出发,逐级去除噪声,最终还原生成目标数据。各时间步的样本更新公式可表示为:
其中
扩散模型通常选用U-Net 作为基础骨干网络。U-Net 具备多尺度特征提取能力与跳跃连接结构,能够辅助模型更好地复原样本细节。本文将模型预测噪声与真实噪声之间的均方误差作为损失函数,具体表达式如下:
3.2 马尔可夫链蒙特卡罗(MCMC)方法
马尔可夫链蒙特卡罗(MCMC)方法是从复杂分布中采样的有效工具。MCMC 的基本原理是构造一条马尔可夫链,使采样数据的状态分布最终收敛至目标分布。在 MCMC 方法中,目标平稳分布与状态转移矩阵满足细致平衡条件,表达式如下:
其中
上述采样流程中,若接受概率
通过上述过程,定义了选定的状态转移矩阵、目标平稳分布以及所需的样本数量。通过采样即可获得对应于所需分布的样本集。
3.3 对比学习
对比学习是一种旨在增强数据可分性的无监督技术。其目标是最大化不同样本之间的一致性,同时最小化与其他实例的一致性。目前,对比学习在故障诊断领域已引起广泛关注。尽管该方法展现出了相当的性能,但在对比样本的构建方面缺乏优化,而这对后续特征提取具有显著影响。为弥补这一空白,本文设计了一种多分布融合对比学习机制,以在样本构建阶段增强判别性特征提取。
在本节中,首先介绍零样本诊断的相关定义与符号。随后,描述所提出的TPF方法的诊断流程。接下来,对该框架中提出的三个关键模块进行详细说明:模型-样本稳定扩散模型、多分布融合对比学习机制以及三元原型融合模块。
4.1 问题定义
零样本复合故障诊断任务可定义如下:该任务的目标是通过可见单一故障识别未见复合故障。可见单一故障类别可表示为
其中
4.2 整体框架
TPF模型的整体框架如图2所示。首先,生成正样本与生成原型。正样本的构建是对比学习的重要组成部分。在TPF模型中,正样本由所提出的多分布引导的MSSDM生成。从可见故障样本中提取单分布特征并进行融合。使用MCMC采样生成所需数量的多分布表示,随后将这些表示嵌入MSSDM中以合成所需的正样本。具体生成过程详见第4.3节。此外,生成的原型由MSSDM、特征提取器和聚类算法共同构建。
图2 所提出的TPF模型的整体框架
第二步是对比训练与特征对齐。对比学习框架中的三元组样本由锚点样本、正样本和负样本组成,可表示为
其中
均方误差损失用于将属性特征
其中
总体训练损失可表示为:
其中
然后,使用自适应矩估计优化器更新全连接层的参数
图2中的第三部分展示了所提出框架的测试阶段。首先,通过训练好的特征提取器获取未见故障的特征
最后,利用最近邻估计对最终原型与未见特征进行匹配。第
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)融合分布
第一步是获取数据的分布特征
接下来,从均匀分布中生成一个随机数
其中
4.5 原型融合模块
本节介绍三种原型(生成原型、特征原型和理论原型)的构建与融合方法。三种原型构建的整体框架如图6所示。理论原型
图6 原型融合模块的整体框架
每种类型的故障原型包含九个不同属性,涵盖故障类型、脉冲特征及脉冲幅值,详见表2。
表2 为滚动轴承设计的九个属性
幅值相关属性源自信号的均方根(RMS)值。其计算公式定义如下:
其中
频率相关属性通过对信号频谱的分析进行提取:
其中
对于单一故障,可直接获取其数据及对应的属性。对于复合故障,其属性通过对单一故障属性进行逻辑运算推导得出。对于幅值相关属性,在融合不同单一故障的幅值相关属性时,取平均幅值的上限作为复合故障的幅值相关属性。故障类型和频率相关属性通过以下方式获得:
其中
生成原型
其中
GMM使用期望最大化算法迭代参数。数据
GMM中均值向量
在所提出的框架中,生成的复合故障特征的均值向量被用作生成原型:
特征原型
特征原型的标签取为其最近理论原型的标签。然后,所有特征原型的标签
特征原型在测试阶段构建,不使用任何未见类别的标签。此外,训练阶段不涉及任何未见类别的数据。该过程完全符合标准的零样本学习范式。最后,将三种不同的原型融合在一起,融合原型可以表述如下:
其中
本节通过两个轴承复合故障诊断实验对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,多分布融合对比学习权重
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 参数敏感性分析
为研究不同系数
图14 (a) BCF和(b) NEEPU数据集上四种负载条件下等值融合系数的敏感性分析
随后,为进一步分析敏感性,将
图15 两个数据集上四种负载条件下参数的敏感性分析实验:(a) BCF数据集;(b) NEEPU数据集
结果表明,当
为应对零样本故障诊断中的域偏移与特征融合挑战,本文开展了以下工作。首先,开发了一种三原型融合框架以缓解域偏移问题。该框架融合了完全由模型合成的生成原型、从故障数据中导出的特征原型以及基于先验知识构建的理论原型。所提出的TPF模型整合了来自先验知识、实际特征与训练模型三个维度的信息,有效缓解了单一原型信息不足所导致的域偏移问题。其次,针对特征融合问题,提出了一种多分布融合对比学习机制。该机制在生成正样本时引入了多种分布的先验信息,使构建的正样本更具判别性,从而引导提取更具判别力的特征。此外,还提出了一种模型-样本稳定扩散模型,以提高样本生成的稳定性。所提出的模型从模型训练和样本生成两个方面引入稳定机制以提升稳定性。在两个轴承复合故障实验中验证了所提出TPF的有效性。与八种典型模型及当前最优模型相比,所提出模型具有最高的诊断精度。