US Patent 11954572 Optimizing training data for image classification

A method for machine learning-based classification may include training a machine learning model with a full training data set, the full training data set comprising a plurality of data points, to generate a first model state of the machine learning model, generating respective embeddings for the data points in the full training data set with the first model state of the machine learning model, applying a clustering algorithm to the respective embeddings to generate one or more clusters of the embeddings, identifying outlier embeddings from the one or more clusters of the embeddings, generating a reduced training data set comprising the full training data set less the data points associated with the outlier embeddings, training the machine learning model with the reduced training data set to a second model state, and applying the second model state to one or more data sets to classify the one or more data sets.