Abstract:(abridged) Mass loss is a key parameter in the evolution of massive stars, with discrepancies between theory and observations and with unknown importance of the episodic mass loss. To address this we need increased numbers of classified sources stars spanning a range of metallicity environments. We aim to remedy the situation by applying machine learning techniques to recently available extensive photometric catalogs. We used IR/Spitzer and optical/Pan-STARRS, with Gaia astrometric information, to compile a large catalog of known massive stars in M31 and M33, which were grouped in Blue, Red, Yellow, B[e] supergiants, Luminous Blue Variables, Wolf-Rayet, and background galaxies. Due to the high imbalance, we implemented synthetic data generation to populate the underrepresented classes and improve separation by undersampling the majority class. We built an ensemble classifier using color indices. The probabilities from Support Vector Classification, Random Forests, and Multi-layer Perceptron were combined for the final classification. The overall weighted balanced accuracy is ~83%, recovering Red supergiants at ~94%, Blue/Yellow/B[e] supergiants and background galaxies at ~50-80%, Wolf-Rayets at ~45%, and Luminous Blue Variables at ~30%, mainly due to their small sample sizes. The mixing of spectral types (no strict boundaries in their color indices) complicates the classification. Independent application to IC 1613, WLM, and Sextans A galaxies resulted in an overall lower accuracy of ~70%, attributed to metallicity and extinction effects. The missing data imputation was explored using simple replacement with mean values and an iterative imputor, which proved more capable. We also found that r-i and y-[3.6] were the most important features. Our method, although limited by the sampling of the feature space, is efficient in classifying sources with missing data and at lower metallicitites.

A Random Forest Approach to Identifying Young Stellar Object Candidates in the Lupus Star-forming Region

Machine Learning Methods for Automated Interstellar Object Classification with LSST

Using Machine Learning to Identify Extragalactic Globular Cluster Candidates from Ground-Based Photometric Surveys of M87

Identifying Lyα emitter candidates with Random Forest: learning from galaxies in CANDELS survey

RR Lyrae Star Candidates from SDSS Databases by Cost-sensitive Random Forests

A kinematic study and membership analysis of the Lupus star-forming region

A Machine-Learning Photometric Classifier for Massive Stars in Nearby Galaxies I. the Method

Preparing for advanced LIGO: A Star-Galaxy Separation Catalog for the Palomar Transient Factory

A Gaia Survey for Young Stars Associated with the Lupus Clouds

Discrimination of Point Sources Using Radial Basis Function Network and Random Forest

X-Shooter spectroscopy of young stellar objects in Lupus. Atmospheric parameters, membership and activity diagnostics

Machine learning methods for the search for L&T brown dwarfs in the data of modern sky surveys

A Multiwavelength Machine-learning Approach to Classifying X-ray Sources in the Fields of Unidentified 4FGL-DR4 sources

Probabilistic Classification of Infrared-selected targets for SPHEREx mission: In search of YSOs

Machine learning based stellar classification with highly sparse photometry data

A proper motion study of the Lupus clouds using VO tools

New members of the Lupus I cloud based on Gaia astrometry Physical and accretion properties from X-Shooter spectra

New Low-Mass Stars and Brown Dwarfs with Disks in Lupus

High-redshift LBG selection from broadband and wide photometric surveys using a Random Forest algorithm

Classification and Ranking of Fermi LAT Gamma-ray Sources from the 3FGL Catalog using Machine Learning Techniques