基于注意力机制和不确定性损失的太阳磁图超分
Super-Resolution of Solar Magnetograms Based on Uncertainty Loss and Attention Mechanism
-
摘要: 为了全面了解太阳活动规律, 需要连续观测时长覆盖多个太阳活动周的、高质量数据. 太阳磁图是研究太阳活动的重要数据, 连续、长时间和高空间分辨率的太阳磁图能够提供更精细的太阳磁场演化信息, 有助于更准确地预报太阳活动和空间天气事件. 因此, 提出一种基于深度学习的超分率算法, 对MDI (Michelson Doppler Imager)磁图进行超分, 取得与日震和磁成像仪(Helioseismic and Magnetic Imager, HMI)磁图一致的分辨率, 从而能够获得持续时长将近两个太阳活动周的高质量太阳磁图数据库. 为了引导网络学习磁图中有效的特征信息, 将注意力机制引入到网络中, 学习注意力权重图. 此外, 采用了不确定性损失作为模型训练的损失函数, 该方法能够对于带有磁场变化的纹理和边缘分配更大的权重, 同时不增加网络参数和计算量. 实验证明, 提出的算法显著提高了超分太阳磁图的质量, 在峰值信噪比(Peak Signal-to-Noise Ratio, PSNR: 33.3168)、结构相似性(Structure Similarity Index Measure, SSIM: 0.8754)、相关性(Correlation Coefficient, CC: 0.9323)和均方根误差(Root Mean Square Error, RMSE: 21.8544)等指标上取得了最优的结果.Abstract: To comprehensively understand the regularities of solar activity, it is necessary to continuously collect high-quality data covering multiple solar activities cycles over a long period of time. Solar magnetograms are the most important data for studying solar activity. Continuous, long-term, and high spatial resolution solar magnetograms can provide more detailed information on solar magnetic field evolution, which is helpful for more accurate prediction of solar activity and space weather events. Therefore, this paper proposes a deep learning-based super-resolution algorithm to perform super-resolution on MDI (Michelson Doppler Imager) magnetograms to achieve the same resolution as HMI (Helioseismic and Magnetic Imager) magnetograms, thus obtaining a high-quality solar magnetogram database covering nearly two solar activity cycles. To improve the performance of the model, this paper introduces an attention mechanism in the deep neural network, which effectively guides the network to enhance effective feature information by learning attention weight maps in the channel and spatial dimensions. In addition, this paper uses uncertainty loss as the loss function for model training, which can assign greater weight to textures and edges with magnetic field changes, without increasing the network parameters and computational complexity. The experimental results show that the proposed algorithm significantly improves the quality of super-resolution solar magnetograms and achieves the best results in peak signal-to-noise ratio (PSNR: 33.3168), Structure Similarity Index Measure (SSIM: 0.8754), correlation coefficient (CC: 0.9323), root mean square error (RMSE: 21.8544) and other indicators.