Estimating reservoir production rates using machine learning models for wellbore operation control

A system is described for estimating well production and injection rates of a subterranean reservoir using machine learning models. The system may include a processor and a non-transitory computer-readable medium comprising instructions that are executable by the processor to cause the processor to perform various operations. The processor may receive a set of static geological data about at least one subterranean reservoir in a subterranean formation. The processor may apply a trained convolutional neural network to the set of static geological data and data on initial states of dynamic reservoir properties to determine dynamic outputs of the subterranean reservoir. The processor may determine well data by extracting the set of static geological data and the dynamic outputs at well trajectories. And, the processor may apply a trained artificial neural network to the well data and subterranean grid information about the subterranean reservoir to generate estimated well production and injection rates.