Protostellar classification using supervised machine learning algorithms

Abstract

Classification of young stellar objects (YSOs) into different evolutionary stages helps us to understand the formation process of new stars and planetary systems. Such classification has traditionally been based on spectral energy distributions (SEDs). An alternative approach is provided by supervised machine learning algorithms. We attempt to classify a sample of Orion YSOs into different classes, where each source has already been classified using multiwavelength SED analysis. We used 8 different learning algorithms to classify the target YSOs, namely a decision tree, random forest, gradient boosting machine (GBM), logistic regression, naïve Bayes classifier, $k$-nearest neighbour classifier, support vector machine, and neural network. The classifiers were trained and tested by using a 10-fold cross-validation procedure. As the learning features, we employed ten continuum flux densities spanning from the near-IR to submm. With a classification accuracy of 82% (with respect to the SED-based classes), a GBM algorithm was found to exhibit the best performance. The lowest accuracy of 47% was obtained with a naïve Bayes classifier. Our analysis suggests that the inclusion of the 3.6 $\mu$m and 24 $\mu$m flux densities is useful to maximise the YSO classification accuracy. Although machine learning has the potential to provide a rapid and fairly reliable way to classify YSOs, an SED analysis is still needed to derive the physical properties of the sources, and to create the labelled training data. The classification accuracies can be improved with respect to the present results by using larger data sets, more detailed missing value imputation, and advanced ensemble methods. Overall, the application of machine learning is expected to be very useful in the era of big astronomical data, for example to quickly assemble interesting target source samples for follow-up studies.

17 pages, 6 figures, 2 tables, accepted for publication in Astrophysics and Space Science, abstract abridged for arXiv

References in corpus (21)

• The Large APEX Bolometer Camera LABOCA
• The Herschel Orion Protostar Survey: Spectral Energy Distributions and Fits Using a Grid of Protostellar Models
• Photometric Supernova Classification With Machine Learning
• An all-sky Support Vector Machine selection of WISE YSO Candidates
• Classifying Radio Galaxies with Convolutional Neural Network
• Searching for Exoplanets Using Artificial Intelligence
• The Herschel Orion Protostar Survey: Luminosity and Envelope Evolution
• Transfer learning for galaxy morphology from one survey to another
• An automatic taxonomy of galaxy morphology using unsupervised machine learning
• Radio Galaxy Zoo: Compact and extended radio source classification with deep learning
• Integrating human and machine intelligence in galaxy morphology classification tasks
• Machine learning search for variable stars
• Spitzer IRAC Detection and Analysis of Shocked Molecular Hydrogen Emission
• Machine-learning identification of galaxies in the WISExSuperCOSMOS all-sky catalogue
• A machine-learning method for identifying multi-wavelength counterparts of submillimeter galaxies: training and testing using AS2UDS and ALESS
• Galaxy And Mass Assembly: Automatic Morphological Classification of Galaxies Using Statistical Learning
• A Machine Learning Approach to Galaxy-LSS Classification I: Imprints on Halo Merger Trees
• Improved Selection Criteria for HII Regions, based on IRAS Sources
• Operations and Performance of the PACS Instrument 3He Sorption Cooler on board of the Herschel Space Observatory
• Characterising the physical and chemical properties of a young Class 0 protostellar core embedded in the Orion B9 filament
• The Science Case for an Extended Spitzer Mission

Cited by in corpus (9)

• Surveying the reach and maturity of machine learning and artificial intelligence in astronomy
• A Naive Bayes Classifier for identifying Class II YSOs
• A neural network-based methodology to select young stellar object candidates from IR surveys
• A Random Forest Approach to Identifying Young Stellar Object Candidates in the Lupus Star-Forming Region
• A Classifier to Detect Elusive Astronomical Objects through Photometry
• Probabilistic Classification of Infrared-selected targets for SPHEREx mission: In search of YSOs
• A Possible Converter to Denoise the Images of Exoplanet Candidates through Machine Learning Techniques
• Machine Learning Classification of Young Stellar Objects and Evolved Stars in the Magellanic Clouds Using the Probabilistic Random Forest Classifier
• The viewing angle in AGN SED models, a data-driven analysis