Dusty Stellar Source Classification by Implementing Machine Learning Methods Based on Spectroscopic Observations in the Magellanic Clouds

Ghaziasgar, Sepideh; Abdollahi, Mahdi; Javadi, Atefeh; van Loon, Jacco Th.; McDonald, Iain; Oliveira, Joana; Masoudnezhad, Amirhossein; Khosroshahi, Habib G.; Foing, Bernard H.; Fazel Hesar, Fatemeh

Abstract

Dusty stellar point sources are a significant stage in stellar evolution and contribute to the metal enrichment of galaxies. These objects can be classified using photometric and spectroscopic observations using color–magnitude diagrams and infrared excesses in spectral energy distributions. We have employed supervised machine learning spectral classification to categorize dusty stellar point sources, including young stellar objects (YSOs) and evolved stars comprising oxygen- and carbon-rich asymptotic giant branch stars (AGBs), red supergiants (RSGs), and post-AGB stars (PAGBs) in the Large and Small Magellanic Clouds, based on spectroscopic labeled data derived from the Surveying the Agents of Galaxy Evolution (SAGE) project, which involved 12 multiwavelength filters and 618 stellar objects. Despite dealing with missing values and uncertainties in the SAGE spectral data sets, we achieved accurate classifications of these sources. To address the challenge of working with small and imbalanced spectral catalogs, we utilized the Synthetic Minority Oversampling Technique (SMOTE), which generates synthetic data points. Subsequently, among all the models applied before and after data augmentation, the probabilistic random forest (PRF) classifier, a tuned random forest, demonstrated the highest total accuracy, reaching 89% based on the recall metric in categorizing dusty stellar sources. In this study, using the SMOTE technique does not improve the accuracy of the best model for the CAGB, PAGB, and RSG classes; it stays at 100%, 100%, and 88%, respectively. However, there are variations in the OAGB and YSO classes. Accordingly, we collected photometrically labeled data with properties similar to the training data set and classified them using the top four PRF models with an accuracy of more than 87%. We also collected multiwavelength data from several studies to classify them using our consensus model, which integrates the four top models to present common labels as the final prediction.

Citation

Ghaziasgar, S., Abdollahi, M., Javadi, A., van Loon, J. T., McDonald, I., Oliveira, J., Masoudnezhad, A., Khosroshahi, H. G., Foing, B. H., & Fazel Hesar, F. (2025). Dusty Stellar Source Classification by Implementing Machine Learning Methods Based on Spectroscopic Observations in the Magellanic Clouds. The Astrophysical Journal, 986(2), 1-22. https://doi.org/10.3847/1538-4357/adceeb

• Journal Article Type: Article
• Acceptance Date: Apr 14, 2025
• Online Publication Date: Jun 16, 2025
• Publication Date: Jun 16, 2025
• Deposit Date: Jun 25, 2025
• Publicly Available Date: Jun 25, 2025
• Journal: The Astrophysical Journal
• Print ISSN: 0004-637X
• Electronic ISSN: 1538-4357
• Publisher: American Astronautical Society
• Peer Reviewed: Peer Reviewed
• Volume: 986
• Issue: 2
• Article Number: 168
• Pages: 1-22
• DOI: https://doi.org/10.3847/1538-4357/adceeb
• Keywords: Small Magellanic Cloud, Red supergiant stars, Asymptotic giant branch stars, Asymptotic giant branch, Large Magellanic Cloud
• Public URL: https://keele-repository.worktribe.com/output/1280962