集成注意力机制的旋涡星系旋臂数量识别方法
An Attention Mechanism-based Method for Identifying the Number of Spiral Arms in Spiral Galaxies
-
摘要: 利用深度学习方法精准识别旋涡星系的旋臂数量, 对追溯星系的演化轨迹, 理解旋涡结构在不同宇宙时期的演变规律具有重要作用. 然而, 当前深度学习模型并未充分考虑旋涡星系图像的特点, 尤其是旋涡中心和旋臂亮度的分布特征. 为此, 提出基于亮度的双向融合曼哈顿注意力机制, 并将该注意力机制与EfficentNet等多个网络模型结合, 构建了集成注意力机制的旋涡星系旋臂数量识别方法. 以Galaxy Zoo 2数据集中筛选的旋涡星系图像为例, 进行了实验对比分析. 结果表明, 增加注意力机制后, EfficientNet模型在测试数据集的总体准确率可达95.52%, 尤其是对数量为3、4及以上的旋臂识别效果更好. 研究表明, 良好的注意力机制设计能够挖掘旋涡星系的亮度分布特征, 并能提升旋臂数量识别的效果.Abstract: The precise identification of the number of spiral arms in spiral galaxies using deep learning methods plays an important role in tracing the evolution trajectory of galaxies and understanding the evolution laws of spiral structures in different cosmic periods. However, the features of spiral galaxy images, particularly the distribution characteristics of spiral center and arm brightness, have not been properly taken into account by existing deep learning models. Therefore, this article proposes a brightness-based bidirectional fusion Manhattan attention mechanism, and combines this attention mechanism with the EfficentNet model for identifying the number of spiral arms in spiral galaxies. Experimental study was carried out using the spiral galaxy images chosen from the Galaxy Zoo 2 dataset as an example. The results show that the overall accuracy of the EfficientNet model on the test dataset can reach 95.52% by integrating an attention mechanism, especially good for the recognition of 3, 4, and more spiral arms. Research has shown that well-designed attention mechanisms can reveal the brightness distribution characteristics of spiral galaxies and improve the effectiveness of identifying the number of spiral arms.
下载: