Learning impurity spectral functions from density of states.

Using numerical renormalization group calculation, we construct a dataset with 100 K samples, and train six different neural networks for the prediction of spectral functions from density of states (DOS) of the host material. We find that a combination of gated recurrent unit (GRU) network and bidirectional GRU (BiGRU) performances the best among all the six neural networks. The mean absolute error of the GRU + BiGRU network can reach 0.052 and 0.043 when this network is evaluated on the original dataset and two other independent datasets. The average time of spectral function predictions from machine learning is on the scale of 10-10that of traditional impurity solvers for Anderson impurity model. This investigation pave the way for the application of recurrent neural network and convolutional neural network in the prediction of spectral functions from DOSs in machine learning solvers of magnetic impurity problems.