基于Swin Transformer的太阳射电频谱分类
天体物理仪器与方法
2025-02-07 v1 太阳与恒星天体物理
摘要
太阳射电观测是一种研究太阳的方法。实时自动分类太阳射电频谱并判断是否存在太阳射电暴,对于空间天气预警和太阳物理研究至关重要。由于太阳射电暴频谱数量少且分布不均,本文提出了一种基于Swin Transformer的太阳射电频谱分类方法。首先,该方法将预训练模型的参数迁移到Swin Transformer模型中。然后,冻结Swin Transformer的隐藏层权重,并在目标数据集上训练Swin Transformer的全连接层。最后,进行参数微调。实验结果表明,该方法能够达到100%的真阳性率,比以往的方法更准确。此外,我们的模型参数量仅为2000万,比传统VGG16卷积神经网络超过1.3亿的参数量降低了80%。
引用
@article{arxiv.2502.03782,
title = {Classification of Solar Radio Spectrum Based on Swin Transformer},
author = {Jian Chen and Guowu Yuan and Hao Zhou and Chengming Tan and Lei Yang and Siqi Li},
journal= {arXiv preprint arXiv:2502.03782},
year = {2025}
}