Skip to content

rohansun/LSTM-flare-prediction

 
 

Repository files navigation

Predicting Solar Flares Using a Long Short-term Memory Network

DOI

Authors

Hao Liu, Chang Liu, Jason T. L. Wang and Haimin Wang

Contributor

Yasser Abduallah, Yixi Zhang

Abstract

We present a long short-term memory (LSTM) network for predicting whether an active region (AR) would produce a ϒ-class flare within the next 24 hr. We consider three ϒ classes, namely ≥M5.0 class, ≥M class, and ≥C class, and build three LSTM models separately, each corresponding to a ϒ class. Each LSTM model is used to make predictions of its corresponding ϒ-class flares. The essence of our approach is to model data samples in an AR as time series and use LSTMs to capture temporal information of the data samples. Each data sample has 40 features including 25 magnetic parameters obtained from the Space-weather HMI Active Region Patches and related data products as well as 15 flare history parameters. We survey the flare events that occurred from 2010 May to 2018 May, using the Geostationary Operational Environmental Satellite X-ray flare catalogs provided by the National Centers for Environmental Information (NCEI), and select flares with identified ARs in the NCEI flare catalogs. These flare events are used to build the labels (positive versus negative) of the data samples. Experimental results show that (i) using only 14–22 most important features including both flare history and magnetic parameters can achieve better performance than using all 40 features together; (ii) our LSTM network outperforms related machine-learning methods in predicting the labels of the data samples. To our knowledge, this is the first time that LSTMs have been used for solar-flare prediction.

Binder

This notebook is Binder enabled and can be run on mybinder.org by using the link below.

run_FlarePredict.ipynb (Jupyter Notebook for FlarePredict)

Binder

Please note that starting Binder might take some time to create and start the image.

For the latest updates of FlarePredict refer to https://github.com/deepsuncode/LSTM-flare-prediction

Installation on local machine

Create a virtual environment.

python -m pip virtualenv < env name >

Using Pip

  • pip install pandas
  • python -m pip install tensorflow (Refer https://www.tensorflow.org/install/pip )
  • python -c 'import tensorflow as tf ;print(tf.version__)'
  • pip install scikit-learn
  • pip install keras

Using conda

Library Version Description
pandas 1.1.5 Data analysis and manipulation tool
h5py 2.10.0 Pythonic interface to the HDF5 binary data format
keras 2.2.4 Artificial neural networks API
numpy 1.19.2 Array manipulation
scikit-learn 0.24.2 Machine learning
tensorflow 1.15.0 Neural network libraries

References

Predicting Solar Flares Using a Long Short-term Memory Network. Liu, H., Liu, C., Wang, J. T. L., Wang, H., ApJ., 877:121, 2019.

https://iopscience.iop.org/article/10.3847/1538-4357/ab1b3c

https://arxiv.org/abs/1905.07095

https://web.njit.edu/~wangj/LSTMpredict/

About

DeepSun open source software: LSTM-flare-prediction

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 89.4%
  • Jupyter Notebook 10.6%