Walmart Inc.
SYSTEMS AND METHODS FOR PROVIDING PLUG-AND-PLAY FRAMEWORKS FOR TRAINING MODELS USING SEMI-SUPERVISED LEARNING TECHNIQUES

Abstract:

Systems and methods including one or more processors and one or more non-transitory storage devices storing computing instructions configured to run on the one or more processors and perform acts of providing a semi-supervised learning abstraction model that includes an API; receiving, via the API, pre-training parameters at least identifying (a) a first set of unlabeled images and (b) an encoder model selected from the plurality of encoder models; executing a pre-training procedure that trains the encoder model using the first set of unlabeled images; receiving, via the API, supervised training parameters at least identifying (a) a second set of labeled images and (b) the encoder model that is pre-trained using the pre-training procedure; executing a supervised training procedure that further trains the encoder model using the second set of labeled images; and storing a encoder model checkpoint for the encoder model. Other embodiments are disclosed herein.