摘要:针对电机长尾数据故障诊断中尾部类别特征学习不足和头尾部梯度竞争失衡的问题, 提出一种融合余弦退火热重启衰减学习率策略(CDLR)的时空消息传递神经网络(STMPNN)电机长尾故障诊断模型(STMPNN-CDLR). 首先, 通过多节点拓扑结构建模多变量时间序列数据样本同一时间窗中不同时间阶段(子时间窗)传感器间的时空依赖关系, 并设计动态时空关系加权矩阵刻画传感器特征在时间维度上的演化模式, 强化头尾部类别潜在时空交互的特征表示; 其次, 利用STMPNN的消息传递机制实现跨子时间窗的特征更新, 提升模型对局部和全局信息的感知能力; 最后, 通过CDLR策略周期性地重启和衰减学习率, 缓解长尾分布导致的梯度竞争失衡问题, 增强模型稳定性和对尾部类别的敏感性. 在4组不同长尾比率的电机故障诊断实验中, 所提出的方法在不牺牲头部正常类别诊断性能的前提下, 对尾部故障类别展现出优异的诊断性能和稳定性, Accuracy高于94.57%, 验证了该方法在解决电机长尾故障诊断问题中的有效性和优越性.