ORACLE: A Real-Time, Hierarchical, Deep-Learning Photometric Classifier for the LSST

Abstract

We present ORACLE, the first hierarchical deep-learning model for real-time, context-aware classification of transient and variable astrophysical phenomena. ORACLE is a recurrent neural network with Gated Recurrent Units (GRUs), and has been trained using a custom hierarchical cross-entropy loss function to provide high-confidence classifications along an observationally-driven taxonomy with as little as a single photometric observation. Contextual information for each object, including host galaxy photometric redshift, offset, ellipticity and brightness, is concatenated to the light curve embedding and used to make a final prediction. Training on $\sim$0.5M events from the Extended LSST Astronomical Time-Series Classification Challenge, we achieve a top-level (Transient vs Variable) macro-averaged precision of 0.96 using only 1 day of photometric observations after the first detection in addition to contextual information, for each event; this increases to $>$0.99 once 64 days of the light curve has been obtained, and 0.83 at 1024 days after first detection for 19-way classification (including supernova sub-types, active galactic nuclei, variable stars, microlensing events, and kilonovae). We also compare ORACLE with other state-of-the-art classifiers and report comparable performance for the 19-way classification task, in addition to delivering accurate top-level classifications much earlier. The code and model weights used in this work are publicly available at our associated GitHub repository (https://github.com/uiucsn/ELAsTiCC-Classification).

30 pages, 20 figures, 9 tables. Accepted to ApJ

References in corpus (24)

• SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
• Array Programming with NumPy
• Astropy: A Community Python Package for Astronomy
• The Astropy Project: Building an inclusive, open-science project and status of the v2.0 core package
• The Astropy Project: Sustaining and Growing a Community-oriented Open-source Project and the Latest Major Release (v5.0) of the Core Package
• LSST: from Science Drivers to Reference Design and Anticipated Data Products
• The Zwicky Transient Facility: Science Objectives
• Type Iax Supernovae: A New Class of Stellar Explosion
• SNANA: A Public Software Package for Supernova Analysis
• Type Ia Supernovae Strongly Interacting with Their Circumstellar Medium
• Photometric Supernova Classification With Machine Learning
• SuperNNova: an open-source framework for Bayesian, Neural Network based supernova classification
• RAPID: Early Classification of Explosive Transients using Deep Learning
• Alert Classification for the ALeRCE Broker System: The Light Curve Classifier
• Avocado: Photometric Classification of Astronomical Transients with Gaussian Process Augmentation
• Optimizing spectroscopic follow-up strategies for supernova photometric classification with active learning
• The ANTARES Astronomical Time-Domain Event Broker
• A Deep Learning Approach for Active Anomaly Detection of Extragalactic Transients
• GHOST: Using Only Host Galaxy Information to Accurately Associate and Distinguish Supernovae
• Fink: early supernovae Ia classification using active learning
• ATAT: Astronomical Transformer for time series And Tabular data
• Enabling the discovery of fast transients: A kilonova science module for the Fink broker
• Transient Classifiers for Fink: Benchmarks for LSST
• A Classifier-Based Approach to Multi-Class Anomaly Detection for Astronomical Transients

Cited by in corpus (2)

• The automation of optical transient discovery and classification in Rubin-era time-domain astronomy
• ABC-SN: Attention Based Classifier for Supernova Spectra