A neural network-based methodology to select young stellar object candidates from IR surveys

Abstract

Observed Young Stellar Objects (YSOs) are used to study star formation and characterize star forming regions. For this purpose, YSO candidate catalogs are compiled from various surveys, especially in the infrared (IR), and simple selection schemes in colour-magnitude diagrams (CMDs) are often used to identify and classify YSOs. We propose a methodology for YSO classification through Machine Learning (ML) using Spitzer IR data. We detail our approach in order to ensure reproducibility and provide an in-depth example on how to efficiently apply ML to an astrophysical classification. We used feedforward Artificial Neural Networks (ANNs) that use the four IRAC bands ($3.6,4.5,5.8$ and $8\mu m$) and the $24\mu m$ MIPS band from Spitzer to classify point source objects into CI and CII YSO candidates or as contaminants. We found that ANNs can efficiently be applied to YSO classification with a contained number of neurons ($\sim$ 25). Knowledge gathered on one star-forming region has shown to be partly efficient for prediction in new regions. Best generalization capacity was achieved using a combination of several star-forming regions to train the network. Carefully rebalancing the training proportions was necessary to achieve good results. We achieved above 90% and 97% recovery rate for CI and CII YSOs, respectively, with precision above 80% and 90% for our most general result. We took advantage of the ANN great flexibility to define, for each object, an effective membership probability to each output class. Using a threshold in this probability was found to efficiently improve the classification results at a reasonable cost of object exclusion. With this selection, we reached 90% precision on CI YSOs, for more than half of them. Our catalog of YSO candidates in Orion (365 CI, 2381 CII) and NGC 2264 (101 CI, 469 CII) predicted by our final ANN is publicly available at CDS.

References in corpus (30)

• Gaia Data Release 2. Summary of the contents and survey properties
• The Wide-field Infrared Survey Explorer (WISE): Mission Description and Initial On-orbit Performance
• The Spitzer Space Telescope Mission
• Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification
• Interpreting Spectral Energy Distributions from Young Stellar Objects. I. A grid of 200,000 YSO model SEDs
• Rotation-invariant convolutional neural networks for galaxy morphology prediction
• A Spitzer Survey of Young Stellar Clusters within One Kiloparsec of the Sun: Cluster Core Extraction and Basic Structural Analysis
• The Star Formation Rate and Gas Surface Density Relation in the Milky Way: Implications for Extragalactic Studies
• The Spitzer Space Telescope Survey of the Orion A & B Molecular Clouds - Part I: A Census of Dusty Young Stellar Objects and a Study of their Mid-IR Variability
• Global Hierarchical Collapse In Molecular Clouds. Towards a Comprehensive Scenario
• unWISE: unblurred coadds of the WISE imaging
• The VLA/ALMA Nascent Disk and Multiplicity (VANDAM) Survey of Orion Protostars. A Statistical Characterization of Class 0 and I Protostellar Disks
• A catalog of visual-like morphologies in the 5 CANDELS fields using deep-learning
• A Correlation Between Surface Densities of Young Stellar Objects and Gas in Eight Nearby Molecular Clouds
• Revisiting the Hubble sequence in the SDSS DR7 spectroscopic sample: a publicly available bayesian automated classification
• A Herschel and APEX Census of the Reddest Sources in Orion: Searching for the Youngest Protostars
• An all-sky Support Vector Machine selection of WISE YSO Candidates
• 3D shape of Orion A from Gaia DR2
• Overview of the Massive Young Star-Forming Complex Study in Infrared and X-ray (MYStIX) Project
• Newton vs the machine: solving the chaotic three-body problem using deep neural networks
• ALMA survey of Class II protoplanetary disks in Corona Australis: a young region with low disk masses
• Identification of Young Stellar Object candidates in the $Gaia$ DR2 x AllWISE catalogue with machine learning methods
• Supernova Driving. IV. The Star Formation Rate of Molecular Clouds
• Alma Survey Of Circumstellar Disks In The Young Stellar Cluster IC 348
• Pre-main sequence stars in the Cepheus flare region
• A Spitzer View of Mon OB1 East/NGC 2264
• Do Androids Dream of Magnetic Fields? Using Neural Networks to Interpret the Turbulent Interstellar Medium
• Incorporating Astrochemistry into Molecular Line Modelling via Emulation
• Protostellar classification using supervised machine learning algorithms
• MaNN: Multiple Artificial Neural Networks for modelling the Interstellar Medium

Cited by in corpus (3)

• A Naive Bayes Classifier for identifying Class II YSOs
• Probabilistic Classification of Infrared-selected targets for SPHEREx mission: In search of YSOs
• YOLO-CL cluster detection in the Rubin/LSST DC2 simulation