Using Active Learning to Improve Quasar Identification for the DESI Spectra Processing Pipeline
Abstract
The Dark Energy Spectroscopic Instrument (DESI) survey uses an automatic spectral classification pipeline to classify spectra. QuasarNET is a convolutional neural network used as part of this pipeline originally trained using data from the Baryon Oscillation Spectroscopic Survey (BOSS). In this paper we implement an active learning algorithm to optimally select spectra to use for training a new version of the QuasarNET weights file using only DESI data, specifically to improve classification accuracy. This active learning algorithm includes a novel outlier rejection step using a Self-Organizing Map to ensure we label spectra representative of the larger quasar sample observed in DESI. We perform two iterations of the active learning pipeline, assembling a final dataset of 5600 labeled spectra, a small subset of the approx 1.3 million quasar targets in DESI's Data Release 1. When splitting the spectra into training and validation subsets we meet or exceed the previously trained weights file in completeness and purity calculated on the validation dataset with less than one tenth of the amount of training data. The new weights also more consistently classify objects in the same way when used on unlabeled data compared to the old weights file. In the process of improving QuasarNET's classification accuracy we discovered a systemic error in QuasarNET's redshift estimation and used our findings to improve our understanding of QuasarNET's redshifts.
Cite
@article{arxiv.2505.01596,
title = {Using Active Learning to Improve Quasar Identification for the DESI Spectra Processing Pipeline},
author = {Dylan Green and David Kirkby and J. Aguilar and S. Ahlen and D. M. Alexander and E. Armengaud and S. Bailey and A. Bault and D. Bianchi and A. Brodzeller and D. Brooks and T. Claybaugh and R. de Belsunce and A. de la Macorra and P. Doel and V. A. Fawcett and S. Ferraro and A. Font-Ribera and J. E. Forero-Romero and E. Gaztañaga and S. Gontcho A Gontcho and G. Gutierrez and M. Ishak and S. Juneau and R. Kehoe and T. Kisner and A. Kremin and A. Lambert and M. Landriau and L. Le Guillou and M. E. Levi and M. Manera and A. Meisner and R. Miquel and J. Moustakas and A. D. Myers and N. Palanque-Delabrouille and F. Prada and I. Pérez-Ràfols and G. Rossi and E. Sanchez and C. Saulder and D. Schlegel and M. Schubnell and H. Seo and F. Sinigaglia and D. Sprayberry and T. Tan and G. Tarlé and B. A. Weaver and S. Youles and J. Yu and R. Zhou and H. Zou},
journal= {arXiv preprint arXiv:2505.01596},
year = {2025}
}
Comments
26 pages, 10 figures. Prepared for submission to JCAP