基于自注意力变形网络的磁暴预测方法、装置及存储介质

摘要

本发明公开了基于自注意力变形网络的磁暴预测方法、装置及存储介质，其中方法包括输入卫星磁测数据和太阳高能粒子数据；提取卫星磁测数据的第一特征；提取太阳高能粒子数据的第二特征；基于第一特征和第二特征进行分类得到地磁指数；第二特征提取模块包括设有多个编码子层的编码器和设有多个解码子层的解码器，每个编码子层包括第一多头自注意力结构和第一全连接前向网络；每个解码子层包括多头自注意力结构、第二多头自注意力结构和第二全连接前向网络处理。能实现准确的磁暴预测；能并行计算，减少计算复杂度，提高计算效率；使模型更具可解释性。