论文题目:A trackable multi-domain collaborative generative adversarial network for rotating machinery fault diagnosis
论文期刊:Mechanical Systems and Signal Processing
论文日期:2024年
论文链接:
https://doi.org/10.1016/j.ymssp.2024.111950
作者:Xin Wang, Hongkai Jiang *, Mingzhe Mu, Yutong Dong
机构:School of Civil Aviation, Northwestern Polytechnical University, 710072 Xi’an, China
通讯作者邮箱: jianghk@nwpu.edu.cn
作者简介:
姜洪开,男,1972年生,博士,西北工业大学教授/博导。西安交通大学仪器科学与技术专业博士学位。2006年5月进入西北工业大学航空宇航科学与技术博士后流动站从事博士后研究工作,并留校任教。主持航天科技创新基金、中国博士后科学基金等国家和省部级基金项目4项。参加国家973子课题项目1项、国家自然科学基金重点项目1项、国家自然科学基金面上项目2项、国家863高技术研究发展计划项目1项。(来源百度百科)
摘要
1 引言
2 生成对抗网络
3 所提出的方法
3.1 多域协同对抗策略
3.2 频率同步丢失
3.3 流跟踪因子
3.4 所提方法的流程
4 案例一:高速航空轴承数据集
4.1 数据集描述
4.2 故障诊断的实验验证
4.3 生成数据的可信度分析
5 案例二:电力机车轴承数据集
5.1 数据集描述
5.2 故障诊断的实验验证
5.3 生成数据的可信度分析
5.4 内部工作机制可信度分析
6 结论
(以上标记章节为本文内容)
获取足够的平衡数据在实际旋转机械故障诊断任务中颇具挑战。准确地从不平衡数据中诊断故障是一个紧迫的实际问题。生成对抗网络已成为解决这一问题的主流方法。然而,其复杂的训练机制和不透明的架构引发了信任危机,导致用户无法完全信任输出结果。因此,提出了一种可追踪的多领域协作生成对抗网络(TMCGAN)用于旋转机械故障诊断。TMCGAN的核心贡献在于实现全局可解释生成和可信分类,这包括三个具体方面。首先,构建了多领域协同对抗策略,依次学习来自不同领域的信号的关键特征信息,从而实现多领域协同能源供应的全面训练。其次,设计了并行频率损失,以整合多维频率细节信息,丰富反馈,形成更高效的对抗训练闭环。最后,开发了流跟踪因子,阐明内部工作机制,提供实时跟踪反馈,解释底层决策原理,从而提高可解释性。两个案例研究表明,由TMCGAN赋能的分类器在旋转机械故障诊断中表现出色,同时保持高度可信。
案例一使用了来自高速航空轴承[40]的数据集来验证TMCGAN的有效性。需要强调的是,实验的比例因子设定为1,权重因子为0.1。生成器和判别器之间的训练平衡保持在1:4的比例。在VAE中,批次限制为100,每个数据点由一个样本表示。变分自编码器(VAE)、梯度惩罚和GAN的Wasserstein距离(WGAN-GP)、变分自编码生成对抗网络(VAEGAN)、DFEGAN[41]和WCGAN[33]与TMCGAN进行了比较。WGAN-GP和VAEGAN的架构与所提出的方法对齐,而DFEGAN和WCGAN的架构则遵循其原始配置。此外,所有模型中的生成器和判别器均使用Adam优化算法进行优化,学习率为0.0002,在60000次训练迭代(本文中一次训练迭代指一个iteration,下同)中,批量大小为8。这种统一配置确保了实验比较的公平性。
4.1 数据集描述
图6展示了实验装置。该装置由三个主要部件组成:主轴、加速度计和轴承。主轴的速度调整通过频率转换器实现,采样率为51.2kHz。表2提供了全面的数据集描述。实验选择了多种数据类型,根据负载、故障大小和速度的不同而变化。
图6 实验设备
表2 数据集描述
备注:"内圈"表示内圈故障,"滚子"代表滚子故障。
4.2 故障诊断的实验验证
为了构建一个不平衡比例为10:1的实验场景,初始不平衡数据集A包含一个训练集,该训练集由200个正常状态样本和每个故障状态各12个样本组成,每个样本包含1024个数据点。每个故障状态的12个样本通过重叠采样扩展至20个样本。此外,测试集的样本量是训练集的一半。数据集B至G分别对应通过使用VAE、WGAN-GP、VAEGAN、DFEGAN、WCGAN和TMCGAN的增强技术获得的平衡数据集。关于这些数据集的详细信息见表3。
表3 数据集的构建
作为TMCGAN的候选分类器,采用了CNN、SVM和SAE。SVM使用了RBF核,该核函数的半径为0.2,惩罚因子为50。对于SAE,学习率设为0.1,编程了120次迭代。此外,稀疏参数设置为0.3,惩罚因子值为5。CNN的配置包括三个卷积层、三个池化层和两个全连接层。卷积核大小、步长和池化层的选择作为参数选择的基础进行了实验。这些参数实验的结果详见表4。最终参数的详细描述见表5。
表4 CNN参数选择测试
表5 CNN分类器的参数设置
备注:在该架构中,“C”代表卷积层,其中初始参数指定滤波器数量,随后参数指定卷积核尺寸,最后参数指定激活函数。“M”代表最大池化层,使用尺寸为2的卷积核。“Fc”表示全连接层,参数依次为偏置、权重和激活函数。此外,“R”表示ReLU函数,“S”代表Softmax函数。
除了关键分类器参数的影响外,TMCGAN的迭代次数也显著影响其诊断性能。通过在20,000到70,000次迭代范围内生成数据并构建平衡数据集,进行了诊断实验。如表6所示,从20,000次迭代开始,诊断性能持续提升,在60,000次迭代时达到最佳状态。然而,当迭代次数达到70,000次时,性能开始下降。因此,60,000次迭代被认为是最佳训练配置,这也解释了本节开头选择60,000次迭代的原因。
表6 实验的迭代选择允许进行实验
确定参数后,通过不同方法对各类数据集进行分类,并应用GAN-train技术进行诊断实验,同时进行10次不同方法分类研究的迭代,表7给出了平均准确率和标准差。
表7 诊断实验结果的统计
对于不平衡数据集A,使用了三种分类器。由于CNN的影响最大,因此在随后的数据集诊断测试中被用作分类器。在诊断实验中,TMCGAN增强的数据集实现了更优的分类性能。多领域协同对抗策略和并行频率损失的积极影响体现在标准差和准确性的提高上,证明了TMCGAN相对于现有方法的优势。
此外,引入了消融研究来评估TMCGAN中多域组件的贡献,旨在分析不同域对模型性能的影响。原始模型被指定为R,实验按时间域、时频域和频率域的影响顺序依次进行。这些消融研究的结果见表8。时间域信息的整合提高了诊断质量,并有助于增强稳定性。时频域的加入虽然没有显著提升诊断性能,但在提高模型的可解释性方面发挥了关键作用。最后,通过引入频域,构建了更高效的反馈回路,显著提升了诊断效果。这三个领域的顺序交互形成了协作对抗机制,共同增强了TMCGAN的性能。
表8 多领域协同对抗策略下的消融实验
4.3 生成数据的可信度分析
尽管研究证实了TMCGAN在故障识别中的有效性和优越性,但其优势的原因尚未得到解释。在此,通过使用几个核心的数据质量指标和可视化技术,对TMCGAN生成的数据进行了可信度分析,这有助于阐明TMCGAN在故障诊断中表现优异的确切原因。
三个关键的评估指标,KLD、MMD和GAN测试,用于评估数据的可信度。MMD指的是最大平均差异,而KLD则代表 Kullback-Leibler 散度。通过KLD评估,可以将生成的数据分布与真实数据的分布进行比较,从而对比两个概率分布。基于核函数的度量MMD用于量化真实数据分布与生成数据分布之间的距离。GAN检验利用分类结果来判断生成的数据是否来自与真实数据相同的分布[42]。GAN检验值越高,KLD和MMD值越低,表明相应生成数据的质量越高。
图7至图9展示了不同模型的指标值。尽管VAE、WGAN-GP和VAEGAN的MMD和GAN测试结果大致相似,但WGAN-GP的KLD值高于其他对比模型。在现有模型中,TMCGAN的KLD和MMD值最低,同时在GAN测试中得分最高。这表明TMCGAN生成的数据优于其他三个模型,因为它更接近原始数据。
图7 不同模型的KLD
图8 不同模型的MMD
图9 不同模型的GAN测试
数据可信度可以通过使用可视化技术来提高,除了评估指标之外。时间域和频率域的细节有助于突出生成数据与原始数据之间的差异。图10中以视觉方式展示了七种独特的实验数据类别。右侧图展示了频率域的细节,左侧图则显示了时间域的具体情况。生成的数据用红色标记,而蓝色代表原始数据。这一事实表明了原始数据和生成数据之间惊人的相似性,证实了生成数据在时间维度和频率维度上的可信度。
图10 数据域维度引导的可信度分析
t-SNE可视化解释了为什么诊断结果在数据之间的高维关系方面是合理的,而数据可视化则证明了数据本身是合理的[43]。使用t-SNE展示不同形式数据的特征差异,可以检索到七类不同数据的高维特征。图11显示了WGAN-GP、VAEGAN和TMCGAN的高维可视化结果。WGAN-GP展现了各种数据方面的相互耦合现象,这是对诊断结果影响的重要体现。VAEGAN特征耦合现象消失,但相同类型的数据特征分散仍然存在。在TMCGAN中,耦合现象消失,而相同类型的聚合效果良好。这说明TMCGAN中各种数据类型的特征可以被准确提取,同类型数据的特征可以被清晰区分,这也解释了为什么TMCGAN可以达到最佳的诊断结果,可视化解释增强了诊断结果的可信度。
图11 由t-SNE引导的高维可信度分析
多领域协作对抗方法和并行频率损失提供的高效反馈循环,使TMCGAN在跨多种数据集的特征学习方面表现出色。这使得TMCGAN能够从不同类型的资料中提取出关键的区分性特征,减少特征融合现象,从而实现更优的数据生成效果。
本节中,生成的数据本身及其在故障诊断中的应用是可信度分析的主题,解释了生成数据可靠性的背后原因。对于模型内部部分的可信度分析将在案例二中进行,以解释TMCGAN的内部工作机制。
本案例采用我团队自行采集的电力机车轴承数据集,实验设置及模型,参数与案例一中的参数相同。
5.1 数据集描述
案例二采用了电力机车采集的轴承数据。在实际场景中,噪声会对数据采集造成干扰。电力机车滚动轴承所面临的恶劣工况环境使其故障模式趋于复杂化,导致各类单一及复合故障频发。
图12显示了设备特定信息。此外,以12.8kHz的采样频率收集的数据如图所示表9。该信号经过第4节所述的处理。
一个可信的模型应当能够产生可靠的输出。首先,需要评估TMCGAN生成的数据在故障诊断中的有效性。类似于案例一,设置了一个不平衡比为10:1的实验场景。除了故障数据本身的差异外,样本数量、样本长度和采样方法与案例一相同。值得注意的是,数据集B至E是通过VAE、WGAN-GP、VAEGAN和TMCGAN增强的平衡数据集。构建的数据集详情见表10。
图12 数据采集实验台
表9 数据集描述
表10 构建数据集的说明
5.2 故障诊断的实验验证
TMCGAN的初始验证基于诊断实验的结果。TMCGAN产生的数据被加入到初始的不平衡数据集,创建一个平衡的数据集,目的是进行故障诊断研究。每个实验重复十次,以消除其他影响的可能性。结果如图13所示。
该图展示了多种方法的实验结果,使用中位数和四分位数描绘了多次实验的数据特征。每个矩形内的十个黑点分别代表每种方法的具体值。每个矩形上方的线表示上四分位数,下方的线表示下四分位数,中间的线则表示中位数。方法在垂直方向上的位置越高,诊断准确性越好。同时,水平方向上的形状越宽,诊断稳定性越高。基于数据集A的故障诊断中,CNN表现出最佳性能。因此,在后续涉及各种数据集的实验中,CNN被用作分类器。在众多研究中,TMCGAN不仅具有最低的四分位数变异,还拥有最高的诊断准确性。TMCGAN带来的诊断结果比现有技术更加可靠和准确。因此,在诊断准确性和结果波动性方面,TMCGAN生成的数据更为可信。
图13 诊断实验结果分析
混淆矩阵详细展示了诊断数据的研究情况,指出了错误分类现象发生的区域,并呈现了每种故障类型的识别结果。图14展示了混淆矩阵。尽管现有方法在处理原始数据时已取得显著进步,但在与标签1和3相关的数据中,特征纠缠问题仍然存在,导致诊断不准确。TMCGAN通过多领域协作对抗策略的全面训练和并行频率损失的有效反馈,成功解决了由特征纠缠引起的误诊问题,从而实现了更优的诊断性能。
诊断数据的评价除了准确度之外,还考虑了精确度、召回率和F1分数。分类模型的精度定义为正确预测的正类样本数与成功预测的正类样本数之比。分类模型准确预测为正类的样本占实际正类样本的比例称为召回率。衡量召回率和准确性表现的综合指标是F1分数。表11和表12展示了各种方法的结果。尽管VAEGAN和WGAN-GP在标签2中表现出轻微优势,但TMCGAN在评估整个数据集时始终表现出更优性能。这突显了多领域协作对抗策略和平行频率损失的积极影响。
图14 不同方法混淆矩阵的比较
表11 各种方法的精确度和召回率
备注:‘Pr’表示精确率,‘Re’代表召回率
表12 各种方法的F1分数
5.3 生成数据的可信度分析
借助客观指标进行评估可以增强用户对TMCGAN的信任。这也是解释TMCGAN生成数据高质量的合理原因。类似于第4节,确定TMCGAN输出可信度的第二阶段是对结果的指标评估。首先,图15显示了七个状态下的时域和频域信息分析。时域信息显示在左侧图中,而频域信息则显示在右侧图中。在这些可视化表示中,生成的数据用红色标记,原始数据则用蓝色显示。该图展示了两组数据属性的可比性,表明在电动机车发生故障时,TMCGAN的可靠性得到了保证。
图15 不同数据源的信号生成比较
此外,图16-18展示了使用三个指标KLD、MMD和GAN测试来评估数据质量。TMCGAN生成的数据质量卓越,这一点通过在GAN测试中获得最高分以及在其他两个指标中记录最低值得到了证明。定量指标表明,TMCGAN产生的输出最为可靠,且具有更优的输出效果。质量。这也可能是TMCGAN在诊断实验中表现最好的原因。
图16 KLD值的比较
图17 MMD值比较
图18 GAN测试值的比较
在对数据本身进行领域分析和度量评估之后,展示增强数据集的高维方面是另一种有用的解释TMCGAN所实现的优越诊断的技术。
因此,这里采用t-SNE进行聚类,以直观地识别不同数据组的独特特征。图19展示了TMCGAN和VAE的特征可视化。虽然VAE数据在同一类别内的特征表现出一致性,但在不同类别的特征之间缺乏明显的区分,这可能影响诊断准确性。TMCGAN出色的诊断效果可以通过其可视化来解释,该可视化显示相同的数据被很好地聚类在一起,并且可以区分出不同的故障数据特征。这表明,TMCGAN生成的数据是明显可分离的,并在高维空间中表现出一个明确定义的分类结构。
图19 诊断数据的高维特征分析
5.4 内部工作机制可信度分析
前一节对TMCGAN的生成数据进行了可信分析。TMCGAN生成的数据因其可靠性,增强了用户对实际应用的信任。然而,尽管这种方法产生了良好的结果,但由于其复杂的训练机制和不透明的设计,实际的工作过程并未详细说明。这是GAN应用面临的主要问题之一。
为了解决这一问题,TMCGAN开发了一种流跟踪因子,该因子可以贯穿整个架构,用于描述模型的内部运作。一方面,流跟踪因子阐明了TMCGAN在对抗优化过程中的训练机制,并记录了系统在整个训练过程中所做的内部决策。另一方面,流跟踪因子的决策信息在机制中从多个角度进行验证,从而阐明内部决策的正确性。本小节对相关内容进行了详细描述。
在训练迭代过程中,数据由集成到TMCGAN架构中的流跟踪因子进行跟踪。如图20和图21所示,随机选择一个原始样本,TMCGAN中的流跟踪因子记录其独特的生成过程。图中详细展示了TMCGAN在第30,000次和第50,000次迭代时的内部决策过程。这涵盖了空间、时频和频率域中的生成器、判别器和损失反馈。因此,后续阶段的优化目标以及每个样本在单次训练中的学习重点对TMCGAN架构中的各个网络都是透明的。
图20 第30000次跟踪结果
图21 第50000次跟踪结果
生成器接收样本,在学习了各个数据通道的空间域后,根据流跟踪因子揭示关键信息。重要细节被突出并量化,以便未来的学习。生成器生成的最新学习结果是提供给判别器的新数据。判别器利用流跟踪因子,专注于生成样本的时间-频率域信息,在三个高频(H1-H3)和一个低频(L)下显示每个样本区域的兴趣分布。来自两方面的信息作为损失计算的基础,在损失反馈影响判别器和生成器之前。损失反馈中的流跟踪因子定义了后续迭代阶段样本的最佳训练目标,指导TMCGAN在训练过程中不断改进。
生成器、判别器和损失反馈的内部决策通过流跟踪因子量化,在训练过程中,每个组件中重要点变化的细节可以被详细记录。这以直观的方式展示了TMCGAN的内部工作机制。例如,在30,000次和50,000次迭代中,内部决策的重要方面被清晰且易于理解地呈现出来。生成样本的质量随着每次迭代的进行而提高,表明TMCGAN的性能在不断优化。
在TMCGAN的内部工作机制中,对每个样本的训练决策信息进行了规定,该决策信息的正确性对训练过程至关重要。
从宏观训练的角度来看,通过迭代,从10,000到60,000次,产生的样本质量得到改善。生成的数据在图22中以红色显示,而原始数据则以蓝色显示。更高的健身水平可以提高代际质量。在时间域的左侧和频率域的右侧都明显存在持续上升的趋势。这一积极趋势表明,TMCGAN决策是正确的。
图22 不同迭代次数的生成质量
回到模型的基本工作原理,对抗训练的核心在于判别器。在TMCGAN中,使用多层卷积结构作为判别器。这使得可以评估反卷积结构如何影响输出的权重和激活值。然而,图15和图22所示的结果也直接证实了内部决策过程能够准确捕捉数据的本质特征。内部工作机理的正确性,下面将重点讨论数据源和故障机制方面的故障场景。
为了验证,选择了外环故障(标签3)。实验轴承参数的具体数值表明,理论上的故障特征频率为57.8赫兹。图23通过不同迭代次数下的包络谱展示了TMCGAN在不同阶段如何识别特征频率。随着训练的深入,故障特征频率逐渐显现。当训练达到60,000次时,这一频率几乎与理论值一致,此时TMCGAN已经掌握了关键的故障特征。从故障机制的角度来看,这解释了TMCGAN生成质量的稳步提升。
图23 迭代中包络谱的变化
重点关注位于模型内部结构中的判别器。从机制的角度来看,局部轴承故障机制表明,故障部件和其他部件会导致周期性的短期冲击,并使轴承系统以共振频率振动,伴随高频自由衰减。此外,最重要的故障是轴承差异化的指标是各种轴承故障类别产生的低频故障特征频率。判别器采用多层卷积与小波相结合的方法,对信号进行滤波并确定高低频,重点在于样本的时间-频率域信息。如图24所示,判别器的三层L1至L3均得以展示,逐层提取数据的高低频信息。为了展示TMCGAN对不同频率特征的响应,选择了对角高频滤波(左)和低频滤波(右)。在显示之前,数据已进行了归一化处理。在第一层中,特定细节显现出来;而在第二层和第三层中,则展示了所有信息。显然,TMCGAN学习了与故障相关的低频特征,并在逐渐挖掘深层特征时建模了与故障相关的深层低频特征。此外,TMCGAN通过学习有限数量的高频特征来获得精确的故障特征检测。在实验中,低频特征的学习趋势与故障机制相吻合。从轴承故障机制的角度来看,这也阐明了TMCGAN在特征学习选择上的合理性和决策正确性。

