Photometric redshift-aided classification using ensemble learning

Abstract

We present SHEEP, a new machine learning approach to the classic problem of astronomical source classification, which combines the outputs from the XGBoost, LightGBM, and CatBoost learning algorithms to create stronger classifiers. A novel step in our pipeline is that prior to performing the classification, SHEEP first estimates photometric redshifts, which are then placed into the data set as an additional feature for classification model training; this results in significant improvements in the subsequent classification performance. SHEEP contains two distinct classification methodologies: (i) Multi-class and (ii) one versus all with correction by a meta-learner. We demonstrate the performance of SHEEP for the classification of stars, galaxies, and quasars using a data set composed of SDSS and WISE photometry of 3.5 million astronomical sources. The resulting F1-scores are as follows: 0.992 for galaxies; 0.967 for quasars; and 0.985 for stars. In terms of the F1-scores for the three classes, SHEEP is found to outperform a recent RandomForest-based classification approach using an essentially identical data set. Our methodology also facilitates model and data set explainability via feature importances; it also allows the selection of sources whose uncertain classifications may make them interesting sources for follow-up observations.

Accepted for publication in A&A; 10 pages, 8 figures, 6 tables

References in corpus (11)

• The James Webb Space Telescope
• Unsupervised star, galaxy, qso classification: Application of HDBSCAN
• Star formation rates and stellar masses from machine learning
• Machine-learning identification of galaxies in the WISExSuperCOSMOS all-sky catalogue
• A machine learning approach to galaxy properties: joint redshift-stellar mass probability distributions with Random Forest
• On the discovery of stars, quasars, and galaxies in the Southern Hemisphere with S-PLUS DR2
• Star Formation Rates for photometric samples of galaxies using machine learning methods
• The miniJPAS survey: the photometric redshift catalogue
• Inferring physical properties of galaxies from their emission line spectra
• Identification of BASS DR3 Sources as Stars, Galaxies and Quasars by XGBoost
• AstronomicAL: An interactive dashboard for visualisation, integration and classification of data using Active Learning

Cited by in corpus (18)

• Machine-learning classification of astronomical sources: estimating F1-score in the absence of ground truth
• Transferring spectroscopic stellar labels to 217 million Gaia DR3 XP stars with SHBoost
• Euclid preparation: XXII. Selection of Quiescent Galaxies from Mock Photometry using Machine Learning
• PhotoRedshift-MML: a multimodal machine learning method for estimating photometric redshifts of quasars
• Selection of powerful radio galaxies with machine learning
• Euclid preparation. LI. Forecasting the recovery of galaxy physical properties and their relations with template-fitting and machine-learning methods
• Semi-supervised classification of stars, galaxies and quasars using K-means and random-forest approaches
• Machine learning applications for the study of AGN physical properties using photometric observations
• Identifying type II quasars at intermediate redshift with few-shot learning photometric classification
• Machine Learning Classification of COSMOS2020 Galaxies: Quiescent vs. Star-Forming
• Machine learning based stellar classification with highly sparse photometry data
• Photometric classification of QSOs from ALHAMBRA survey using random forest
• Euclid preparation. Estimating galaxy physical properties using CatBoost chained regressors with attention
• COSMOS2025: Machine Learning Classification of Early- and Late-type Galaxies at 0 < z < 3
• Hybrid-z: Enhancing Kilo-Degree Survey bright galaxy sample photometric redshifts with deep learning
• COSMOS2025: A Machine Learning Census of Massive Quiescent Galaxies at $2.5<\text{z}<5$
• Efficient neutral-IGM inference from noisy 21-cm forest spectra with latent-space U-Net encoding and XGBoost
• A gradient boosting and broadband approach to finding Lyman-α emitting galaxies beyond narrowband surveys