本期给大家分享微 信社群大佬写的1区top代码开源文章。如果有故障诊断相关方向研究人员希望宣传自己研究成果,或者希望加入微 信社群,近距离和各高校大牛交流,欢迎大家在公众 号后台与小编联系,大家一起交流学习。
铁路道岔转辙机是保障列车进路安全的关键设备,其故障诊断是当前铁路智能运维领域的研究热点。本期推荐的这篇文章来自哈尔滨工业大学与北京交通大学/清华大学/华中科技大学团队,聚焦 ZDJ9 转辙机在跨类别小样本场景下的故障诊断问题,提出了“动态图元学习 + 多传感器信息融合”的新型诊断框架。文章系统构建了基于类别迁移的元学习策略,实现对 训练阶段未出现故障类型 的快速适应;同时设计了图结构多传感器信息融合网络 GMSIFN,显式 建模减速器、动作杆、自动开闭器等关键部位之间的 空间依赖与故障传播路径。在试验线与实际运营线路两套九通道振动数据上的 12 个跨类别小样本任务中,该方法在诊断精度、收敛速度和噪声鲁棒性方面均显著优于多种图学习、元学习、小样本与持续学习基线。其核心价值在于 打通小样本故障诊断、新故障模式快速适应与多传感器空间建模之间的空白,为安全关键铁路基础设施构建可部署的智能运维与诊断系统提供了有力支撑。
论文题目: Dynamic Graph Meta-Learning with Multi-Sensor Spatial Dependencies for Cross-Category Small-Sample Fault Diagnosis in ZDJ9-RTAs
论文期刊:Advanced Engineering Informatics
Doi:https://doi.org/10.1016/j.aei.2025.104132
1 问题背景
2 整体诊断框架
2.1 跨类别元学习训练策略
2.2 基于图的多传感器信息融合网络
3 实验
3.1 铁路试验线数据验证
3.2 铁路现场数据验证
4 结论
围绕这一现实瓶颈,近年已有工作开始探索RTA小样本故障诊断。代表性研究通常基于度量学习或元学习框架,引入原型网络、半监督元学习、注意力机制等,以在有限样本下提高特征区分度和诊断精度;部分工作也尝试将多源信号进行数据或特征层融合,以缓解单一信号信息不足的问题。然而,这些方法大多仍然建立在训练类与测试类同质的假设之上,即训练阶段和测试阶段共享相同的故障类型集 合,只是在样本数量上体现“小样本”。这与实际RTA运维场景存在明显偏差:随着设备老化、环境扰动和检修策略变化,新型故障模式会不断涌现,运维系统往往需要在仅有极少量新故障样本的条件下,迅速完成对未见类别的诊断与区分;而现有方法缺乏针对“跨类别小样本适应”的显式建模,很难在类别集 合发生变化时保持稳定可靠的性能。另一方面,已发表的RTA小样本研究多仍以单传感器信号为主要数据源,或者将多源信号简单做串接/加权融合,未显式建模各关键结构部位(减速箱、尖轨/传动杆、转辙控制电路等)之间的空间拓扑关系与故障传播路径,因而难以在复杂机械耦合环境下充分挖掘多传感器之间的互补信息和传播机理。
需要特别强调的是,虽然持续学习(Continual Learning)在形式上同样涉及“新增类别”的学习,但其目标侧重点与本文场景并不一致。典型持续学习方法(如EWC、CEC)旨在实现“在不断引入新任务/新类别的同时尽量保持对所有历史类别的识别能力”,核心是缓解灾难性遗忘,因此会通过参数正则化、重要权重冻结或记忆重放等机制显式约束模型更新,这在本质上牺牲了部分对新类别的优化自由度,以换取旧类别性能的稳定。而在RTA运维实践中,我们关注的关键问题是:当现场首次出现此前从未记录过的新故障类型时,如何在极小样本条件下尽快获得对这些新故障的高置信度诊断结果,以支撑运维人员的快速处置;至于旧故障类型,往往已积累了充足的历史经验与规则库,可以通过离线模型重训或规则融合进行弥补。因此,在这一“新类快速适应优先,而非长期记忆平衡”的任务设定下,引入持续学习框架不仅收益有限,反而可能由于对旧参数的过度约束而拖慢新故障类别的适应速度、降低最终判别上限,并不契合本文所要解决的核心工程需求。
整体而言,本文提出的诊断框架面向多传感器ZDJ9-RTA的跨类别小样本故障诊断,由“跨类别元训练–图结构多传感器建模–小样本快速自适应诊断”三部分有机构成。首先,在数据层面从减速箱、传动杆和转辙控制器等关键部位同步采集九通道振动信号,并将完整故障类别空间严格划分为互不重叠的元训练类集和元测试类集,从而构建训练与测试故障类型不相交的跨类别小样本诊断场景。其次,在训练阶段基于MAML框架引入类别迁移元学习策略:在已知故障类空间内动态构造由少量样本组成且类别组合不断变化的元任务,借助“支持集内循环更新+查询集外循环优化”的双层优化过程,学习得到对类别分布变化高度敏感、可快速迁移至未见故障类别的参数初始化。同时,针对每个元任务中的多传感器时序信号,采用所提出的GMSIFN,通过无监督图构建模块根据信号相似度自适应推断传感器拓扑关系,并在GRU门控注意力机制支持下完成节点特征聚合与图级表征学习,从而显式刻画传感器间的空间依赖与故障传播路径。最后,在测试与实际应用阶段,对仅包含极少量新故障样本的小规模适应集进行快速微调,以元学习得到的初始化为起点更新GMSIFN参数,随后在包含全部剩余样本的大规模评估集上执行图级故障识别,实现从多传感器数据采集到跨类别小样本精准诊断的端到端闭环流程。
2.1 跨类别元学习训练策略

