Feb, 2024

基于贝叶斯深度学习方法和不确定性量化的 SYM-H 指数预测

TL;DR我们提出了一种新颖的深度学习框架,名为 SYMHnet,利用图神经网络和双向长短期记忆网络协同学习来自太阳风和行星际磁场参数的模式,用于基于 1 分钟和 5 分钟的分辨率数据对 SYM-H 指数的短期预测。SYMHnet 以 NASA 的空间科学数据协调存档提供的参数值的时间序列作为输入,并预测给定时间点 t 的 t + w 小时的 SYM-H 指数值,其中 w 为 1 或 2。通过将贝叶斯推断纳入学习框架中,SYMHnet 可以在预测未来 SYM-H 指数时量化数据不确定性和模型不确定性。实验结果显示,SYMHnet 在安静时段和风暴时段,以及 1 分钟和 5 分钟的分辨率数据上表现良好。结果还显示,与相关的机器学习方法相比,SYMHnet 通常表现更好。例如,当使用 5 分钟分辨率数据在大风暴 (SYM-H = -393 nT) 中预测 SYM-H 指数 (提前 1 小时) 时,SYMHnet 的预测技能得分 (FSS) 为 0.343,而最近的梯度提升机 (GBM) 方法的 FSS 为 0.074。当预测大风暴中的 SYM-H 指数 (提前 2 小时) 时,SYMHnet 的 FSS 为 0.553,而 GBM 方法的 FSS 为 0.087。此外,SYMHnet 可以提供数据和模型不确定性的结果量化,而相关方法则不能。