Heartbeat Stars Recognition Based on Recurrent Neural Networks: Method and Validation

TL;DR

This paper introduces a machine learning approach using recurrent neural networks and harmonic features for automatic detection and classification of heartbeat stars from light curves, achieving high accuracy and discovering new HBSs.

Contribution

The study presents a novel feature extraction method based on orbital harmonics and demonstrates the effectiveness of RNNs in classifying heartbeat stars with high accuracy.

Findings

Achieved 95% accuracy on synthetic light curves.

Achieved 86% average accuracy on real survey data.

Successfully identified four new heartbeat stars.

Abstract

Since the variety of their light curve morphologies, the vast majority of the known heartbeat stars (HBSs) have been discovered by manual inspection. Machine learning, which has already been successfully applied to the classification of variable stars based on light curves, offers another possibility for the automatic detection of HBSs. We propose a novel feature extraction approach for HBSs. First, the orbital frequencies are calculated automatically according to the Fourier spectra of the light curves. Then, the amplitudes of the first 100 harmonics are extracted. Finally, these harmonics are normalized as feature vectors of the light curve. A training data set of synthetic light curves is constructed using ELLC, and their features are fed into recurrent neural networks (RNNs) for supervised learning, with the expected output being the eccentricity of these light curves. The performance of the RNNs is evaluated using a test data set of synthetic light curves, achieving 95$\%$ accuracy. When applied to known HBSs from the OGLE, Kepler, and TESS surveys, the networks achieve an average accuracy of 86$\%$. This method successfully identifies four new HBSs within the eclipsing binary catalog of Kirk et al. The use of orbital harmonics as features for HBSs proves to be a practical approach that significantly reduces the computational cost of neural networks. RNNs show excellent performance in recognizing this type of time series data. This method not only allows efficient identification of HBSs but can also be extended to recognize other types of periodic variable stars.