USING LIGHTWEIGHT MACHINE-LEARNING MODEL ON SMART NIC
Some embodiments provide a method for using a machine learning (ML) model to respond to a query, at a smart NIC of a computer. The method receives a query including an input. The method applies a first ML model to the input to generate an output and a confidence measure for the output. When the confidence measure for the output is below a threshold, the method discards the output and provides the query to the computer for the computer to apply a second ML model to the input.
1 . A method for using a machine learning (ML) model to respond to a query, the method comprising:
at a smart NIC of a computer:
receiving a query comprising an input;
applying a first ML model to the input to generate an output and a confidence measure for the output; and
when the confidence measure for the output is below a threshold, discarding the output and providing the query to the computer for the computer to apply a second ML model to the input.
2 . The method of claim 1 , wherein:
the first ML model is a first neural network; and
the second ML model is a second neural network trained with a same dataset as the first neural network.
3 . The method of claim 2 , wherein the first neural network has fewer nodes than the second neural network.
4 . The method of claim 2 , wherein:
the first neural network comprises a first set of weight parameters;
the second neural network comprises a second set of weight parameters; and
a greater percentage of weight parameters are equal to zero in the first set of weight parameters than in the second set of weight parameters.
5 . The method of claim 1 , wherein:
the first and second ML models are random forest (RF) models for classifying inputs;
the second RF model comprises a particular number of decision trees; and
the first RF model comprises only a subset of the decision trees of the second version.
6 . The method of claim 1 , wherein:
the first and second ML models are boosting models for classifying inputs;
the second boosting model comprises a particular number of decision trees; and
the first boosting model comprises only a subset of the decision trees of the second version.
7 . The method of claim 1 , wherein the first ML model is a first type of model and the second ML model is a second, different type of model.
8 . The method of claim 1 , wherein the output of each of the first and second ML models comprises a classification of the input into one category of a plurality of categories.
9 . The method of claim 8 , wherein the confidence measure comprises a probability that the classification by the first ML model identifies a correct category for the input.
10 . The method of claim 1 , wherein the output of each of the first and second ML models comprises a classification of the input into one or more categories of a plurality of categories.
11 . The method of claim 1 , wherein when the confidence measure for the output is above the threshold, the smart NIC provides the output generated by the first ML model as a response to the query without providing the input to the server.
12 . The method of claim 11 , wherein the smart NIC providing the response to the query without providing the input to the computer enables the response to the query to be sent faster than when the smart NIC provides the query to the computer.
13 . A non-transitory machine-readable medium storing a program for execution by at least one processing unit of a smart network interface card (NIC) of a computer, the program for using a machine learning (ML) model to respond to a query, the program comprising sets of instructions for:
receiving a query comprising an input;
applying a first ML model to the input to generate an output and a confidence measure for the output; and
when the confidence measure for the output is below a threshold, discarding the output and providing the query to the computer for the computer to apply a second ML model to the input.
14 . The non-transitory machine-readable medium of claim 13 , wherein:
the first ML model is a first neural network;
the second ML model is a second neural network trained with a same dataset as the first neural network; and
the first neural network has fewer nodes than the second neural network.
15 . The non-transitory machine-readable medium of claim 13 , wherein:
the first ML model is a first neural network comprising a first set of weight parameters;
the second ML model is a second neural network, trained with a same dataset as the first neural network, comprising a second set of weight parameters; and
a greater percentage of weight parameters are equal to zero in the first set of weight parameters than in the second set of weight parameters.
16 . The non-transitory machine-readable medium of claim 13 , wherein:
the first and second ML models are random forest (RF) models for classifying inputs;
the second RF model comprises a particular number of decision trees; and
the first RF model comprises only a subset of the decision trees of the second version.
17 . The non-transitory machine-readable medium of claim 13 , wherein:
the first and second ML models are boosting models for classifying inputs;
the second boosting model comprises a particular number of decision trees; and
the first boosting model comprises only a subset of the decision trees of the second version.
18 . The non-transitory machine-readable medium of claim 13 , wherein:
the output of each of the first and second ML models comprises a classification of the input into one category of a plurality of categories; and
the confidence measure comprises a probability that the classification by the first ML model identifies a correct category for the input.
19 . The non-transitory machine-readable medium of claim 13 , wherein the program further comprises a set of instructions for providing the output generated by the first ML model as a response to the query without providing the input to the server when the confidence measure for the output is above the threshold.
20 . The non-transitory machine-readable medium of claim 13 , wherein the program is executed by a central processing unit of the smart NIC.
21 . The non-transitory machine-readable medium of claim 20 , wherein the set of instructions for applying the first ML model to the input comprises a set of instructions for providing the input to a separate processing unit of the smart NIC that executes the lightweight ML model and receiving the output and the confidence measure from the separate processing unit.