一种面向列车传动系统的类脑持续学习故障诊断方法

技术领域

本发明属于轨道交通安全保障领域，具体涉及一种面向列车传动系统的类脑持续学习故障诊断方法。

背景技术

轨道车辆传动系统主要包括牵引电机、齿轮箱、轮对轴箱等，起到传递牵引力实现车辆行进的重要作用。由于列车运行环境复杂多变，传动系统不可避免地发生各类故障，给轨道车辆运行带来安全风险。当前列车传动系统的检修通常以里程为依据，达到一定里程数后，对各部件进行维修和更换，这种基于经验的“定期修”往往伴随着过维修和欠维修的问题。随着多源全息感知和智能物联技术的发展，基于深度学习的智能故障诊断技术通过对监测数据的自动辨识实时发现故障，在轨道交通安全保障和智能运维领域受到青睐，引发了从传统的“定期修”到费效比更高的“按需修”的变革。然而，现有的基于深度学习的智能诊断模型在训练完成后，可诊断的故障类型便被固定，如果想增加可诊断故障类别，则需要重新训练模型。轨道车辆传动系统的部件繁多、型号各异，很难预先得到全部潜在故障类型的监测数据，而受制于计算和存储资源、数据传输等限制重新训练模型在很多工程场景下无法实现，限制了智能诊断模型的应用潜力。目前，针对列车传动系统故障诊断问题，缺乏一种类脑持续学习范式，使诊断模型部署后，面对非独立同分布的新故障数据集时，无须从头训练，通过模仿人脑神经突触的学习与记忆机制，在保持既有故障诊断能力的同时，拓展可诊断故障类型的边界。

发明内容

为了克服现有技术的局限，本发明的目的在于提出一种面向列车传动系统的类脑持续学习故障诊断方法，诊断模型部署后可以在不依赖人为重新训练的情况下，对渐进式获取的新类别故障数据集进行持续学习，从而拓展可诊断故障类型的边界。该方法具有如下优点：1)无须存储不断膨胀的故障数据，仅利用少量额外存储空间保存典型旧类别故障数据就可以有效缓解因新知识与旧知识分布不同而造成的“灾难性遗忘”问题；2)模型随着类脑持续学习进入新阶段而不断生成新的类突触表征结构，在多目标优化指导下，专注学习新类特征，面对故障特征模态多变的列车传动系统故障诊断问题时，不会因欠参数而引发诊断性能退化；3)在模型新表征结构生成过程中，嵌入了与学习同步的剪枝，可根据不同阶段学习任务的难度自适应地调节表征结构规模，避免结构化冗余的加剧。

为了达到上述目的，本发明采取的技术方案为：

一种面向列车传动系统的类脑持续学习故障诊断方法，包括以下步骤：

1)设当前持续学习阶段s＝0，获取既有的轨道车辆传动系统故障数据集

构造智能诊断模型F

设置损失函数

其中，l()为指示函数，y

优化流程如下：

i.输入

ii.初始化θ

iii.利用反向传播算法训练模型，基于下式将θ

iv.重复iii直到训练代数达到t；

v.得到训练完成的模型F

优化完成后，从

2)设当前持续学习阶段s＝1，获取新一阶段的列车走行系统故障数据集

构造智能诊断模型F

采用训练同步的动态剪枝策略，具体地：

2.1)获取B

其中，Y

2.2)在各批归一化层内，基于P

sc

2.3)根据采样得到的sc

2.4)将m

其中，Z

设置损失函数

如下所示：

其中，

优化流程如下：

i.输入

ii.初始化θ

iii.如果当前训练代数大于20且可被10整除，则执行iv-vii，否则执行vi-vii

iv.执行2.1)-2.4)剪枝过程

v.剪枝阈值根据序列τ＝{τ

vi.利用反向传播算法训练模型，具体地，根据下式更新θ

vii.跳回至iii直到训练代数达到t；

viii.得到训练完成的模型F

优化完成后，从

本发明的有益效果为：

本发明能够在诊断模型部署后对序列形式的新故障数据集进行类脑持续学习，从而不断拓展其可诊断故障的范围，并且具备如下优点，1)无须基于全历史数据集对模型进行重训练，利用少量旧类别示例配合既有模型结构的冻结就可以有效缓解新知识与旧知识分布不同造成的“灾难性遗忘”问题；2)模型随着类脑持续学习进入新阶段而不断生成新的类突触表征结构，在多目标优化指导下，可以高效学习新类特征，不会因特征模态多样造成欠参数问题继而引发诊断性能退化；3)在模型新表征结构生成过程中，嵌入了与学习同步的剪枝，可根据不同阶段学习任务的难度自适应地调节表征结构规模，避免模型的超冗余。