图24 判别器的逐层频率特征响应
综上所述,流跟踪因子的设计通过揭示TMCGAN的内部工作机制增强了模型的透明度。通过透明的训练机制,内部决策变得显而易见,在映射和解释决策正确性时,样本、样本特征和故障机制均被考虑在内。这也阐明了TMCGAN卓越性能的具体原因。
本文提出了一种解释GAN内部工作机制的方法——TMCGAN,该方法有效提高了模型的透明度,并增强了生成数据和诊断结果的可信度。模型由三个关键组件构成。首先,多域协同对抗策略顺序捕捉了多域的关键特征信息,形成了一个有机融合的对抗系统,为GAN训练提供了更丰富的信息。其次,平行频率损失引入了多维频率细节信息,加强了多域对抗系统,有效提升了对抗训练闭环的效果。最后,流跟踪因子解释了GAN的内部工作机制,揭示了动态关注点和实时决策方向。
此外,许多场景已被用于验证TMCGAN在不平衡数据旋转机械故障诊断中的优越性和有效性。从数据可信度和模型可信度的角度出发,通过涵盖评估指标、可视化分析和机制分析来解释模型内部决策的原因。这实现了全局可解释生成和可信分类。目前,TMCGAN在生成器、判别器、损失反馈和结果分析等主要领域已达到可解释性,其当前范围足够广泛。然而,GAN的解释深度仍需增加。然而,作者的下一步研究将集中在创建能够深入解释的透明GAN,这将增强用户信任并开启更广泛的工程应用可能性。