TRAINING A MACHINE-LEARNING MODEL FOR CONSTRAINT-COMPLIANCE PREDICTION USING AN ACTION-BASED LOSS FUNCTION
An online system trains a constraint prediction machine-learning model using an action-based loss function. The action-based loss function computes a weighted sum of (1) an accuracy of the constraint prediction model in predicting whether a user's interaction complies with a set of constraints and (2) an action score representing a number of actions taken by users to determine whether the user's action complies with the set of constraints. The online system may apply this trained constraint prediction model to future interaction data received from third-party systems to predict whether user interactions with those third-party systems comply with the set of constraints.
1 . A non-transitory computer-readable medium storing parameters for a constraint prediction model, wherein the parameters for the constraint prediction model are produced by a process comprising:
initializing a set of parameters for the constraint prediction model;
accessing training data for the constraint prediction model, wherein the training data comprises a plurality of training examples, wherein each training example of the plurality of training examples comprises:
interaction data describing an interaction of a user with a third-party system; and
a label for the training example that indicates whether the interaction of the user with the third-party system complies with a set of constraints;
applying the constraint prediction model to the interaction data of each training example of the plurality of training examples to generate a set of constraint predictions for the plurality of training examples, wherein each constraint prediction indicates a predicted likelihood that the interaction of the corresponding training example complies with the set of constraints;
accessing action logs from an online system that describe actions taken by users of the online system to determine whether interactions of the plurality of training examples comply with the set of constraints;
computing an action score for each training example of the plurality of training examples based on the action logs, wherein each action score represents a number of actions taken by operators of the online system to determine whether the interaction of the corresponding training example complies with the set of constraints;
computing a loss score for each of the plurality of training examples by applying an action-based loss function to the constraint prediction, action score, and label for each of the plurality of training examples;
updating the set of parameters for the constraint prediction model through a backpropagation process using the computed loss scores for the plurality of training examples; and
storing a final set of parameters for the constraint prediction model to the computer-readable medium.
2 . The computer-readable medium of claim 1 , wherein the constraint prediction model comprises a neural network with a plurality of layers, wherein the plurality of layers comprises an input layer, a set of intermediate layers, and an output layer, wherein the input layer is connected to the set of intermediate layers, and the set of intermediate layers are connected to the output layer, and wherein applying the constraint prediction model to the interaction data of a training example comprises:
inputting, to the input layer, a feature vector associated with the interaction data of the training data;
receiving a constraint prediction from the output layer.
3 . The computer-readable medium of claim 1 , wherein computing the action score for a training example of the plurality of training examples comprises:
identifying a set of actions described in the action logs that are associated with the interaction associated with the interaction data of the training example.
4 . The computer-readable medium of claim 3 , wherein computing the action score for a training example of the plurality of training examples comprises:
assigning a score to each action of the identified set of actions; and
computing an aggregated score based on the assigned scores.
5 . The computer-readable medium of claim 1 , wherein applying the action-based loss function to a constraint prediction, action score, and label comprises:
computing a weighted sum based on the constraint prediction and the action score.
6 . The computer-readable medium of claim 1 , wherein accessing the action logs comprises:
receiving action logs from another third-party system, wherein the action logs describe actions of users on that third-party system.
7 . A method comprising:
receiving interaction data from a third-party system describing an interaction of the user with the third-party system;
applying a constraint prediction model to the interaction data by accessing parameters for the constraint prediction model stored by the non-transitory computer-readable medium of claim 1 ; and
predicting whether the interaction complies with a set of constraints based on the application of the constraint prediction model to the interaction data.
8 . A method comprising:
initializing a set of parameters for the constraint prediction model;
accessing training data for the constraint prediction model, wherein the training data comprises a plurality of training examples, wherein each training example of the plurality of training examples comprises:
interaction data describing an interaction of a user with a third-party system; and
a label for the training example that indicates whether the interaction of the user with the third-party system complies with a set of constraints;
applying the constraint prediction model to the interaction data of each training example of the plurality of training examples to generate a set of constraint predictions for the plurality of training examples, wherein each constraint prediction indicates a predicted likelihood that the interaction of the corresponding training example complies with the set of constraints;
accessing action logs from an online system that describe actions taken by users of the online system to determine whether interactions of the plurality of training examples comply with the set of constraints;
computing an action score for each training example of the plurality of training examples based on the action logs, wherein each action score represents a number of actions taken by operators of the online system to determine whether the interaction of the corresponding training example complies with the set of constraints;
computing a loss score for each of the plurality of training examples by applying an action-based loss function to the constraint prediction, action score, and label for each of the plurality of training examples;
updating the set of parameters for the constraint prediction model through a backpropagation process using the computed loss scores for the plurality of training examples; and
storing a final set of parameters for the constraint prediction model to a non-transitory computer-readable medium.
9 . The method of claim 8 , wherein the constraint prediction model comprises a neural network with a plurality of layers, wherein the plurality of layers comprises an input layer, a set of intermediate layers, and an output layer, wherein the input layer is connected to the set of intermediate layers, and the set of intermediate layers are connected to the output layer, and wherein applying the constraint prediction model to the interaction data of a training example comprises:
inputting, to the input layer, a feature vector associated with the interaction data of the training data;
receiving a constraint prediction from the output layer.
10 . The method of claim 8 , wherein computing the action score for a training example of the plurality of training examples comprises:
identifying a set of actions described in the action logs that are associated with the interaction associated with the interaction data of the training example.
11 . The method of claim 10 , wherein computing the action score for a training example of the plurality of training examples comprises:
assigning a score to each action of the identified set of actions; and
computing an aggregated score based on the assigned scores.
12 . The method of claim 8 , wherein applying the action-based loss function to a constraint prediction, action score, and label comprises:
computing a weighted sum based on the constraint prediction and the action score.
13 . The method of claim 8 , wherein accessing the action logs comprises:
receiving action logs from another third-party system, wherein the action logs describe actions of users on that third-party system.
14 . A method of claim 8 , further comprising:
receiving interaction data from a third-party system describing an interaction of the user with the third-party system;
applying a constraint prediction model to the interaction data by accessing parameters for the constraint prediction model stored by the non-transitory computer-readable medium; and
predicting whether the interaction complies with a set of constraints based on the application of the constraint prediction model to the interaction data.
15 . A non-transitory computer-readable medium storing instructions that, when executed by a processor, cause the processor to perform operations comprising:
initializing a set of parameters for the constraint prediction model;
accessing training data for the constraint prediction model, wherein the training data comprises a plurality of training examples, wherein each training example of the plurality of training examples comprises:
interaction data describing an interaction of a user with a third-party system; and
a label for the training example that indicates whether the interaction of the user with the third-party system complies with a set of constraints;
applying the constraint prediction model to the interaction data of each training example of the plurality of training examples to generate a set of constraint predictions for the plurality of training examples, wherein each constraint prediction indicates a predicted likelihood that the interaction of the corresponding training example complies with the set of constraints;
accessing action logs from an online system that describe actions taken by users of the online system to determine whether interactions of the plurality of training examples comply with the set of constraints;
computing an action score for each training example of the plurality of training examples based on the action logs, wherein each action score represents a number of actions taken by operators of the online system to determine whether the interaction of the corresponding training example complies with the set of constraints;
computing a loss score for each of the plurality of training examples by applying an action-based loss function to the constraint prediction, action score, and label for each of the plurality of training examples;
updating the set of parameters for the constraint prediction model through a backpropagation process using the computed loss scores for the plurality of training examples; and
storing a final set of parameters for the constraint prediction model to the computer-readable medium.
16 . The computer-readable medium of claim 15 , wherein the constraint prediction model comprises a neural network with a plurality of layers, wherein the plurality of layers comprises an input layer, a set of intermediate layers, and an output layer, wherein the input layer is connected to the set of intermediate layers, and the set of intermediate layers are connected to the output layer, and wherein applying the constraint prediction model to the interaction data of a training example comprises:
inputting, to the input layer, a feature vector associated with the interaction data of the training data;
receiving a constraint prediction from the output layer.
17 . The computer-readable medium of claim 15 , wherein computing the action score for a training example of the plurality of training examples comprises:
identifying a set of actions described in the action logs that are associated with the interaction associated with the interaction data of the training example.
18 . The computer-readable medium of claim 17 , wherein computing the action score for a training example of the plurality of training examples comprises:
assigning a score to each action of the identified set of actions; and
computing an aggregated score based on the assigned scores.
19 . The computer-readable medium of claim 15 , wherein applying the action-based loss function to a constraint prediction, action score, and label comprises:
computing a weighted sum based on the constraint prediction and the action score.
20 . The computer-readable medium of claim 15 , wherein accessing the action logs comprises:
receiving action logs from another third-party system, wherein the action logs describe actions of users on that third-party system.