Identifying Light-curve Signals with a Deep Learning Based Object Detection Algorithm. II. A General Light Curve Classification Framework

Abstract

Vast amounts of astronomical photometric data are generated from various projects, requiring significant effort to identify variable stars and other object classes. In light of this, a general, widely applicable classification framework would simplify the process of designing specific classifiers for various astronomical objects. We present a novel deep learning framework for classifying light curves using a weakly supervised object detection model. Our framework identifies the optimal windows for both light curves and power spectra automatically, and zooms in on their corresponding data. This allows for automatic feature extraction from both time and frequency domains, enabling our model to handle data across different scales and sampling intervals. We train our model on data sets obtained from Kepler, TESS, and Zwicky Transient Facility multiband observations of variable stars and transients. We achieve an accuracy of 87% for combined variables and transient events, which is comparable to the performance of previous feature-based models. Our trained model can be utilized directly for other missions, such as the All-sky Automated Survey for Supernovae, without requiring any retraining or fine-tuning. To address known issues with miscalibrated predictive probabilities, we apply conformal prediction to generate robust predictive sets that guarantee true-label coverage with a given probability. Additionally, we incorporate various anomaly detection algorithms to empower our model with the ability to identify out-of-distribution objects. Our framework is implemented in the Deep-LC toolkit, which is an open-source Python package hosted on Github (https://github.com/ckm3/Deep-LC) and PyPI.

26 pages, 19 figures, 6 tables. Published on ApJS. Code is available on https://github.com/ckm3/Deep-LC

References in corpus (18)

• Array Programming with NumPy
• The Astropy Project: Sustaining and Growing a Community-oriented Open-source Project and the Latest Major Release (v5.0) of the Core Package
• The Transiting Exoplanet Survey Satellite
• The Zwicky Transient Facility: System Overview, Performance, and First Results
• The K2 Mission: Characterization and Early results
• Assessing statistical significance of periodogram peaks
• A recurrent neural network for classification of unevenly sampled variable stars
• K2 Variable Catalogue: Variable Stars and Eclipsing Binaries in K2 Campaigns 1 and 0
• ExoMiner: A Highly Accurate and Explainable Deep Learning Classifier that Validates 301 New Exoplanets
• Identifying Exoplanets with Deep Learning III: Automated Triage and Vetting of TESS Candidates
• Anomaly detection in the Zwicky Transient Facility DR3
• Scalable End-to-end Recurrent Neural Network for Variable star classification
• PELICAN: deeP architecturE for the LIght Curve ANalysis
• Classifying Kepler light curves for 12,000 A and F stars using supervised feature-based machine learning
• TESS Data for Asteroseismology (T'DA) Stellar Variability Classification Pipeline: Set-Up and Application to the Kepler Q9 Data
• Image-based Classification of Variable Stars: First Results from Optical Gravitational Lensing Experiment Data
• Accurate and Robust Stellar Rotation Periods catalog for 82771 Kepler stars using deep learning
• Astroconformer: The Prospects of Analyzing Stellar Light Curves with Transformer-Based Deep Learning Models

Cited by in corpus (3)

• FALCO: a Foundation model of Astronomical Light Curves for time dOmain astronomy
• Heartbeat Stars Recognition Based on Recurrent Neural Networks: Method and Validation
• Predicting Stellar Parameters of Massive Stars from Light Curves with Machine Learning