Systems and methods for generating augmented training data for machine learning models

Abstract

Methods and systems are provided for augmenting ultrasound image training data, which may be used to train one or more machine learning models. One example method for augmenting ultrasound training data comprises, selecting an ultrasound image and a ground truth output corresponding to the ultrasound image, determining a first modification to apply to the ultrasound image, applying the first modification to the ultrasound image to produce an augmented ultrasound image, modifying the ground truth output based on the first modification to produce an augmented ground truth output corresponding to the augmented ultrasound image, and training a machine learning model using the augmented ultrasound image and the augmented ground truth output. In this way, a machine learning model may learn a more robust mapping from ultrasound image features to expected output, with less probability of overfitting, and with increased generalizability to noisy ultrasound images, or ultrasound images containing artifacts.