YOLO-CL: Galaxy cluster detection in the SDSS with deep machine learning

Abstract

Galaxy clusters are powerful probes for cosmological models. Next-generation, large-scale optical and infrared surveys are poised to reach unprecedented depths and, thus, they require highly complete and pure cluster catalogs, with a well-defined selection function. We have developed a new cluster detection algorithm named YOLO for CLuster detection (YOLO-CL), which is a modified version of the state-of-the-art object detection deep convolutional network named You only look once (YOLO) that has been optimized for the detection of galaxy clusters. We trained YOLO-CL on the red-sequence Matched-filter Probabilistic Percolation (redMaPPer) cluster catalog, based on Sloan Digital Sky Survey (SDSS) color images. We find that YOLO-CL detects 95-98% of the redMaPPer clusters, with a purity of 95-98%, that is calculated by applying the network to SDSS blank fields. When compared to the Meta-Catalog of X-Ray Detected Clusters of Galaxies 2021 (MCXC2021) X-ray catalog in the SDSS footprint, YOLO-CL recovers all clusters at LX ≳ 2-3 × 1044 erg s-1, M500 ≳ 2-3 × 1014M⊙, R500≳0.75-0.8 Mpc and 0.4 ≲ z ≲ 0.6. When compared to the redMaPPer detection of the same MCXC2021 clusters, we find that YOLO-CL is more complete than redMaPPer, which means that the neural network has indeed improved the cluster detection efficiency of its training sample. In fact, YOLO-CL detects ∼98% of the MCXC2021 clusters with an X-ray surface brightness of IX,500 ≳ 20 × 10-15 erg s-1 cm-2 arcmin-2 at 0.2 ≲ z ≲ 0.6 and ∼100% of the MCXC2021 clusters with IX,500 ≳ 30 × 10-15 erg s-1 cm-2 arcmin-2 at 0.3 ≲ z ≲ 0.6; while redMaPPer detects ∼98% of the MCXC2021 clusters with IX,500 ≳ 55 × 10-15 erg s-1 cm-2 arcmin-2 at 0.2 ≲ z ≲ 0.6 and ∼100% of the MCXC2021 clusters with IX,500 ≳ 20 × 10-15 erg s-1 cm-2 arcmin-2 at 0.5 ≲ z ≲ 0.6. The YOLO-CL selection function is approximately constant with redshift, with respect to the MCXC2021 cluster X-ray surface brightness. YOLO-CL exhibits a high level of performance when compared to traditional detection algorithms applied to SDSS. Deep learning networks display a strong advantage over traditional galaxy cluster detection techniques because they do not require the galaxy's photometric and photometric redshift catalogs. This eliminates systematic uncertainties that may be introduced during source detections and photometry, as well as photometric redshift measurements. Our results show that YOLO-CL is an efficient alternative to traditional cluster detection methods. In general, this work shows that it is worth exploring the performance of deep convolution networks for future cosmological cluster surveys, such as the Rubin/Legacy Survey of Space and Time (Rubin/LSST), Euclid, and Roman Space Telescope surveys.