Using machine learning to explore the long-term evolution of IGR J17091-3624
Reporter:Ming-Chun Hung
Supervisor:Albert Kong, Alvina On, Surojit Saha
Motivation
Overview of X-ray Binaries (XRBs)
Overview of IGR J17091-3624
Target
To identify different QPO light curves from the 2011–2013 long-term data of IGR J17091-3624.
Method
To train a 1D CNN neural network classifier using the labels from the latter part of the paper "An atlas of exotic variability in IGR J17091−3624: a comparison with GRS 1915+105" and the RXTE results as input
Problem
The long-term data from RXTE has an exposure time of 16 seconds, whereas the short-term data (QPO light curve)records the exact arrival times of individual photons.
Each data point has a 16-second exposure.
Problem
Each point represents a single photon and its energy. The horizontal axis indicates the time elapsed since the first photon (the original data records the absolute arrival time).
Solution
Since the short-term data records the exact arrival times at the telescope, I can segment the data to count the number of photons arriving within 16-second intervals. Subsequently, I can use this processed data to retrain the neural network to match the long-term data.
Solution
The figure shows the raw RXTE data. The interval between the blue lines is 16 seconds. I summed the energy within these intervals and converted it to mCrab to match the unit of the long-term data.
Solution
I segmented the raw RXTE data into 1-D arrays of 1,024 seconds each (with a bin size of 16 * 64) to serve as input for the neural network. Segments were generated from the same observation using a sliding window approach to increase the dataset size.
Neural Network Architecture: CNN Classifier
Input
1D CNN
1D CNN
1D CNN
1D CNN
Linear
Output
Result:CNN Classifier
Device:RTX 3090
AI Platfrom:Pytorch
Max Iteration:500
Iteration:500
Accuracy: 87.5%
Result:CNN Classifier
I applied label smoothing to restrict the model's confidence to a maximum of 96%, which consequently capped the lowest achievable loss at around 0.6.
Problem: Confusion between Class I and Class IX
As shown in the confusion matrix, Class IX has a relatively high probability of being classified as Class IX.
Solution:Confusion between Class I and Class IX
We incorporated FFT into the architecture to see if it could capture subtle differences among QPOs, thereby improving classification accuracy.
Neural Network Architecture:CNN+FFT
Input
1D CNN
1D CNN
1D CNN
1D CNN
Linear
Output
FFT
Linear
Result:CNN+FFT Classifier
Device:RTX 3090
AI Platfrom:Pytorch
Max Iteration:500
Iteration:500
Accuracy: 94%
As shown in the confusion matrix, incorporating FFT significantly reduced the misclassification rate between Class I and Class IX.
Result
CNN Classifier vs CNN+FFT Classifier
Next Step
Next, I plan to apply a sliding-window-like algorithm to have the neural network classify the 2011–2013 input data into their respective classes.
By using the light curve of this segment as the input to the neural network, we can determine its corresponding class.
Result:Class I
Result:Class II
Result:Class III
Result:Class IV
Result:Class V
Result:Class VI
Result:Class VII
Result:Class VIII
Result:Class IX
Conclusion
Conclusion
Thank You for Listening.�Q&A