RESOURCE EFFICIENT TRAINING OF MACHINE LEARNING MODELS THAT PREDICT STOCHASTIC SPREAD

Methods, systems, and apparatus for obtaining input features representative of a region of space, processing an input comprising the input features through the ML model to generate a prediction describing predicted features of the region of space, obtaining result features describing the region of space, determining a value of at least one evaluation metric that relates the predicted features and the result features, that at least one evaluation metric including one of a distance score, a pyramiding density error, and min-max intersection over union (IOU) score, and training the ML model responsive to the at least one evaluation metric. Other implementations of this aspect include corresponding systems, apparatus, and computer programs, configured to perform the actions of the methods, encoded on computer storage devices.