QUEST (Quasar Unsupervised Encoder and Synthesis Tool): A machine learning framework to generate quasar spectra

Abstract

Quasars at the redshift frontier (z > 7.0) are fundamental probes of black hole (BH) growth and evolution but notoriously difficult to identify. At these redshifts, machine learning-based selection methods have proven to be efficient, but require appropriate training sets to express their full potential. Here, we present QUEST, a Variational Auto-Encoder capable of generating realistic quasar spectra that can be post-processed for generating synthetic photometry and for spectral imputation. We start from the SDSS DR16Q catalogue, pre-process the spectra, and vet the sample to obtain a clean data set. After training the model, we investigate the properties of its latent space to understand whether it has learnt relevant physics. We provide a pipeline to generate photometry from the sampled spectra, compare it with actual quasar photometry, and showcase the capabilities of the model in reconstructing and extending quasar spectra. The trained network faithfully reproduces the input spectrum, both in terms of sample median and variance. By examining the latent space, we find correlations with continuum and bolometric luminosity, BH mass, redshift, continuum slope, and emission line properties. When used to generate photometry, we find results in excellent agreement with the control sample. The model provides satisfactory results in reconstructing emission lines: estimates of the BH mass from the reconstructed spectra are in good agreement with those from the original spectra. Furthermore, when spectra with broad absorption line features are reconstructed, the model successfully interpolates over the absorption systems. Compared with previous work, we find excellent agreement between the spectra sampled from our model and the output of their results. However, QUEST does not require any ad-hoc tuning, and is capable of reproducing the full variety of spectra available in the training set.