Reinforcement learning for machine learning models using dynamic confidence thresholds

Abstract

Various embodiments of the present disclosure provide reinforcement learning for machine learning using dynamic confidence thresholds. In one example, an embodiment provides for generating a plurality of training datasets for a machine learning model by augmenting a labeled dataset for the machine learning model with a synthetic labeled dataset, generating a plurality of retrained model versions of the machine learning model based on the plurality of training datasets, generating a reward indicator for a retrained model version of the plurality of retrained versions of the machine learning model based on a comparison between a validation dataset for the machine learning model and a respective output dataset for the retrained model version, and modifying the defined confidence threshold based on the reward indicator for the retrained model version to generate a modified confidence threshold for the machine learning model.