Lyman Break Galaxy selection and redshift measurement with supervised contrastive learning

Abstract

Some of the next steps for high-precision cosmology lie within the high-redshift, high-density universe. Spectroscopic survey experiments such as the Dark Energy Spectroscopic Instrument (DESI)'s second phase DESI Run 2 will shift towards probing Lyman Break Galaxy (LBG) populations from z$\sim$2 to z$\sim$4.5. For this faint sample, spectroscopic redshift measurement and sample decontamination remains a challenge, even after target selection. We propose an approach based on supervised weighted contrastive learning, in order to both learn a redshift representation for spectra and decontaminate the sample from quasars and low redshift emission line galaxies. This strategy generalizes the contrastive learning loss approach with continuous relationship weights, such that the network simultaneously learns redshift and classification tasks. The model shows stronger outlier classification and comparable redshift identification performances when compared to the previous network used for DESI (a modified version of QuasarNET) on the same dataset. In particular, contrastive learning is well suited to the small, visually-inspected sample used for training and testing, especially given the multi-task nature of this work.