Leveraging Machine Learning for Accurate and Fast Stellar Mass Estimation of Galaxies

TL;DR

This paper demonstrates that machine learning algorithms can estimate galaxy stellar masses with accuracy comparable to traditional methods but with vastly improved speed, making them suitable for large astronomical datasets.

Contribution

The study compares various ML algorithms to traditional SED-fitting, showing ML's potential as a faster alternative for galaxy stellar mass estimation.

Findings

ML algorithms achieve similar accuracy to LePhare.

ML methods are 1,000 to 100,000 times faster.

K-means and HDBSCAN are top performers.

Abstract

Unveiling the evolutionary history of galaxies necessitates a precise understanding of their physical properties. Traditionally, astronomers achieve this through spectral energy distribution (SED) fitting. However, this approach can be computationally intensive and time-consuming, particularly for large datasets. This study investigates the viability of machine learning (ML) algorithms as an alternative to traditional SED-fitting for estimating stellar masses in galaxies. We compare a diverse range of unsupervised and supervised learning approaches including prominent algorithms such as K-means, HDBSCAN, Parametric t-Distributed Stochastic Neighbor Embedding (Pt-SNE), Principal Component Analysis (PCA), Random Forest, and Self-Organizing Maps (SOM) against the well-established LePhare code, which performs SED-fitting as a benchmark. We train various ML algorithms using simple model SEDs in photometric space, generated with the BC03 code. These trained algorithms are then employed to estimate the stellar masses of galaxies within a subset of the COSMOS survey dataset. The performance of these ML methods is subsequently evaluated and compared with the results obtained from LePhare, focusing on both accuracy and execution time. Our evaluation reveals that ML algorithms can achieve comparable accuracy to LePhare while offering significant speed advantages (1,000 to 100,000 times faster). K-means and HDBSCAN emerge as top performers among our selected ML algorithms. Supervised learning algorithms like Random Forest and manifold learning techniques such as Pt-SNE and SOM also show promising results. These findings suggest that ML algorithms hold significant promise as a viable alternative to traditional SED-fitting methods for estimating the stellar masses of galaxies.