TUNING HYPERPARAMETERS FOR POSTPROCESSING OUTPUT OF MACHINE LEARNING MODELS
A system performs tuning of hyperparameters used for postprocessing of outputs of machine learning models. The system initializes a population of vectors representing values of postprocessing hyperparameters. The system repeatedly modifies the population by adding and removing members of the population. A fitness metric is used to identify vectors that are removed from the population. The system selects a vector from the population of vectors based on the fitness metric values and uses the values of postprocessing hyperparameters from the vector for postprocessing output of the machine learning model.
1 . A computer-implemented method for tuning postprocessing hyperparameters of a machine learning model, the computer-implemented method comprising:
initializing a population of vectors, each vector representing values of postprocessing hyperparameters for processing output score of the machine learning model;
repeating a plurality of times:
adding one or more vectors to the population of vectors, the one or more vectors generated from existing vectors of the population;
determining a fitness metric value for each of the population of vectors, comprising:
executing the machine learning model for an input data to obtain an output score, the input data associated with a ground truth label;
processing the output score using postprocessing hyperparameters represented by the vector to obtain a final output result; and
determining the fitness metric value based on a difference between the final output result and the ground truth label;
removing one or more vectors from the population based on fitness metric values of the one or more vectors;
selecting a vector from the population of vectors based on the fitness metric values; and
using values of postprocessing hyperparameters from the vector selected for postprocessing of output of the machine learning model.
2 . The computer-implemented method of claim 1 , wherein the machine learning model classifies an input data to one of a plurality of categories, wherein the output score of the machine learning model is mapped to a category based on a plurality of thresholds, each threshold representing a postprocessing hyperparameter.
3 . The computer-implemented method of claim 2 , wherein each vector of the population of vectors represents a plurality of values, each value corresponding to a threshold from the plurality of thresholds.
4 . The computer-implemented method of claim 1 , wherein the fitness metric value is determined as an aggregate measure of difference between each final output result and corresponding ground truth label for a plurality of samples, each sample representing an input data with a ground truth label.
5 . The computer-implemented method of claim 1 , wherein generating one or more vectors from existing vectors of the population comprises, identifying a vector from the population of vectors and modifying one or more elements of the vector to obtain a new vector.
6 . The computer-implemented method of claim 1 , wherein generating one or more vectors from existing vectors of the population comprises, identifying a first vector and a second vector from the population of vectors and swapping an element of the first vector with corresponding element of the second vector to obtain a modified first vector and a modified second vector.
7 . The computer-implemented method of claim 1 , wherein a postprocessing hyperparameter represents size of a contour for filtering out noise from the output of the machine learning model.
8 . A non-transitory computer readable storage medium storing instructions that when executed by one or more computer processors, cause the one or more computer processors to perform steps for tuning postprocessing hyperparameters of a machine learning model, the steps comprising:
initializing a population of vectors, each vector representing values of postprocessing hyperparameters for processing output score of the machine learning model;
repeating a plurality of times:
adding one or more vectors to the population of vectors, the one or more vectors generated from existing vectors of the population;
determining a fitness metric value for each of the population of vectors, comprising:
executing the machine learning model for an input data to obtain an output score, the input data associated with a ground truth label;
processing the output score using postprocessing hyperparameters represented by the vector to obtain a final output result; and
determining the fitness metric value based on a difference between the final output result and the ground truth label;
removing one or more vectors from the population based on fitness metric values of the one or more vectors;
selecting a vector from the population of vectors based on the fitness metric values; and
using values of postprocessing hyperparameters from the vector selected for postprocessing of output of the machine learning model.
9 . The non-transitory computer readable storage medium of claim 8 , wherein the machine learning model classifies an input data to one of a plurality of categories, wherein the output score of the machine learning model is mapped to a category based on a plurality of thresholds, each threshold representing a postprocessing hyperparameter.
10 . The non-transitory computer readable storage medium of claim 9 , wherein each vector of the population of vectors represents a plurality of values, each value corresponding to a threshold from the plurality of thresholds.
11 . The non-transitory computer readable storage medium of claim 8 , wherein the fitness metric value is determined as an aggregate measure of difference between each final output result and corresponding ground truth label for a plurality of samples, each sample representing an input data with a ground truth label.
12 . The non-transitory computer readable storage medium of claim 8 , wherein generating one or more vectors from existing vectors of the population comprises, identifying a vector from the population of vectors and modifying one or more elements of the vector to obtain a new vector.
13 . The non-transitory computer readable storage medium of claim 8 , wherein generating one or more vectors from existing vectors of the population comprises, identifying a first vector and a second vector from the population of vectors and swapping an element of the first vector with corresponding element of the second vector to obtain a modified first vector and a modified second vector.
14 . The non-transitory computer readable storage medium of claim 8 , wherein a postprocessing hyperparameter represents size of a contour for filtering out noise from the output of the machine learning model.
15 . A computer system comprising:
one or more computer processors; and
a non-transitory computer readable storage medium storing instructions that when executed by one or more computer processors, cause the one or more computer processors to perform steps for tuning postprocessing hyperparameters of a machine learning model, the steps comprising:
initializing a population of vectors, each vector representing values of postprocessing hyperparameters for processing output score of the machine learning model;
repeating a plurality of times:
adding one or more vectors to the population of vectors, the one or more vectors generated from existing vectors of the population;
determining a fitness metric value for each of the population of vectors, comprising:
executing the machine learning model for an input data to obtain an output score, the input data associated with a ground truth label;
processing the output score using postprocessing hyperparameters represented by the vector to obtain a final output result; and
determining the fitness metric value based on a difference between the final output result and the ground truth label;
removing one or more vectors from the population based on fitness metric values of the one or more vectors;
selecting a vector from the population of vectors based on the fitness metric values; and
using values of postprocessing hyperparameters from the vector selected for postprocessing of output of the machine learning model.
16 . The computer system of claim 15 , wherein the machine learning model classifies an input data to one of a plurality of categories, wherein the output score of the machine learning model is mapped to a category based on a plurality of thresholds, each threshold representing a postprocessing hyperparameter.
17 . The computer system of claim 16 , wherein each vector of the population of vectors represents a plurality of values, each value corresponding to a threshold from the plurality of thresholds.
18 . The computer system of claim 15 , wherein the fitness metric value is determined as an aggregate measure of difference between each final output result and corresponding ground truth label for a plurality of samples, each sample representing an input data with a ground truth label.
19 . The computer system of claim 15 , wherein generating one or more vectors from existing vectors of the population comprises, identifying a vector from the population of vectors and modifying one or more elements of the vector to obtain a new vector.
20 . The computer system of claim 15 , wherein generating one or more vectors from existing vectors of the population comprises, identifying a first vector and a second vector from the population of vectors and swapping an element of the first vector with corresponding element of the second vector to obtain a modified first vector and a modified second vector.