Using Active Learning to Improve Quasar Identification for the DESI Spectra Processing Pipeline
Dylan Green, David Kirkby, J. Aguilar, S. Ahlen, D. M. Alexander, E. Armengaud, S. Bailey, A. Bault, D. Bianchi, A. Brodzeller, D. Brooks, T. Claybaugh, R. de Belsunce, A. de la Macorra, P. Doel, V. A. Fawcett, S. Ferraro, A. Font-Ribera, J. E. Forero-Romero, E. Gaztañaga, S. Gontcho A Gontcho, G. Gutierrez, M. Ishak, S. Juneau, R. Kehoe, T. Kisner, A. Kremin, A. Lambert, M. Landriau, L. Le Guillou, M. E. Levi, M. Manera, A. Meisner, R. Miquel, J. Moustakas, A. D. Myers, N. Palanque-Delabrouille, F. Prada, I. Pérez-Ràfols, G. Rossi, E. Sanchez, C. Saulder, D. Schlegel, M. Schubnell, H. Seo, F. Sinigaglia, D. Sprayberry, T. Tan, G. Tarlé, B. A. Weaver, S. Youles, J. Yu, R. Zhou, H. Zou
Published: 2025/5/2
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.