Machine Learning Classification of Young Stellar Objects and Evolved Stars in the Magellanic Clouds Using the Probabilistic Random Forest Classifier

Abstract

The Magellanic Clouds (MCs) are excellent locations to study stellar dust emission and its contribution to galaxy evolution. Through spectral and photometric classification, MCs can serve as a unique environment for studying stellar evolution and galaxies enriched by dusty stellar point sources. We applied machine learning classifiers to spectroscopically labeled data from the Surveying the Agents of Galaxy Evolution (SAGE) project, which involved 12 multiwavelength filters and 618 stellar objects at the MCs. We classified stars into five categories: young stellar objects (YSOs), carbon-rich asymptotic giant branch (CAGB) stars, oxygen-rich AGB (OAGB) stars, red supergiants (RSG), and post-AGB (PAGB) stars. Following this, we augmented the distribution of imbalanced classes using the Synthetic Minority Oversampling Technique (SMOTE). Therefore, the Probabilistic Random Forest (PRF) classifier achieved the highest overall accuracy, reaching $89\%$ based on the recall metric, in categorizing dusty stellar sources before and after data augmentation. In this study, SMOTE did not impact the classification accuracy for the CAGB, PAGB, and RSG categories but led to changes in the performance of the OAGB and YSO classes.

6 pages, 2 figures, 2 tables, Published in the Communications of BAO (ComBAO)

References in corpus (19)

• A new generation of PARSEC-COLIBRI stellar isochrones including the TP-AGB phase
• Probabilistic Random Forest: A machine learning algorithm for noisy datasets
• The Evolution of Massive Stars. I. Red Supergiants in the Magellanic Clouds
• The wind speeds, dust content, and mass-loss rates of evolved AGB and RSG stars at varying metallicity
• The weirdest SDSS galaxies: results from an outlier detection algorithm
• Optically visible post-AGB/RGB stars and young stellar objects in the Small Magellanic Cloud: candidate selection, spectral energy distributions and spectroscopic examination
• Optically Visible Post-AGB Stars, Post-RGB Stars and Young Stellar Objects in the Large Magellanic Cloud
• Spitzer Infrared Spectrographic point source classification in the Small Magellanic Cloud
• The UK Infrared Telescope M33 monitoring project. III. Feedback from dusty stellar winds in the central square kiloparsec
• Pulsation-triggered mass loss from AGB stars: the 60-day critical period
• The UK Infrared Telescope M33 monitoring project. I. Variable red giant stars in the central square kiloparsec
• Massive young stellar objects in the Local Group irregular galaxy NGC6822 identified using machine learning
• The Isaac Newton Telescope monitoring survey of Local Group dwarf galaxies -- IV. The star formation history of Andromeda VII derived from long period variable stars
• Protostellar classification using supervised machine learning algorithms
• Massive young stellar objects in the Local Group spiral galaxy M33 identified using machine learning
• Properties of star formation of the Large Magellanic Cloud as probed by young stellar objects
• AGB population as probes of galaxy structure and evolution
• Mass-loss rates of cool evolved stars in M33 galaxy
• The Physical Properties of Red Supergiants