Automated machine learning model selection
AssigneeInternational Business Machines Corporation
InventorsAna Paula Appel, Renato Luiz de Freitas Cunha, Bruno Silva, PAULO RODRIGO CAVALIN
Life of this patent
Estimating the term…
Abstract
An approach to identifying architectures of machine learning models meeting a user defined constraint. The approach can receive input associated with evaluating machine learning models from a user. The approach can determine acceptable architectural templates to evaluate the machine learning models based on the input and determine a list of architectures and metrics based on a calculation of maximum neural network sizes of the acceptable architectural templates not exceeding the constraint. The approach can send the list of architectures and metrics to the user for selection.
Claims20
A computer-implemented method for identifying architectures of machine learning models meeting a user defined constraint, the computer-implemented method comprising:
- receiving, by one or more processors, input associated with evaluating machine learning models from a user;
- determining, by the one or more processors, acceptable architectural templates to evaluate the machine learning models based on the input;
- determining, by the one or more processors, a first list of architectures and metrics based on a calculation of maximum neural network sizes of the acceptable architectural templates not exceeding the constraint;
- and
- sending, by the one or more processors, the first list of architectures and metrics to the user for selection.
A computer program product for identifying architectures of machine learning models meeting a user defined constraint, the computer program product comprising:
- one or more non-transitory computer readable storage media and program instructions stored on the one or more non-transitory computer readable storage media, the program instructions comprising:
- program instructions to receive input associated with evaluating machine learning models from a user;
- program instructions to determine acceptable architectural templates to evaluate the machine learning models based on the input;
- program instructions to determine a first list of architectures and metrics based on a calculation of maximum neural network sizes of the acceptable architectural templates not exceeding the constraint;
- and
- program instructions to send the first list of architectures and metrics to the user for selection.
A computer system for identifying architectures of machine learning models meeting a user defined constraint, the computer system comprising:
- one or more computer processors;
- one or more computer readable storage media;
- and
- program instructions stored on the one or more computer readable storage media for execution by at least one of the one or more processors, the program instructions comprising:
- program instructions to receive input associated with evaluating machine learning models from a user;
- program instructions to determine acceptable architectural templates to evaluate the machine learning models based on the input;
- program instructions to determine a first list of architectures and metrics based on a calculation of maximum neural network sizes of the acceptable architectural templates not exceeding the constraint;
- and
- program instructions to send the first list of architectures and metrics to the user for selection.
Dependencies are inferred from the claim text.