[发明专利]基于交叉注意力和自注意力的分类参数分布的故障诊断方法

摘要

本发明公开了基于交叉注意力和自注意力的分类参数分布的故障诊断方法，具体步骤为：S1、对于支持集中的每一个样本x，按照序列关系计算位置权重；S2、计算支持集中的每一个样本x加权位置权重后表示；S3、对x′计算对应的多头自注意力机制对应的Q1、K1、V1；S4、计算基于Q1、K1、V1的注意力；S5、重复步骤4和步骤5；S6、对测试集样本加权位置权重后表示；S7、计算对应的类别预测值。本发明利用多头交叉注意力机制，将支持集中数据作为上下文知识，对测试数据进行知识补充，训练分类器的参数分布，可以有效克服故障数据少带来的问题；并且在解决序列数据问题的同时，兼顾支持集带来的上下文知识，解决小样本故障数据的问题。