Transferring spectroscopic stellar labels to 217 million Gaia DR3 XP stars with SHBoost

TL;DR

This study demonstrates that machine learning regression using gradient-boosted trees can accurately derive stellar parameters and extinctions from Gaia DR3 XP spectra for over 217 million stars, without ground-based spectra.

Contribution

It introduces a novel application of gradient-boosted random-forest regression to extract stellar parameters from Gaia XP spectra at an unprecedented scale, with reliable uncertainties and interpretability.

Findings

Achieved median uncertainties of 0.20 mag in extinction and 0.01 dex in temperature.

Predicted stellar parameters with competitive accuracy compared to classical methods.

Extended the stellar parameter estimation to over 217 million Gaia stars.

Abstract

In this paper, we explore the feasibility of using machine learning regression as a method of extracting basic stellar parameters and line-of-sight extinctions from spectro-photometric data. We built a stable gradient-boosted random-forest regressor (xgboost), trained on spectroscopic data, capable of producing output parameters with reliable uncertainties from Gaia DR3 data (most notably the low-resolution XP spectra), without ground-based spectroscopic observations. Using Shapley additive explanations, we interpret how the predictions for each star are influenced by each data feature. For the training and testing of the network, we used high-quality parameters obtained from the StarHorse code for a sample of around eight million stars observed by major spectroscopic stellar surveys, complemented by curated samples of hot stars, very metal-poor stars, white dwarfs, and hot sub-dwarfs. The training data cover the whole sky, all Galactic components, and almost the full magnitude range of the Gaia DR3 XP sample of more than 217 million objects that also have reported parallaxes. We have achieved median uncertainties of 0.20 mag in V-band extinction, 0.01 dex in logarithmic effective temperature, 0.20 dex in surface gravity, 0.18 dex in metallicity, and $12\%$ in mass (over the full Gaia DR3 XP sample, with considerable variations in precision as a function of magnitude and stellar type). We succeeded in predicting competitive results based on Gaia DR3 XP spectra compared to classical isochrone or spectral-energy distribution fitting methods we employed in earlier works, especially for parameters $A_V$ and $T_{\rm eff}$, along with the metallicity values. Finally, we showcase some potential applications of this new catalogue, including extinction maps, metallicity trends in the Milky Way, and extended maps of young massive stars, metal-poor stars, and metal-rich stars). [abridged]