附图说明

图1为本发明方法工作流程图。

图2为类脑持续学习智能诊断原理示意图。

图3为动态剪枝工作原理图。

图4为本发明方法同现有方法的持续学习性能对比图。

具体实施方式

下面结合附图对本发明做进一步详细阐述。

参照图1、图2和图3，一种面向列车传动系统的类脑持续学习故障诊断方法，包括以下步骤：

1)设当前持续学习阶段s＝0。获取既有的轨道车辆传动系统故障数据集

其中，

其中，l()为指示函数，y

优化流程如下：

i.输入

ii.初始化θ

iii.利用反向传播算法训练模型，具体地，基于下式更新θ

iv.重复iii直到训练代数达到t；

v.得到训练完成的模型F

优化完成后，从

2)设当前持续学习阶段s＝1。获取新一阶段的列车走行系统故障数据集

2.1)获取B

其中，γ

2.2)在各批归一化层内，基于P

sc

2.3)根据采样得到的sc

2.4)将m

其中，Z

需要说明的是，模型训练的初始阶段，参数更新幅度较大，剪枝往往表现出剧烈波动。如果将2.1)-2.4)直接嵌入到模型训练中，则可能剪除有潜力的通道。为此，建议上述剪枝流程在训练的第20代之后引入。除此之外，剪枝带来模型结构的变化，造成训练优化的不平滑，因此，每10个epoch进行一次剪枝。随着代数的增加，τ应以渐进的方式减少。当剪枝后10个epoch的诊断精度如果明显下降，则将网络参数回溯到剪枝之前，并保持上次剪枝的模型结构继续训练，停止进一步的剪枝。

设置损失函数

如下所示：

其中，

优化流程如下：

i.输入

ii.初始化θ

iii.如果当前训练代数大于20且可被10整除，则执行iv-vii，否则执行vi-vii

iv.执行2.1)-2.4)剪枝过程

v.剪枝阈值根据序列τ＝{τ

vi.利用反向传播算法训练模型，具体地，根据下式更新θ

vii.跳回至iii直到训练代数达到Tviii.得到训练完成的模型F

优化完成后，从

基于上述方法，诊断模型则可以以高诊断精度和低冗余度不断拓展可诊断故障的类型，并且不会发生明显的“灾难性遗忘”。

实施例

以轨道车辆传动系统的牵引电机故障诊断为案例，基于牵引电机模拟实验台数据，对本发明方法的有效性进行了验证。本实施例所采用的实验台由利用变频器控制交流电机的转速，通过电涡流制动器对电机施加扭矩，以此模拟牵引电机的各种工况。在电机驱动端和非驱动端分别安装了三向加速度传感器，在电源线处安装了电流钳采集电机工作电压，并在电机旁安装了声传感器，共采集10通道传感数据。实验中考虑了11种电机健康状态，包括正常状态、4种轴承故障(内圈故障、外圈故障、滚动体故障和保持架故障)、3种转子故障(转子不平衡、转子弯曲和转子断条)和3种电气故障(电机缺相、电压不平衡和短路)。每种健康状态包括1000个样本，每个样本包括1024个采样点。在此基础上，根据轴承故障、转子故障和电气故障将数据集划分为三个阶段，每阶段数据按照1：1比例用于训练和测试。对于每阶段的训练集，包括之前阶段训练数据采样得到的少量示例原本和当前阶段训练数据，对于测试集，包括之前阶段和当前阶段的全部测试数据。实验对比了本发明提出的方法与全网络参数微调和分类器参数微调两种在线学习常用方法，网络基本结构见表1，训练相关参数见表2。实验结果被汇总于表3及图4。通过实验结果可以看出，本发明方法随着持续学习阶段的增加，诊断效果显著优于两种对比方法。这得益于诊断模型的生长拓展机制有效缓解了“灾难性遗忘”问题。此外，该机制还为模型注入了可塑性，提升了模型学习能力上限，从而对新故障的诊断也有更好的效果。上述实验结果证明了方法的优越性。

表1.网络基本结构汇总表

注：对比方法表征学习部分不拓展，全网络参数微调基于前一阶段参数对E和CL的全部可训练参数更新，而分类器微调则仅更新CL的参数。

表2.训练相关参数汇总表

表3.持续学习阶段s＝2准确率汇总

/>