A White Dwarf Search Model Based on a Deep Transfer-learning Method

Abstract

Abstract White dwarfs represent the ultimate stage of evolution for over 97% of stars and play a crucial role in studies of the Milky Way’s structure and evolution. Recent years have witnessed significant progress in using deep-learning methods for identifying unique objects in large-scale data. In this paper, we present a model based on transfer learning for identifying white dwarfs. We constructed a data set using the spectra released by LAMOST DR9 and trained a convolutional neural network model. The model was then further trained using a transfer-learning approach for a binary classification model. Our final model is comprised of a seven-class classification model and a binary classification model. The testing set yielded an accuracy rate of 96.08%. Our proposed model successfully identifies 4314 of the 4479 white dwarfs published in previous papers. We applied this model to filter the 1,121,128 spectral data from the LAMOST DR9 V1 catalog. Subsequently, we obtained 6317 white dwarf candidates, of which 5014 were cross-validated and found to be known white dwarfs. We finally identified 489 new white dwarfs out of the remaining 1303 candidates, containing 377 DAs, 1 DB, 4 DZs, 1 magnetic WD, 101 DA+M binaries, and 1 DB+M binary. Our study also compared transfer-learning methods with non-transfer-learning methods, and the results show that transfer learning provides faster training speed and a higher accuracy rate. We provide the trained model and a corresponding usage program for subsequent studies.