An Interpretable LSTM Network for Solar Flare Prediction

Gautam Varma Datla,Haodi Jiang,Jason T. L. Wang

2023 IEEE 35TH INTERNATIONAL CONFERENCE ON TOOLS WITH ARTIFICIAL INTELLIGENCE, ICTAI(2023)

引用 0|浏览3
暂无评分
摘要
Deep learning models are often considered black box models as their internal workings tend to be opaque to the user. Because of this lack of transparency, it is challenging to understand the reasoning behind the model's predictions. Here, we present an approach to making a solar flare prediction model interpretable. This model, built based on a long short-term memory (LSTM) network with an attention mechanism, aims to predict whether an active region (AR) on the Sun's surface would produce a large flare, namely an M- or X-class flare, within 24 hours. The flare events used in this study are collected from the Geostationary Operational Environmental Satellite X-ray flare catalogs provided by the National Centers for Environmental Information. The crux of our approach is to model data samples in an AR as time series and use the LSTM network to capture the temporal dynamics of the data samples. Each data sample has 22 features including magnetic parameters and flare history parameters. To make the model's predictions accountable and reliable, we leverage post hoc model-agnostic techniques, which help elucidate the factors contributing to the predicted output for an input sequence and provide insights into the model's behavior across multiple sequences within an AR. To our knowledge, this is the first time that interpretability has been added to an LSTM-based flare prediction model.
更多
查看译文
关键词
Interpretable deep learning,Solar flares,LSTM
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要