Automatic Estimation of Stellar Atmospheric Parameters Based on Extremely Randomized Trees

Accurate determination of the stellar atmospheric physical parameters is crucial for Galactic archaeology via large-scale spectroscopic surveys. Because of the high dimension of celestial spectra, the computational complexity of processing them is relatively high. In this paper, we use the physical signature of each stellar spectrum—line indices as the input features of the Extremely Randomized Trees model (ERT) to estimate the atmospheric physical parameters. We firstly do the correlation analysis between the 26 line indices with the three stellar atmospheric parameters— effective temperature Teff, surface gravity log g and metallicity [Fe/H], respectively. And then, according to the correlation analysis results, we use different combinations of the line indices as the input features of the ERT model to estimate the three parameters. Finally, we analyze the interaction of the peak wavelengths for the three physical parameters, and we found that it could affect the estimation precision of the Teff and [Fe/H], but not the log g. Base on the above the analysis results, we obtained the three parameters with a precision of 59.17 K for Teff, 0.071 dex for log g, 0.0971 dex for [Fe/H]. At the same time, we design a new method to estimate three physical parameters. The results show that this method can be applied to estimate the stellar atmospheric physical parameters effectively.