[发明专利]不平衡小样本数据的滚动轴承故障诊断方法及系统

摘要

本发明公开了一种不平衡小样本数据的滚动轴承故障诊断方法及系统，包括：获取滚动轴承的声纹信号数据；对采集到的数据进行降维、位置编码和填充处理；基于预训练的故障诊断模型以及所述滚动轴承的声纹信号数据，进行故障诊断，得到滚动轴承的故障类型；其中，所述故障诊断模型由预训练好的基于多头自注意力机制的自“编码‑解码”模型中的编码部分与多头自注意力机制分类器搭建。本发明通过基于多头自注意力机制的自“编码‑解码”模型实现了模型预训练，解决了将多头自注意力机制应用到多模态任务数据需求问题，并有效提升了模型训练效率与故障诊断准确率。