Scalable End-to-end Recurrent Neural Network for Variable star classification

Abstract

During the last decade, considerable effort has been made to perform automatic classification of variable stars using machine learning techniques. Traditionally, light curves are represented as a vector of descriptors or features used as input for many algorithms. Some features are computationally expensive, cannot be updated quickly and hence for large datasets such as the LSST cannot be applied. Previous work has been done to develop alternative unsupervised feature extraction algorithms for light curves, but the cost of doing so still remains high. In this work, we propose an end-to-end algorithm that automatically learns the representation of light curves that allows an accurate automatic classification. We study a series of deep learning architectures based on Recurrent Neural Networks and test them in automated classification scenarios. Our method uses minimal data preprocessing, can be updated with a low computational cost for new observations and light curves, and can scale up to massive datasets. We transform each light curve into an input matrix representation whose elements are the differences in time and magnitude, and the outputs are classification probabilities. We test our method in three surveys: OGLE-III, Gaia and WISE. We obtain accuracies of about $95\%$ in the main classes and $75\%$ in the majority of subclasses. We compare our results with the Random Forest classifier and obtain competitive accuracies while being faster and scalable. The analysis shows that the computational complexity of our approach grows up linearly with the light curve size, while the traditional approach cost grows as $N\log (N)$.

15 pages, 17 figures. To be published in MNRAS

References in corpus (11)

• The Gaia mission
• Rotation-invariant convolutional neural networks for galaxy morphology prediction
• Automated supervised classification of variable stars I. Methodology
• A recurrent neural network for classification of unevenly sampled variable stars
• Deep-HiTS: Rotation Invariant Convolutional Neural Network for Transient Detection
• The EPOCH Project: I. Periodic variable stars in the EROS-2 LMC database
• A machine learned classifier for RR Lyrae in the VVV survey
• Convolutional Neural Networks for Transient Candidate Vetting in Large-Scale Surveys
• An improved quasar detection method in EROS-2 and MACHO LMC datasets
• Unsupervised Classification of Variable Stars
• Automatic Survey-Invariant Variable Star Classification