US Patent 11790231 Determining optimal augmentations for a training data set

A computer-implemented method according to one embodiment includes applying a predetermined augmentation to the sample set of training data to create an augmented sample set, training a model with the augmented sample set, determining a performance of the trained model, and assigning a weight to the predetermined augmentation for the training data set based on the determined performance. A determination is made as to whether to apply the predetermined augmentation to a larger training data set before the training data set is applied to the model, based on the weight assigned to the predetermined augmentation.