用双向LSTM分析天文瞬变光变曲线,提升分类准确率。
Classification of Transient Astronomical Object Light Curves Using LSTM Neural Networks
- 采用双向LSTM结合掩码层处理光变曲线序列
- 对周期类和S型类识别效果好,AUC达0.99和0.95
- 在数据量少或类别不平衡时表现下降,适合天体分类研究者
本研究提出一种双向长短期记忆(LSTM)神经网络,用于分类来自光电LSST天文时间序列分类挑战(PLAsTiCC)数据集的瞬变天体光变曲线。原始14类被合并为5个泛化类别(S-Like、Fast、Long、Periodic、Non-Periodic)以缓解类别不平衡问题。经填充、时间重缩放和通量归一化预处理后,使用带有掩码层的双向LSTM在19,920个对象的测试集上训练与评估。模型对S-Like和周期类表现优异,ROC-AUC分别为0.95和0.99,精确率-召回率AUC分别为0.98和0.89。但对Fast和Long类表现较差(长类ROC-AUC仅0.68),且难以区分周期与非周期类。在仅使用检测后5、10、20天的数据下,性能显著下降,误判倾向指向S-Like类。结果表明,类别不平衡和时间信息不足是主要瓶颈,建议采用类别平衡策略及聚焦探测时刻的预处理方法提升性能。
原文摘要 · Abstract (English)
This study presents a bidirectional Long Short-Term Memory (LSTM) neural network for classifying transient astronomical object light curves from the Photometric LSST Astronomical Time-series Classification Challenge (PLAsTiCC) dataset. The original fourteen object classes were reorganized into five generalized categories (S-Like, Fast, Long, Periodic, and Non-Periodic) to address class imbalance. After preprocessing with padding, temporal rescaling, and flux normalization, a bidirectional LSTM network with masking layers was trained and evaluated on a test set of 19,920 objects. The model achieved strong performance for S-Like and Periodic classes, with ROC area under the curve (AUC) values of 0.95 and 0.99, and Precision-Recall AUC values of 0.98 and 0.89, respectively. However, performance was significantly lower for Fast and Long classes (ROC AUC of 0.68 for Long class), and the model exhibited difficulty distinguishing between Periodic and Non-Periodic objects. Evaluation on partial light curve data (5, 10,and 20 days from detection) revealed substantial performance degradation, with increased misclassification toward the S-Like class. These findings indicate that class imbalance and limited temporal information are primary limitations, suggesting that class balancing strategies and preprocessing techniques focusing on detection moments could improve performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。