图2 跨类别元学习训练策略
满足ZDJ9-RTA在现实运维场景中“新故障不断涌现、小样本快速诊断”的需求,本文在MAML框架基础上,构建了类别迁移元学习训练策略。其核心思想是在训练阶段主动制造“任务间类别不断变化”的学习环境,使模型在已知故障类别上形成一种面向“类别分布变化”的泛化能力,从而在测试阶段能够利用极少量新故障样本实现快速跨类别适应。 具体而言,在已知故障类集 合 内,本文通过随机采样构造 个元任务 ,每个任务从 中选取 个类别,并为每个类别采集 个样本,将其划分为用于内循环更新的支持集 与用于外循环优化的查询集 。由于每一轮任务均重新随机采样类别,任务间类别交集 遵从参数为 的二项分布,从而在训练过程中形成显式的“类别迁移”过程。 在优化层面,类别迁移元学习采取双层梯度更新机制。对给定任务 ,首先在支持集上进行若干步内循环梯度下降,得到该任务的特定参数:
其中 为当前共享初始化参数, 为内循环学习率, 为基于支持集的交叉熵损失。随后,用 在查询集 上计算损失,并在外循环中对初始化参数 进行更新:
其中 为外循环学习率。通过在大量类别组合各异的任务上反复执行上述内外循环优化,模型逐步获得一组对类别分布变化敏感、且易于在少量样本下快速调整的元初始化参数 。 在测试阶段,针对训练阶段从未出现过的新故障类别集 合 ,本文遵循“小样本快速微调+大样本全面评估”的流程。 首先在新故障的小规模适应集 上,以元初始化 为起点,执行一次快速梯度更新得到适应后参数:
然后利用 在包含所有剩余测试样本的大规模评估集上进行故障识别,并以经验风险衡量跨类别小样本诊断性能。通过这种训练–测试一致的“类别迁移+梯度快速适应”机制,模型在不接触真实新故障的前提下,就已在训练阶段习得“遇到新类别时如何快速调整”的元知识,从而显著提升在实际新故障情形下的小样本诊断能力。
2.2 基于图的多传感器信息融合网络

图3 基于图的多传感器信息融合网络框架
针对传统小样本RTA故障诊断方法普遍依赖单传感器、难以刻画多传感器空间依赖的局限,本文提出GMSIFN。该网络以“传感器为节点、信号相关性为边”,通过端到端的图构建与图级分类过程,将九通道振动信号中的空间结构信息显式引入故障判别。 在输入层,GMSIFN首先对多通道时间序列 进行可学习的线性压缩,将高维时序信息映射到低维传感器特征空间: , 其中 为压缩后的节点特征矩阵, 、 为可训练参数。该变换在保留关键时序模式的同时,显著降低后续图构建的计算复杂度。 随后,GMSIFN在压缩特征空间中通过无监督方式自动推断各传感器节点间的拓扑关系。对于任意两个传感器节点 , ,以特征内积衡量其信号相似度,并引入可学习缩放因子 : , 在此基础上为每个节点选择 个相似度最大的邻居构成局部邻域,并通过Sigmoid归一化得到边权矩阵 。该过程不依赖先验几何布局或人工连边规则,而是完全由多传感器振动信号自身的统计关联性驱动,从而能够自适应地刻画减速箱壳体、转辙杆、转辙控制器之间的动力学耦合关系。 在节点特征聚合阶段,GMSIFN利用“邻居特征 + 边权 + 注意机制 + GRU门控”的组合结构实现多层消息传递:首先将节点的邻居特征与边权拼接并线性变换,以编码“邻居内容 + 连接强度”;随后将中心节点特征与邻域特征对齐,计算注意权重,对不同邻居在特征聚合中的贡献进行自适应加权,从而突出对当前故障模式最敏感的传感器通道。聚合后的上下文特征再与原始节点特征一同输入门控循环单元(GRU),通过更新门和重置门在“历史自特征”与“邻域信息”之间进行平衡,得到更新后的节点表征 。这一门控式聚合能够有效避免简单邻接平均造成的“淹没”现象,在保留关键单节点特征的同时充分利用图结构中的空间依赖。 在图级分类阶段,GMSIFN先对节点特征进行全局汇聚获得初步的图级表示 ,再将 扩展回节点维度,与局部节点特征拼接,并通过可学习注意机制为不同节点分配诊断权重,形成带“节点重要性”的全局特征 。 随后,网络再次利用GRU对初始全局特征 与注意增强后的特征 进行门控融合,得到最终图表示 ,并通过全连接+Softmax层输出故障类别后验概率:
综上,GMSIFN通过“线性压缩+无监督图构建+门控图注意聚合+图级注意分类”形成了一条从原始九通道振动信号到图级故障判别的端到端路径,不仅充分挖掘了多传感器之间的空间拓扑依赖,而且在元学习框架下可与跨类别小样本训练策略协同工作,实现对ZDJ9-RTA多种新发故障类型的高精度快速诊断。
3.1 Case I 铁路试验线数据验证
本节首先利用铁路试验线ZDJ9-RTA多传感器振动数据对所提方法进行验证。该数据集由襄阳信号通信段提供,采集对象为部署于专用试验线上的ZDJ9转辙机,在减速箱壳体、转辙杆侧和转辙控制器侧分别安装一组三轴加速度传感器,实现九通道同步振动测量,采样频率为512 Hz、单次动作周期为10 s(每通道5120个采样点)。数据集包含16种工况,来自8种典型状态的双向变体,既包括3 kN欠力、5/6/8 kN过力等力学异常,也包含“解锁故障”“接近运动后突然失去表示”等机械故障以及空载运行,共采集804条标注样本(每类40–60条),且正反向动作为不同类别编码,具有明显的方向性特征。该数据集既为图结构多传感器融合提供了典型的空间布点信息,又为跨类别小样本学习构造了丰富的“已知–新类”组合空间。
图4 铁路试验线数据采集平台
在上述数据集上,本文按照表1所示的2类、3类、4类共12个跨类别小样本任务(T1–T12)对GMSIFN及多类对比方法进行系统评估。首先通过贝叶斯优化与网格搜索联合确定压缩维度与邻居数的最优组合,图中给出了不同任务下超参数敏感性分析结果。在此基础上,图中展示了与图神经网络/元学习基线(GAT、GraphTrans、Meta-GAT、Meta-MGNN、MAML、MLDSO)、先进RTA/工业小样本方法(FDCSANet、ISONet)以及持续学习方法(CEC、EWC)的对比结果:在12个任务上,本方法在ACC、F1和AUC三个指标上整体领先,3-shot条件下平均准确率约为84%以上,在若干代表性任务中最高准确率可达99.95%(如T3),相对于最优基线平均提升可达十几个百分点。进一步地,我们定量可视化了GGL学习到的九通道邻接权重在典型工况下与机械耦合关系的一致性,t-SNE结果表明在困难的T9四类任务中,GMSIFN学习到的特征类间分离度显著高于对比方法;我们还从收敛速度和消融试验角度验证了“类别迁移元学习 + 图构建 + GRU门控聚合”这一整体框架在铁路试验线数据上的有效性和必要性。总体而言,Case I 的实验结果充分证明了所提方法在理想试验环境下具有稳定且显著的跨类别小样本故障诊断优势。
图5 Case I 实验结果可视化
3.2 Case II 铁路现场数据验证
本文构建了面向ZDJ9 转辙机的跨类别小样本故障诊断新范式,突破了传统方法在“训练/测试故障类别同构”和“单传感器数据依赖”方面的关键局限。为实现跨类别小样本诊断能力,文中提出了一套由动态类别迁移元学习策略与图结构多传感器信息融合网络(GMSIFN)构成的双创新框架:前者通过随机任务构造模拟真实工况中新故障类型的不断涌现,实现基于少量样本的快速跨类别自适应;后者基于信号相似度自适应推断传感器拓扑,并通过GRU门控图注意机制融合多传感器空间依赖关系,显式刻画跨部件的故障传播路径。基于试验线与现场两套ZDJ9-RTA实际数据,在12个跨类别小样本任务上的系统对比结果表明,所提方法在诊断精度、收敛速度以及噪声鲁棒性等方面均较多类图学习、元学习、小样本故障诊断及持续学习代表方法取得了稳定且显著的性能提升,有效弥合了故障样本稀缺与故障模式演化之间的矛盾,将小样本诊断研究拓展至动态变化的工业运维环境。
同时,当前工作仍存在两方面有待改进之处:其一,框架假定新故障以离散类别形式出现,而实际工况中故障形态往往呈连续演化特征;其二,GMSIFN 的图构建完全基于数据驱动的传感器相关性,尚未显式融入转辙机机构链与受力传递等物理约束,也尚未系统分析学习得到的图拓扑与真实故障传播机理之间的对应关系,在一定程度上限制了模型的可解释性。后续研究将依次从两方面展开:一方面引入增量式元学习,实现新故障模式在无需完整重训条件下的持续吸收与更新,增强框架对连续演化故障的处理能力;另一方面探索物理先验约束下的图构建策略,例如结合转辙机运动学/动力学链路与超图表示,将多轴振动信号拓扑与层级机械结构进行融合,以获得物理一致的图拓扑并提升故障传播路径建模的可解释性,从而进一步支撑面向安全关键铁路基础设施的终身学习诊断系统
校核:李正平、陈凯歌、曹希铭、赵学功、白亮、任超、海洋、陈宇航、陈莹洁、Tina、王金、赵诚、肖鑫鑫、张优
该文资料搜集自网络,仅用作学术分享,不做商业用途,若侵权,后台联系小编进行删除