一种基于改进DLKA与Transformer的小样本轴承故障诊断方法

本发明公开了一种基于改进DLKA与Transformer的小样本轴承故障诊断方法，涉及轴承故障诊断技术领域。该方法包括：获取滚动轴承故障诊断数据集；构建基于可变形局部核注意力模块DLKA和Transformer的小样本轴承故障模型，并利用所述数据集对所述小样本轴承故障模型进行训练；所述小样本轴承故障模型包括：多尺度可变形大核特征提取模块MDLKFE、改进的全局Transformer模块、多尺度马氏距离模块和分类器；利用训练好的小样本轴承故障模型进行小样本轴承故障诊断。本发明中的方法在有限数据条件下展现出卓越的性能表现。