DERIVING STAR CLUSTER PARAMETERS WITH CONVOLUTIONAL NEURAL NETWORKS

Jonas Bialopetravičius1, 2, Donatas Narbutis1, 2, Vladas Vansevičius1, 2

1 Vilnius University Observatory, Saulėtekio av. 3, LT-10257 Vilnius, Lithuania

2 Center for Physical Sciences and Technology, Saulėtekio av. 3, LT-10257 Vilnius, Lithuania

[email protected]

Convolutional neural networks (CNNs) have been proven to perform fast object classification and detection on natural images and have the potential to infer astrophysical parameters on the exponentially increasing amount of sky-survey imaging data. The inference pipeline can be trained either from real human-annotated data or simulated mock observations. Until now, star cluster analysis was based on integral or resolved stellar photometry. This limits the amount of information that can be extracted from cluster images.

We aim to develop a CNN-based algorithm capable of simultaneously deriving ages, masses, and sizes of star clusters directly from multi-band images. We also aim to demonstrate CNN capabilities on low-mass semi-resolved star clusters in a low-signal-to-noise-ratio regime.

In this study a CNN was constructed based on the deep residual network (ResNet) architecture [1] and trained on simulated images of star clusters with various ages, masses, and sizes. To provide realistic backgrounds, star fields of M31 galaxy taken from The Panchromatic Hubble Andromeda Treasury (PHAT) survey [2] were added to the mock cluster images. Examples of generated clusters can be seen in Fig. 1.

Figure 1
Fig. 1. Examples of generated clusters on the same real background image. The ages of all displayed clusters are 10 Myr. The mass and $r_h$ (radius that encloses half the number of stars) values are varied as shown on the axes. The intensity scale of the images was normalized with the arcsinh function.
Figure 2
Fig. 2. Test results of CNN performance of clusters with age 10 Myr. Each black dot corresponds to the true parameters of 200 artificial clusters. Ellipses enclose one $\sigma$ (39% of 2D distribution) of their inferred values (precision), with the red arrows pointing to the means of the distributions (bias).

The proposed CNN was verified on mock images of artificial clusters and has demonstrated high precision and no significant bias (Fig. 2) for clusters of ages less than 3 Gyr, masses between 250 and 4000 $M_{\odot}$, and sizes ($r_h$) between 0.15 and 10 arcsec. The CNN-based cluster analysis pipeline is end-to-end, starting from input images all the way to the inferred parameters; no hand-coded steps have to be performed: estimates of parameters are provided by the neural network in one inferential step from raw images [3].

This allows both unresolved and semi-resolved cluster cases to be dealt with homogeneously, and multiple photometric passbands to be used in an integrated manner.


[1] K. He, X. Zhang, S. Ren, J. Sun, Deep Residual Learning for Image Recognition, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[2] PHAT survey data: https://archive.stsci.edu/prepds/phat/.

[3] J. Bialopetravičius, D. Narbutis, V. Vansevičius, Deriving star cluster parameters with convolutional neural networks. I. Age, mass, and size, Astronomy & Astrophysics, doi:10.1051/0004-6361/201833833.