Theory-guided neural network for studying the ground state of 2D spin-orbit coupled Bose–Einstein condensates

Machine learning has been a powerful tool to study various models in many fields, which requires plenty of data to ensure accuracy. It is a challenge to reduce the dependence on the amount of data and ensure that the predictions are both accurate and physical. We develop a theory-guided neural network (TgNN) to explore the ground states of two-component spin–orbit coupled Bose–Einstein condensation in two-dimension. Only randomly selected few data from the twelve ground states as training data, TgNN performs higher accuracy, broader generalization ability and stronger robustness than deep neural network (DNN). Moreover, in the neighborhood of the phase transition where the predictions of DNN have obvious deviations, while TgNN still keeps relative accuracy. TgNN is less dependent on data and exhibits potential for high-dimensional and complicated models. • TgNN accurately predicts all ground states of 2D SOC BEC, which uses only one-tenth of the total data. • TgNN provides higher accuracy, broader generality ability and stronger robustness than DNN. • TgNN has more advantages than DNN in solving critical problems.