Deep learning interpretability analysis for carbon star identification in Gaia DR3

TL;DR

This paper introduces GaiaNet, a CNN-based model for identifying carbon stars using Gaia XP spectra, enhanced with SHAP interpretability to highlight key spectral features and discover new candidates.

Contribution

We developed GaiaNet, an improved CNN model for spectral classification, and integrated SHAP interpretability to identify key features and find new carbon star candidates in Gaia data.

Findings

GaiaNet achieved up to 100% accuracy on validation data.

SHAP analysis revealed key molecular absorption features for classification.

Identified 451 new candidate carbon stars in Gaia data.

Abstract

Context. A large fraction of Asymptotic Giant Branch (AGB) stars develop carbon-rich atmospheres during their evolution. Based on their color and luminosity, these carbon stars can be easily distinguished from many other kinds of stars. However, numerous G, K, and M giants also occupy the same region as carbon stars on the HR diagram. Despite this, their spectra exhibit differences, especially in the prominent CN molecular bands. Target. We aim to distinguish carbon stars from other kinds of stars using $Gaia$'s XP spectra, while providing attributional interpretations of key features necessary for identification, and even discovering additional new spectral key features. Method. We propose a classification model named `GaiaNet', an improved one-dimensional convolutional neural network specifically designed for handling $Gaia$'s XP spectra. We utilized the SHAP interpretability model to determine SHAP values for each feature in a spectrum, enabling us to explain the output of the `GaiaNet' model and provide further meaningful analysis. Compared to four traditional machine-learning methods, the `GaiaNet' model exhibits an average classification accuracy improvement of approximately 0.3% on the validation set, with the highest accuracy reaching 100%. Utilizing the SHAP model, we present a clear spectroscopic heatmap highlighting molecular band absorption features primarily distributed around CN773.3 and CN895.0, and summarize five key feature regions for carbon star identification. Upon applying the trained classification model to the CSTAR sample with Gaia `xp_sampled_mean' spectra, we obtained 451 new candidate carbon stars as a by-product.