一种基于多头注意力网络的短波协议识别方法、装置、设备及存储介质

本申请公开了一种基于多头注意力网络的短波协议识别方法，包括：S1获取用于训练的短波协议数据，利用用于训练的短波协议数据训练基于多头注意力网络的分类模型；S2获取待识别的短波协议数据，将数据经过数据预处理得到标准格式的数据；S3将标准格式的数据按预设的间隔进行单元化，生成单元化序列；S4将单元化序列输入训练好的基于多头注意力网络的分类模型，获取输出序列；S5根据输出序列中的类别单元的值获取待识别的短波协议数据分为各类协议的概率，其中概率最高的对应类别即为待识别数据的类别。本发明还公开了一种基于多头注意力网络的短波协议识别装置、相应的设备及存储介质。本发明能进一步提升短波协议类型识别正确率。