Automatic data curation
AssigneeScale AI, Inc.
InventorsDiego Ardila, Russell Kaplan, Vinjai Saraj Vale, Jihan Yin
Life of this patent
Estimating the term…
Abstract
One embodiment of the present invention sets forth a technique for curating a data sample set. The technique includes determining one or more data sampling criteria based on a sampling objective for a data sample set associated with the machine learning model. The technique also includes selecting, from a set of unlabeled data samples, at least one data sample to be labeled and added to a data sample set associated with the machine learning model based on the one or more data sampling criteria. The technique also includes, for each selected data sample, supplementing the data sample set with the selected data sample and at least one association with a label.
Claims15
A computer-implemented method for curating a data sample set, the method comprising:
- generating a set of outputs by processing at least one data sample using a trained machine learning model;
- concurrently determining on a plurality of computing instances a plurality of relevance scores between a plurality of sampling objectives and the set of outputs generated using the trained machine learning model, wherein each sampling objective in the plurality of sampling objectives is associated with a different computing instance in the plurality of computing instances and is associated with a different objective that is to be achieved during retraining of the trained machine learning model, and wherein each relevance score represents a relevance of a given sampling objective in the plurality of sampling objectives to the trained machine learning model;
- determining a given sampling objective for the data sample set that is to be achieved during retraining of the trained machine learning model, wherein the determining of the given sampling objective is based on the plurality of relevance scores;
- determining one or more data sampling criteria based on the given sampling objective;
- selecting, from a set of unlabeled data samples, at least one data sample for labeling and adding to the data sample set based on the one or more data sampling criteria;
- for each selected data sample, supplementing the data sample set with the selected data sample and at least one association with a label;
- and
- retraining the trained machine learning model using the supplemented data sample set.
One or more non-transitory computer readable media storing instructions that, when executed by one or more processors, cause the one or more processors to perform the steps of:
- generating a set of outputs by processing at least one data sample using a trained machine learning model;
- concurrently determining on a plurality of computing instances a plurality of relevance scores between a plurality of sampling objectives and the set of outputs generated using the trained machine learning model, wherein each sampling objective in the plurality of sampling objectives is associated with a different computing instance in the plurality of computing instances and is associated with a different objective that is to be achieved during retraining of the trained machine learning model, and wherein each relevance score represents a relevance of a given sampling objective in the plurality of sampling objectives to the trained machine learning model;
- determining a given sampling objective for the data sample set that is to be achieved during retraining of the trained machine learning model, wherein the determining of the given sampling objective is based on the plurality of relevance scores;
- determining one or more data sampling criteria based on the given sampling objective;
- selecting, from a set of unlabeled data samples, at least one data sample for labeling and adding to the data sample set based on the one or more data sampling criteria;
- for each selected data sample, supplementing the data sample set with the selected data sample and at least one association with a label;
- and
- retraining the trained machine learning model using the supplemented data sample set.
A system, comprising:
- a memory that stores instructions, and
- a processor that is coupled to the memory and, when executing the instructions, is configured to:
- generate a set of outputs by processing at least one data sample using a trained machine learning model;
- concurrently determine on a plurality of computing instances a plurality of relevance scores between a plurality of sampling objectives and the set of outputs generated using the trained machine learning model, wherein each sampling objective in the plurality of sampling objectives is associated with a different computing instance in the plurality of computing instances and is associated with a different objective that is to be achieved during retraining of the trained machine learning model, and wherein each relevance score represents a relevance of a given sampling objective in the plurality of sampling objectives to the trained machine learning model;
- determine a given sampling objective for the data sample set that is to be achieved during retraining of the trained machine learning model, wherein the determining of the given sampling objective is based on the plurality of relevance scores;
- determine one or more data sampling criteria based on the given sampling objective;
- select, from a set of unlabeled data samples, at least one data sample for labeling and adding to the data sample set based on the one or more data sampling criteria;
- for each selected data sample, supplement the data sample set with the selected data sample and at least one association with a label;
- and
- retraining the trained machine learning model using the supplemented data sample set.
Dependencies are inferred from the claim text.