International Business Machines Corporation
LOSS AUGMENTATION FOR PREDICTIVE MODELING

Abstract:

A machine learning system that incorporates arbitrary constraints into deep learning model is provided. The machine learning system provides a set of penalty data points en a set of arbitrary constraints in addition to a set of original training data points. The machine learning system assigns a penalty to each penalty data point in the set of penalty data points. The machine learning system optimizes a machine learning model by solving an objective function based on an original loss function and a penalty loss function. The original loss function is evaluated over a set of original training data points and the penalty loss function is evaluated over the set of penalty data points. The machine learning system provides the optimized machine learning model based on a solution of the objective function.