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 (7)

• 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
• Spitzer Infrared Spectrographic point source classification in the Small Magellanic Cloud
• 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
• Massive young stellar objects in the Local Group irregular galaxy NGC6822 identified using machine learning
• Massive young stellar objects in the Local Group spiral galaxy M33 identified using machine learning