Machine learning model correction

Abstract

A method, in some implementations, may include obtaining output from a machine learning (ML) model responsive to input data, obtaining initial training data representing training data used to train the ML model, generating, based on the output from the ML model and the initial training data, correction training data that represents a desired alteration to the output from the ML model responsive to one or more particular subgroups in the input data, generating, based on the correction training data, a correction ML model configured to receive, as input, the input data and to output correction values which, when combined with the output from the ML model, perform the desired alteration, and generating corrected output as a combination of the output from the ML model and the output correction values from the correction ML model, and providing, for display, the corrected output.