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.

Determination of metallicities of red giant stars using machine learning techniques applied to the narrow and broadband photometry of the S-PLUS survey The catalogue is available at the CDS.

A machine learning approach to photometric metallicities of giant stars

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

Stellar atmospheric parameters and chemical abundances of about 5 million stars from S-PLUS multi-band photometry

Stellar Loci. VII. Photometric Metallicities of 5 Million FGK Stars Based on GALEX GR6+7 AIS and Gaia EDR3

J-PLUS: Beyond Spectroscopy III. Stellar Parameters and Elemental-abundance Ratios for Five Million Stars from DR3

Stellar loci IV. Red giant stars

Metallicity and $α$-abundance for 48 million stars in low-extinction regions in the Milky Way

Using machine learning to investigate the populations of dusty evolved stars in various metallicities

Robust Data-driven Metallicities for 175 Million Stars from Gaia XP Spectra

J-PLUS: Beyond Spectroscopy. III. Stellar Parameters and Elemental-abundance Ratios for Five Million Stars from DR3

Machine Learning the Tip of the Red Giant Branch

Ca II Triplet Spectroscopy of Small Magellanic Cloud Red Giants. V. Abundances and Velocities for 12 Massive Clusters

NGC 1856: Using Machine Learning Techniques to Uncover Detailed Stellar Abundances from MUSE Data

Spectroscopic follow-up of statistically selected extremely metal-poor star candidates from GALAH DR3

Stellar population astrophysics with the TNG. Abundance analysis of nearby red giants and red clump stars: Combining high-resolution spectroscopy and asteroseismology

NGC1856: Using machine learning techniques to uncover detailed stellar abundances from MUSE data

Measuring Detailed Chemical Abundances from Co-Added Medium-Resolution Spectra. I. Tests Using Milky Way Dwarf Spheroidal Galaxies and Globular Clusters

M giants with IGRINS I. Stellar parameters and $α$-abundance trends of the solar neighborhood population

The Survey of Surveys: machine learning for stellar parametrization

Photometric metallicity for 694233 Galactic giant stars from Gaia DR3 synthetic Stromgren photometry. Metallicity distribution functions of halo sub-structures