Classification of Wolf Rayet stars using Ensemble-based Machine Learning algorithms

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Kar, Subhajit, Bhattacharya, Rajorshi, Das, Ramkrishna, Pihlström, Ylva, Lewis, Megan O.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915028576239616
author Kar, Subhajit
Bhattacharya, Rajorshi
Das, Ramkrishna
Pihlström, Ylva
Lewis, Megan O.
author_facet Kar, Subhajit
Bhattacharya, Rajorshi
Das, Ramkrishna
Pihlström, Ylva
Lewis, Megan O.
contents We develop a robust Machine Learning classifier model utilizing the eXtreme-Gradient Boosting (XGB) algorithm for improved classification of Galactic Wolf-Rayet (WR) stars based on Infrared (IR) colors and positional attributes. For our study, we choose an extensive dataset of 6555 stellar objects (from 2MASS and AllWISE data releases) lying in the Milky Way (MW) with available photometric magnitudes of different types including WR stars. Our XGB classifier model can accurately (with an 86\% detection rate) identify a sufficient number of WR stars against a large sample of non-WR sources. The XGB model outperforms other ensemble classifier models such as the Random Forest. Also, using the XGB algorithm, we develop a WR sub-type classifier model that can differentiate the WR subtypes from the non-WR sources with a high model accuracy ($>60\%$). Further, we apply both XGB-based models to a selection of 6457 stellar objects with unknown object types, detecting 58 new WR star candidates and predicting sub-types for 10 of them. The identified WR sources are mainly located in the Local spiral arm of the MW and mostly lie in the solar neighborhood.
format Preprint
id arxiv_https___arxiv_org_abs_2410_14845
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Classification of Wolf Rayet stars using Ensemble-based Machine Learning algorithms
Kar, Subhajit
Bhattacharya, Rajorshi
Das, Ramkrishna
Pihlström, Ylva
Lewis, Megan O.
Solar and Stellar Astrophysics
Astrophysics of Galaxies
Instrumentation and Methods for Astrophysics
We develop a robust Machine Learning classifier model utilizing the eXtreme-Gradient Boosting (XGB) algorithm for improved classification of Galactic Wolf-Rayet (WR) stars based on Infrared (IR) colors and positional attributes. For our study, we choose an extensive dataset of 6555 stellar objects (from 2MASS and AllWISE data releases) lying in the Milky Way (MW) with available photometric magnitudes of different types including WR stars. Our XGB classifier model can accurately (with an 86\% detection rate) identify a sufficient number of WR stars against a large sample of non-WR sources. The XGB model outperforms other ensemble classifier models such as the Random Forest. Also, using the XGB algorithm, we develop a WR sub-type classifier model that can differentiate the WR subtypes from the non-WR sources with a high model accuracy ($>60\%$). Further, we apply both XGB-based models to a selection of 6457 stellar objects with unknown object types, detecting 58 new WR star candidates and predicting sub-types for 10 of them. The identified WR sources are mainly located in the Local spiral arm of the MW and mostly lie in the solar neighborhood.
title Classification of Wolf Rayet stars using Ensemble-based Machine Learning algorithms
topic Solar and Stellar Astrophysics
Astrophysics of Galaxies
Instrumentation and Methods for Astrophysics
url https://arxiv.org/abs/2410.14845