PRIVACY PRESERVING MACHINE LEARNING VIA GRADIENT BOOSTING

Information

  • Patent Application
  • 20230034384
  • Publication Number
    20230034384
  • Date Filed
    October 08, 2021
    2 years ago
  • Date Published
    February 02, 2023
    a year ago
Abstract
This describes a privacy preserving machine learning platform. In one aspect, a method includes receiving, by a first computing system of multiple multi-party computation (MPC) systems, an inference request including a first share of a given user profile. A predicted label for the given user profile is determined based at least in part on a first machine learning model. A predicted residue value for the given user profile indicating a predicted error in the predicted label is determined. The first computing system determines the first share of the predicted residue value for the given user profile based at least in part on the first share of the given user profile and a second machine learning model. The first computing system receives, from a second computing system of the MPC computing systems, data indicating the second share of the predicted residue value for the given user profile.
Description
CROSS-REFERENCE TO RELATED APPLICATION

This application claims priority to IL Application No. 277910, filed Oct. 9, 2020. The disclosure of the foregoing application is hereby incorporated by reference in its entirety.


TECHNICAL FIELD

This specification relates to a privacy preserving machine learning platform that trains and uses machine learning models using secure multi-party computation.


BACKGROUND

Some machine learning models are trained based on data collected from multiple sources, e.g., across multiple websites and/or native applications. However, this data may include private or sensitive data that should not be shared or allowed to leak to other parties.


SUMMARY

In general, one innovative aspect of the subject matter described in this specification can be embodied in methods that include receiving, by a first computing system of multi-party computation (MPC) computing systems, an inference request that includes a first share of a given user profile; determining a predicted label for the given user profile based at least in part on a first machine learning model trained using multiple user profiles; determining a predicted residue value for the given user profile indicating a predicted error in the predicted label; generating, by the first computing system, a first share of an inference result based at least in part on the predicted label determined for the given user profile and the predicted residue value; and providing, by the first computing system and to a client device, the first share of the inference result and a second share of the inference result received from the second computing system. Determining the predicted residue value for the given user profile includes determining, by the first computing system, a first share of the predicted residue value for the given user profile based at least in part on the first share of the given user profile and a second machine learning model trained using the user profiles and data indicating differences between true labels for the user profiles and predicted labels as determined for the user profiles using the first machine learning model; receiving, by the first computing system and from the second computing system, data indicating a second share of the predicted residue value for the given user profile determined by the second computing system based at least in part on the second share of the given user profile and a second set of one or more machine learning models; and determining the predicted residue value for the given user profile based at least in part on the first and second shares of the predicted residue value. Other implementations of this aspect include corresponding apparatus, systems, and computer programs, configured to perform the aspects of the methods, encoded on computer storage devices.


These and other implementations can each optionally include one or more of the following features. In some aspects, determining the predicted label for the given user profile includes determining, by the first computing system, a first share of the predicted label based at least in part on (i) the first share of the given user profile, (ii) a first machine learning model trained using multiple user profiles, and (iii) one or more of multiple true labels for the user profiles, the true labels including one or more true labels for each user profile in the multiple user profiles, receiving, by the first computing system and from a second computing system of the MPC computing systems, data indicating a second share of the predicted label determined by the second computing system based at least in part on a second share of the given user profile and a first set of one or more machine learning models, and determining the predicted label based at least in part on the first and second shares of the predicted label.


In some implementations, the methods further include applying, by the first computing system, a transformation to the first share of the given user profile to obtain a first transformed share of the given user profile. In such implementations, determining, by the first computing system, the first share of the predicted label includes determining, by the first computing system, a first share of the predicted label based at least in part on the first transformed share of the given user profile. In some such implementations, the transformation is a random projection, such as a Johnson-Lindenstrauss (J-L) transformation. In some of the aforementioned implementations, determining, by the first computing system, the first share of the predicted label includes providing, by the first computing system, the first transformed share of the given user profile as input to the first machine learning model to obtain a first share of the predicted label for the given user profile as output.


In some examples, the methods further include evaluating a performance of the first machine learning model and training the second machine learning model using data determined in evaluating the performance of the first machine learning model. In these examples, evaluating the performance of the first machine learning model includes, for each of the multiple user profiles, determining a predicted label for the user profile and determining a residue value for the user profile indicating an error in the predicted label. Also, in these examples, determining the predicted label for the user profile includes determining, by the first computing system, a first share of a predicted label for the user profile based at least in part on (i) a first share of the user profile, (ii) the first machine learning model, and (iii) one or more of the true labels for the user profiles, receiving, by the first computing system and from the second computing system, data indicating a second share of the predicted label for the user profile determined by the second computing system based at least in part on a second share of the user profile and the first set of one or more machine learning models maintained by the second computing system, and determining the predicted label for the user profile based at least in part on the first and second shares of the predicted label. Additionally, in such examples, determining the residue value for the user profile includes determining, by the first computing system, a first share of the residue value for the user profile based at least in part on the predicted label determined for the user profile and a first share of a true label for the user profile included in the true labels, receiving, by the first computing system and from the second computing system, data indicating a second share of the residue value for the user profile determined by the second computing system based at least in part on the predicted label determined for the user profile and a second share of the true label for the user profile, and determining the residue value for the user profile based at least in part on the first and second shares of the residue value. In the aforementioned examples, training the second machine learning model using data determined in evaluating the performance of the first machine learning model includes training the second machine learning model using data indicating the residue values determined for the user profiles in evaluating the performance of the first machine learning model.


In some of the aforementioned examples, before evaluating the performance of the first machine learning model, the methods further include deriving a set of parameters of a function and configuring the first machine learning model to, given a user profile as input, generate an initial predicted label for the user profile and apply the function, as defined based on the derived set of parameters, to the initial predicted label for the user profile to generate, as output, a first share of a predicted label for the user profile. In at least some of these examples, deriving the set of parameters of the function includes (i) deriving, by the first computing system, a first share of the set of parameters of the function based at least in part on a first share of each of the multiple true labels, (ii) receiving, by the first computing system and from the second computing system, data indicating a second share of the set of parameters of the function derived by the second computing system based at least in part on a second share of each of the multiple true labels, and (iii) deriving the set of parameters of the function based at least in part on the first and second shares of the set of parameters of the function. In at least some of the aforementioned examples, the function is a second degree polynomial function.


In some such examples, the methods further include estimating, by the first computing system, a first share of a set of distribution parameters based at least in part on the first share of each of the multiple true labels. In these examples, deriving, by the first computing system, the first share of the set of parameters of the function based at least in part on the first share of each of the multiple true labels includes deriving, by the first computing system, a first share of the set of parameters of the function based at least in part on the first share of the set of distribution parameters. In at least some of the aforementioned examples, the set of distribution parameters include (i) one or more parameters of a probability distribution of prediction errors for true labels of a first value in the multiple true labels, and (ii) one or more parameters of a probability distribution of prediction errors for true labels of a second value in the multiple true labels. In these examples, the second value is different from the first value. Furthermore, in at least some of the aforementioned examples, the first share of the residue value for the user profile is indicative of a difference in value between the predicted label determined for the user profile and the first share of the true label for the user profile, and the second share of the residue value for the user profile is indicative of a difference in value between the predicted label determined for the user profile and the second share of the true label for the user profile.


In some implementations, (i) the first machine learning model includes a k-nearest neighbor model maintained by the first computing system, (ii) the first set of one or more machine learning models includes a k-nearest neighbor model maintained by the second computing system, (iii) the second machine learning model includes at least one of a deep neural network (DNN) maintained by the first computing system and a gradient-boosting decision tree (GBDT) maintained by the first computing system, and/or (iv) the second set of one or more machine learning models includes at least one of a DNN maintained by the second computing system and a GBDT maintained by the second computing system.


In at least some of these implementations, determining, by the first computing system, the first share of the predicted label includes (i) identifying, by the first computing system, a first set of nearest neighbor user profiles based at least in part on the first share of the given user profile and the k-nearest neighbor model maintained by the first computing system, (ii) receiving, by the first computing system and from the second computing system, data indicating a second set of nearest neighbor profiles identified by the second computing system based at least in part on the second share of the given user profile and the k-nearest neighbor model maintained by the second computing system, (iii) identifying a number k of nearest neighbor user profiles that are considered most similar to the given user profile among the user profiles based at least in part on the first and second sets of nearest neighbor profiles, and (iv) determining, by the first computing system, the first share of the predicted label based at least in part on a true label for each of the k nearest neighbor user profiles.


In at least some of the aforementioned implementations, determining, by the first computing system, the first share of the predicted label further includes (i) determining, by the first computing system, a first share of a sum of the true labels for the k nearest neighbor user profiles, (ii) receiving, by the first computing system and from the second computing system, a second share of the sum of the true labels for the k nearest neighbor user profiles, and (iii) determining the sum of the true labels for the k nearest neighbor user profiles based at least in part on the first and second shares of the sum of the true labels for the k nearest neighbor user profiles. Furthermore, in some such implementations, determining, by the first computing system, the first share of the predicted label further includes applying a function to the sum of the true labels for the k nearest neighbor user profiles to generate the first share of the predicted label for the given user profile. In some of the aforementioned implementations, the first share of the predicted label for the given user profile includes the sum of the true labels for the k nearest neighbor user profiles.


In some of the aforementioned implementations, determining, by the first computing system, the first share of the predicted label based at least in part on the true label for each of the k nearest neighbor user profiles includes determining, by the first computing system, a first share of a set of predicted labels based at least in part on a set of true labels for each of the k nearest neighbor user profiles corresponding to a set of categories, respectively. In these implementations, includes determining, by the first computing system, the first share of the set of predicted labels includes, for each category in the set, (i) determining a first share of a frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of a first value, (ii) receiving, by the first computing system and from the second computing system, a second share of the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value, and (iii) determining the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value based at least in part on the first and second shares of the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value. In some of these implementations, determining, by the first computing system, the first share of the set of predicted labels includes, for each category in the set, applying a function corresponding to the category to the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value to generate a first share of a predicted label corresponding to the category for the given user profile.


Another innovative aspect of the subject matter described in this specification can be embodied in methods that include receiving, by a secure MPC cluster of computing systems, an inference request associated with a given user profile, determining, by the MPC cluster, a predicted label for the given user profile based at least in part on a first machine learning model trained using multiple user profiles, determining, by the MPC cluster, a predicted residue value for the given user profile indicating a predicted error in the predicted label based at least in part on the given user profile and a second machine learning model trained using the user profiles and data indicating differences between true labels for the user profiles and predicted labels as determined for the user profiles using the first machine learning model, generating, by the MPC cluster, data representing an inference result based at least in part on the predicted label determined for the given user profile and the predicted residue value, and providing, by the MPC cluster, the data representing the inference result to a client device. Other implementations of this aspect include corresponding apparatus, systems, and computer programs, configured to perform the aspects of the methods, encoded on computer storage devices.


These and other implementations can each optionally include one or more of the following features. In some aspects, the inference request includes an encrypted second share of the given user profile that was encrypted using an encryption key of the second computing system. Some aspects can include transmitting the encrypted second share of the given user profile to the second computing system.


In some aspects, determining the predicted label for the given user profile includes determining, by the MPC cluster, the predicted label for the given user profile based at least in part on (i) the given user profile, (ii) the first machine learning model trained using the user profiles, and (iii) one or more of the true labels for the user profiles, the true labels including one or more true labels for each user profile in the multiple user profiles.


In some implementations, the methods further include applying, by the MPC cluster, a transformation to the given user profile to obtain a transformed version of the given user profile. In these implementations, determining, by the MPC cluster, the predicted label includes determining, by the MPC cluster, the predicted label based at least in part on the transformed version of the given user profile. In some such implementations, the transformation is a random projection, such as a Johnson-Lindenstrauss (J-L) transformation. In at least some of the aforementioned implementations, determining, by the MPC cluster, the predicted label includes providing, by the MPC cluster, the transformed version of the given user profile as input to the first machine learning model to obtain the predicted label for the given user profile as output.


In some examples, the methods further include evaluating a performance of the first machine learning model and training the second machine learning model using data determined in evaluating the performance of the first machine learning model. In such examples, (1) determining, by the MPC cluster, a predicted label for the user profile based at least in part on (i) the user profile, (ii) the first machine learning model, and (iii) one or more of the true labels for the user profiles, and (2) determining, by the MPC cluster, a residue value for the user profile indicating a predicted error in the predicted label based at least in part on the predicted label determined for the user profile and a true label for the user profile included in the true labels. In the aforementioned examples, training the second machine learning model using data determined in evaluating the performance of the first machine learning model includes training the second machine learning model using data indicating the residue values determined for the user profiles in evaluating the performance of the first machine learning model.


In at least some of the aforementioned examples, before evaluating the performance of the first machine learning model, the methods further include deriving, by the MPC cluster, a set of parameters of a function based at least in part on the true labels, and configuring the first machine learning model to, given a user profile as input, generate an initial predicted label for the user profile and apply the function, as defined based on the derived set of parameters, to the initial predicted label for the user profile to generate, as output, a predicted label for the user profile. In some such examples, the methods further include estimating, by the MPC cluster, a set of normal distribution parameters based at least in part on the true labels. In these examples, deriving, by the MPC cluster, the set of parameters of the function based at least in part on the true labels includes deriving, by the MPC cluster, the set of parameters of the function based at least in part on the estimated set of normal distribution parameters. In some of the aforementioned examples, the set of distribution parameters include one or more parameters of a probability distribution of prediction errors for true labels of a first value in the true labels, and one or more parameters of a probability distribution of prediction errors for true labels of a second value in the true labels, the second value being different from the first value. Furthermore, in some of the aforementioned examples, the function is a second degree polynomial function. In at least some of the aforementioned examples, the residue value for the user profile is indicative of a difference in value between the predicted label determined for the user profile and the true label for the user profile.


In some implementations, the first machine learning model includes a k-nearest neighbor model. In some of these implementations, determining, by the MPC cluster, the predicted label includes (i) identifying, by the MPC cluster, a number k of nearest neighbor user profiles that are considered most similar to the given user profile among the user profiles based at least in part on the given user profile and the k-nearest neighbor model, and (ii) determining, by the MPC cluster, the predicted label based at least in part on a true label for each of the k nearest neighbor user profiles.


In at least some of the aforementioned implementations, determining, by the MPC cluster, the predicted label based at least in part on the true label for each of the k nearest neighbor user profiles includes determining, by the MPC cluster, a sum of the true labels for the k nearest neighbor user profiles. In some such implementations, determining, by the MPC cluster, the predicted label further includes applying a function to the sum of the true labels for the k nearest neighbor user profiles to generate a predicted label for the given user profile. Furthermore, in some of the aforementioned implementations, the predicted label for the given user profile includes the sum of the true labels for the k nearest neighbor user profiles.


In at least some of the aforementioned implementations, determining, by the MPC cluster, the predicted label based at least in part on the true label for each of the k nearest neighbor user profiles includes determining, by the MPC cluster, a set of predicted labels based at least in part on a set of true labels for each of the k nearest neighbor user profiles corresponding to a set of categories, respectively. In these implementations, determining, by the MPC cluster, the set of predicted labels includes, for each category in the set, determining a frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of a first value. In some of these implementations, determining, by the MPC cluster, the set of predicted labels includes, for each category in the set, applying a function corresponding to the category to the determined frequency to generate a predicted label corresponding to the category for the given user profile.


In some examples, each of the true labels is encrypted. In some implementations, the inference result includes a sum of the predicted label and the predicted residue value. In some examples, the second machine learning model includes at least one of a deep neural network, a gradient-boosting decision tree, and a random forest model.


In some examples, the client device computes the given user profile using multiple feature vectors that each include feature values related to events of a user of the client device and a decay rate for each feature vector.


In some examples, the client device computes the given user profile using multiple feature vectors that each include feature values related to events of a user of the client device. Computing the given user profile can include classifying one or more of the feature vectors as sparse feature vectors, classifying one or more of the feature vectors as dense feature vectors. Some aspects can include generating the first share of the given user profile and respective second shares of the given user profile for the one or more second computing systems using the sparse feature vectors and dense feature vectors. Generating the first share and the respective one or more second shares of the given user profile can include splitting the sparse feature vector using a Function Secret Sharing (FSS) technique.


Yet another innovative aspect of the subject matter described in this specification can be embodied in methods that include receiving, by a first computing system of multiple MPC systems, an inference request comprising a first share of a given user profile; identifying a number k of nearest neighbor user profiles that are considered most similar to the given user profile among multiple user profiles, including: identifying, by the first computing system and based on the first share of the given user profile and a first k-nearest neighbor model trained using the user profiles, a first set of nearest neighbor user profiles; receiving, by the first computing system and from each of one or more second computing systems of the multiple MPC systems, data indicating a respective second set of nearest neighbor profiles identified by the second computing system based on a respective second share of the given user profile and a respective second k-nearest neighbor model trained by the second computing system; identifying, by the first computing system and based on the first set of nearest neighbor user profiles and each second set of nearest neighbor user profiles, the number k of nearest neighbor user profiles; generating, by the first computing system, a first share of an inference result based on a respective label for each of the k nearest neighbor user profiles, wherein the label for each user profile is predictive of one or more user groups to which a user corresponding to the user profile will be added, and wherein the inference result indicates whether a given user corresponding to the given user profile is to be added to a given user group; and providing, by the first computing system and to a client device, the first share of the inference result and a respective second share of the inference result received from each of the one or more second computing systems. Other implementations of this aspect include corresponding apparatus, systems, and computer programs, configured to perform the aspects of the methods, encoded on computer storage devices.


These and other implementations can each optionally include one or more of the following features. In some aspects, the inference request includes an encrypted second share of the given user profile that was encrypted using an encryption key of the second computing system. Some aspects can include transmitting the encrypted second share of the given user profile to the second computing system.


In some aspects, the second share of the inference result is encrypted using an encryption key of an application of the client device. In some aspects, the label for each user profile has Boolean type for binary classification. Generating the first share of the inference result can include determining a first share of a sum of the labels for the k nearest neighbor user profiles, receiving, from the second computing system, a second share of the sum of the labels for the k nearest neighbor user profiles, determining, based on the first share of a sum of the labels and the second share of a sum of the labels, the sum of the labels, determining that the sum of the labels exceeds a threshold, in response to determining that the sum of the labels exceeds a threshold determining, as the inference result, to add the given user to the given user group, and generating the first share of the inference result based on the inference result.


In some aspects, the label for each user profile has a numerical value. Generating the first share of the inference result can include determining a first share of a sum of the labels for the k nearest neighbor user profiles, receiving, from the second computing system, a second share of the sum of the labels for the k nearest neighbor user profiles, determining, based on the first share of a sum of the labels and the second share of a sum of the labels, the sum of the labels, determining, as the inference result, based on the sum of the labels, that the given user is to join the given user group, and generating the first share of the inference result based on the inference result.


In some aspects, the label for each user profile has a categorical value. Generating the first share of the inference result can include, for each label in a set of labels, determining a first share of a frequency at which user profiles in the k nearest neighbor profiles have the label, receiving, from the second computing system, a second share of the frequency at which user profiles in the k nearest neighbor profiles have the label, and determining, based on the first share and second share of the frequency at which user profiles in the k nearest neighbor profiles have the label, the frequency at which users profiles in the k nearest neighbor profiles have the label. Some aspects can include identifying the label having the highest frequency, assigning, as the inference result, the given user is to join the given user group corresponding to the label having the highest frequency, and generating the first share of the inference result based on the inference result.


Some aspects can include training the first k-nearest neighbor model. The training an include creating, in collaboration with the second computing system, first shares of a random bit flipping pattern, generating a first share of a bit matrix by projecting a first share of each user profile in the user profiles onto a set of random projection planes, modifying the first share of the bit matrix by modifying one or more bits of the first share of the bit matrix using the first shares of the bit flipping pattern, providing a first portion of the modified first share of the bit matrix to the second computing system, receiving, from the second computing system, a second half of a modified second share of the bit matrix generated by the second computing system using second shares of the user profiles in the multiple user profiles and second shares of the random bit flipping pattern, and reconstructing, by the first computing system, bit vectors for the second half of the first bit matrix using a second half of the modified first share of the bit matrix and the second half of the modified second share of the bit matrix. Creating, in collaboration with the second computing system, first shares of a random bit flipping pattern can include generating a first m-dimensional vector comprising multiple first elements that each have a value of zero or one, splitting the first m-dimensional vector into two shares, providing a first share of the first m-dimensional vector to the second computing system, receiving a first share of a second m-dimensional vector from the second computing system, and computing, in collaboration with the second computing system, the first share of the random bit flipping pattern using shares of the first and second m-dimensional vectors. In some aspects, the multiple MPC computing systems include more than two MPC computing systems.


The subject matter described in this specification can be implemented in particular embodiments so as to realize one or more of the following advantages. The machine learning techniques described in this document can identify users that have similar interests and expand user group membership while preserving the privacy of users, e.g., without leaking users' online activity to any computing systems. This protects user privacy with respect to such platforms and preserves the security of the data from breaches during transmission or from the platforms. Cryptographic techniques, such as secure multi-party computation (MPC), enable the expansion of user groups based on similarities in user profiles without the use of third-party cookies, which preserves user privacy without negatively impacting the ability to expand the user groups and in some cases provides better user group expansion based on more complete profiles than achievable using third-party cookies. The MPC techniques can ensure that, as long as one of the computing systems in an MPC cluster is honest, no user data can be obtained by any of the computing systems or another party in plaintext. As such, the claimed methods allow the identification, grouping and transmission of user data in a secure manner, without requiring the use of third-party cookies to determine any relations between user data. This is a distinct approach from previous, known methods which generally require third-party cookies to determine relationships between data. By grouping user data in this manner, the efficiency of transmitting data content to user devices is improved as data content that is not relevant to a particular user need not be transmitted. Particularly, third-party cookies are not required thereby avoiding the storage of third-party cookies, improving memory usage. Exponential decay techniques can be used to build user profiles at client devices to reduce the data size of the raw data needed to build the user profiles, thereby reducing data storage requirements of client devices, which often have very limited data storage. The accuracy of the classifications, e.g., for user group expansion, can be improved by training a stronger model, e.g., a deep neural network model, based on another model, e.g., a k nearest neighbor model. That is, the techniques described in this document can improve accuracy by training a strong learner based on a weaker learner.


Various features and advantages of the foregoing subject matter is described below with respect to the figures. Additional features and advantages are apparent from the subject matter described herein and the claims.





BRIEF DESCRIPTION OF THE DRAWINGS


FIG. 1 is a block diagram of an environment in which a secure MPC cluster trains machine learning models and the machine learning models are used to expand user groups.



FIG. 2 is a swim lane diagram of an example process for training a machine learning model and using the machine learning model to add users to user groups.



FIG. 3 is a flow diagram that illustrates an example process for generating a user profile and sending shares of the user profile to an MPC cluster.



FIG. 4 is a flow diagram that illustrates an example process for generating a machine learning model.



FIG. 5 is a flow diagram that illustrates an example process for adding a user to user groups using machine learning models.



FIG. 6 is a conceptual diagram of an exemplary framework for generating an inference result for a user profile.



FIG. 7 is a conceptual diagram of an exemplary framework for generating an inference result for a user profile with boosted performance.



FIG. 8 is a flow diagram that illustrates an example process for generating an inference result for a user profile with boosted performance at an MPC cluster.



FIG. 9 is a flow diagram that illustrates an example process for preparing for and carrying out a training of a second machine learning model for boosting inference performance at an MPC cluster.



FIG. 10 is a conceptual diagram of an exemplary framework for evaluating a performance of a first machine learning model.



FIG. 11 is a flow diagram that illustrates an example process for evaluating a performance of a first machine learning model at an MPC cluster.



FIG. 12 is a flow diagram that illustrates an example process for generating an inference result for a user profile with boosted performance at a computing system of an MPC cluster.



FIG. 13 is a block diagram of an example computer system.





Like reference numbers and designations in the various drawings indicate like elements.


DETAILED DESCRIPTION

In general, this document describes systems and techniques for training and using machine learning models to expand user group membership while preserving user privacy and ensuring data security. In general, rather than creating and maintaining user profiles at computing systems of other entities, such as content platforms, the user profiles are maintained at the client devices of the users. To train the machine learning models, the client devices of the users can send their encrypted user profiles (e.g., as secret shares of the user profiles) along with other data to multiple computing systems of a secure multi-party computation (MPC) cluster, optionally via a content platform. For example, each client device can generate two or more secret shares of the user profile and send a respective secret share to each computing system. The computing systems of the MPC cluster can use MPC techniques to train machine learning models for suggesting user groups for the users based on their profiles in ways that prevent any computing system of the MPC cluster (or other party which is not the user itself) from obtaining any user's profile in plaintext, thereby preserving user privacy. For example, using the secret shares and MPC techniques described in this document enables the machine learning models to be trained and used while the user profile data of each user is encrypted at all times when the data it outside of the user's device. The machine learning models can be k-nearest neighbor (k-NN) models.


After the machine learning models are trained, the machine learning models can be used to suggest one or more user groups for each user based on their profiles. For example, the client device of a user can query the MPC cluster for suggested user groups for that user or to determine whether a user should be added to a particular user group. Various inference techniques can be used, such as binary classification, regression (e.g., using arithmetic mean or root mean square), and/or multiclass classification to identify the user groups. The user group membership of a user can be used in privacy preserving and secure ways to provide content to the user.


Example System for Generating and Using Machine Learning Models


FIG. 1 is a block diagram of an environment 100 in which a secure MPC 130 cluster trains machine learning models and the machine learning models are used to expand user groups. The example environment 100 includes a data communication network 105, such as a local area network (LAN), a wide area network (WAN), the Internet, a mobile network, or a combination thereof. The network 105 connects the client devices 110, the secure MPC cluster 130, publishers 140, websites 142, and content platforms 150. The example environment 100 may include many different client devices 110, secure MPC clusters 130, publishers 140, websites 142, and content platforms 150.


A client device 110 is an electronic device that is capable of communicating over the network 105. Example client devices 110 include personal computers, mobile communication devices, e.g., smart phones, and other devices that can send and receive data over the network 105. A client device can also include a digital assistant device that accepts audio input through a microphone and outputs audio output through speakers. The digital assistant can be placed into listen mode (e.g., ready to accept audio input) when the digital assistant detects a “hotword” or “hotphrase” that activates the microphone to accept audio input. The digital assistant device can also include a camera and/or display to capture images and visually present information. The digital assistant can be implemented in different forms of hardware devices including, a wearable device (e.g., watch or glasses), a smart phone, a speaker device, a tablet device, or another hardware device. A client device can also include a digital media device, e.g., a streaming device that plugs into a television or other display to stream videos to the television, or a gaming device or console.


A client device 110 typically includes applications 112, such as web browsers and/or native applications, to facilitate the sending and receiving of data over the network 105. A native application is an application developed for a particular platform or a particular device (e.g., mobile devices having a particular operating system). Publishers 140 can develop and provide, e.g., make available for download, native applications to the client devices 110. A web browser can request a resource 145 from a web server that hosts a website 142 of a publisher 140, e.g., in response to the user of the client device 110 entering the resource address for the resource 145 in an address bar of the web browser or selecting a link that references the resource address. Similarly, a native application can request application content from a remote server of a publisher.


Some resources, application pages, or other application content can include digital component slots for presenting digital components with the resources 145 or application pages. As used throughout this document, the phrase “digital component” refers to a discrete unit of digital content or digital information (e.g., a video clip, audio clip, multimedia clip, image, text, or another unit of content). A digital component can electronically be stored in a physical memory device as a single file or in a collection of files, and digital components can take the form of video files, audio files, multimedia files, image files, or text files and include advertising information, such that an advertisement is a type of digital component. For example, the digital component may be content that is intended to supplement content of a web page or other resource presented by the application 112. More specifically, the digital component may include digital content that is relevant to the resource content (e.g., the digital component may relate to the same topic as the web page content, or to a related topic). The provision of digital components can thus supplement, and generally enhance, the web page or application content.


When the application 112 loads a resource (or application content) that includes one or more digital component slots, the application 112 can request a digital component for each slot. In some implementations, the digital component slot can include code (e.g., scripts) that cause the application 112 to request a digital component from a digital component distribution system that selects a digital component and provides the digital component to the application 112 for presentation to a user of the client device 110.


The content platforms 150 can include supply-side platforms (SSPs) and demand-side platforms (SSPs). In general, the content platforms 150 manage the selection and distribution of digital components on behalf of publishers 140 and digital component providers 160.


Some publishers 140 use an SSP to manage the process of obtaining digital components for digital component slots of its resources and/or applications. An SSP is a technology platform implemented in hardware and/or software that automates the process of obtaining digital components for the resources and/or applications. Each publisher 140 can have a corresponding SSP or multiple SSPs. Some publishers 140 may use the same SSP.


Digital component providers 160 can create (or otherwise publish) digital components that are presented in digital component slots of publisher's resources and applications. The digital component providers 160 can use a DSP to manage the provisioning of its digital components for presentation in digital component slots. A DSP is a technology platform implemented in hardware and/or software that automates the process of distributing digital components for presentation with the resources and/or applications. A DSP can interact with multiple supply-side platforms SSPs on behalf of digital component providers 160 to provide digital components for presentation with the resources and/or applications of multiple different publishers 140. In general, a DSP can receive requests for digital components (e.g., from an SSP), generate (or select) a selection parameter for one or more digital components created by one or more digital component providers based on the request, and provide data related to the digital component (e.g., the digital component itself) and the selection parameter to an SSP. The SSP can then select a digital component for presentation at a client device 110 and provide, to the client device 110, data that causes the client device 110 to present the digital component.


In some cases, it is beneficial to a user to receive digital components related to web pages, application pages, or other electronic resources previously visited and/or interacted with by the user. In order to distribute such digital components to users, the users can be assigned to user groups, e.g., user interest groups, cohorts of similar users, or other group types involving similar user data. For example a user can be assigned to a user interest group when the users visit particular resources or perform particular actions at the resource (e.g., interact with a particular item presented on a web page or add the item to a virtual cart). In another example, a user can be assigned to a user group based on a history of activity, e.g., a history of resources visited and/or actions performed at the resources. In some implementations, the user groups can be generated by the digital component providers 160. That is, each digital component provider 160 can assign users to their user groups when the users visit electronic resources of the digital component providers 160.


To protect user privacy, a user's group membership can be maintained at the user's client device 110, e.g., by one of the applications 112, or the operating system of the client device 110, rather than by a digital component provider, content platform, or other party. In a particular example, a trusted program (e.g., a web browser) or the operating system can maintain a list of user group identifiers (“user group list”) for a user using the web browser or another application. The user group list can include a group identifier for each user group to which the user has been added. The digital component providers 160 that create the user groups can specify the user group identifiers for their user groups. The user group identifier for a user group can be descriptive of the group (e.g., gardening group) or a code that represents the group (e.g., an alphanumeric sequence that is not descriptive). The user group list for a user can be stored in secure storage at the client device 110 and/or can be encrypted when stored to prevent others from accessing the list.


When the application 112 presents a resource or application content related to a digital component provider 160, or a web page on a website 142, the resource can request that the application 112 add one or more user group identifiers to the user group list. In response, the application 112 can add the one or more user group identifiers to the user group list and store the user group list securely.


The content platforms 150 can use the user group membership of a user to select digital components or other content that may be of interest to the user or may be beneficial to the user/user device in another way. For example, such digital components or other content may comprise data that improves a user experience, improves the running of a user device or benefits the user or user device in some other way. However, the user group identifiers of the user group list of a user can be provided in ways that prevent the content platforms 150 from correlating user group identifiers with particular users, thereby preserving user privacy when using user group membership data to select digital components.


The application 112 can provide user group identifiers from the user group list to a trusted computing system that interacts with the content platforms 150 to select digital components for presentation at the client device 110 based on the user group membership in ways that prevent the content platforms 150 or any other entities which are not the user itself from knowing a user's complete user group membership.


In some cases, it is beneficial to users and to digital component providers to expand user groups to include users that have similar interests or other similar data as the users that are already members of the user group.


Advantageously, users can be added to user groups without the use of third-party cookies. As described above, the user profiles can be maintained at the client device 110. This preserves user privacy by precluding a user's cross-domain browsing history to be shared with outside parties, reduces bandwidth consumed by transmitting the cookies over the network 105 (which, aggregated over millions of users is substantial), reduces the storage requirements of content platforms 150 that typically store such information, and reduces battery consumption used by client devices 110 to maintain and transmit the cookies.


For example, a first user may be interested in snow skiing and may be a member of a user group for a particular ski resort. A second user may also be interested in skiing, but unaware of this ski resort and not a member of the ski resort. If the two users have similar interests or data, e.g., similar user profiles, the second user may be added to the user group for the ski resort so that the second user receives content, e.g., digital components, related to the ski resort and that may be of interest or otherwise beneficial to the second user or a user device thereof. In other words, user groups may be expanded to include other users having similar user data.


The secure MPC cluster 130 can train machine learning models that suggest, or can be used to generate suggestions of, user groups to users (or their applications 112) based on the user's profiles. The secure MPC cluster 130 includes two computing systems MPC1 and MPC2 that perform secure MPC techniques to train the machine learning models. Although the example MPC cluster 130 includes two computing systems, more computing systems can also be used as long as the MPC cluster 130 includes more than one computing system. For example, the MPC cluster 130 can include three computing systems, four computing systems, or another appropriate number of computing systems. Using more computing systems in the MPC cluster 130 can provide more security and fault tolerance, but can also increase the complexity of the MPC processes.


The computing systems MPC1 and MPC2 can be operated by different entities. In this way, each entity may not have access to the complete user profiles in plaintext. Plaintext is text that is not computationally tagged, specially formatted, or written in code, or data, including binary files, in a form that can be viewed or used without requiring a key or other decryption device, or other decryption process. For example, one of the computing systems MPC1 or MPC2 can be operated by a trusted party different from the users, the publishers 140, the content platform 150, and the digital component providers 160. For example, an industry group, governmental group, or browser developer may maintain and operate one of the computing systems MPC1 and MPC2. The other computing system may be operated by a different one of these groups, such that a different trusted party operates each computing system MPC1 and MPC2. Preferably, the different parties operating the different computing systems MPC1 and MPC2 have no incentive to collude to endanger user privacy. In some implementations, the computing systems MPC1 and MPC2 are separated architecturally and are monitored to not communicate with each other outside of performing the secure MPC processes described in this document.


In some implementations, the MPC cluster 130 trains one or more k-NN models for each content platform 150 and/or for each digital component provider 160. For example, each content platform 150 can manage the distribution of digital components for one or more digital component providers 160. A content platform 150 can request that the MPC cluster 130 train a k-NN model for one or more of the digital component providers 160 for which the content platform 150 manages the distribution of digital components. In general, a k-NN model represents distances between the user profiles (and optionally additional information) of a set of users. Each k-NN model of a content platform can have a unique model identifier. An example process for training a k-NN model is illustrated in FIG. 4 and described below.


After training a k-NN model for a content platform 150, the content platform 150 can query, or have the application 112 of a client device 110 query the k-NN model to identify one or more user groups for a user of the client device 110. For example, the content platform 150 can query the k-NN model to determine whether a threshold number of the “k” user profiles nearest to the user are members of a particular user group. If so, the content platform 150 may add the user to that user group. If a user group is identified for the user, the content platform 150 or the MPC cluster 130 can request that the application 112 add the user to the user group. If approved by the user and/or the application 112, the application 112 can add a user group identifier for the user group to the user group list stored at the client device 110.


In some implementations, an application 112 can provide a user interface that enables a user to manage the user groups to which the user is assigned. For example, the user interface can enable the user to remove user group identifiers, prevent all or particular resources 145, publishers 140, content platforms 150, digital component providers 160, and/or MPC clusters 130 from adding the user to a user group (e.g., prevent the entity from adding user group identifiers to the list of user group identifiers maintained by the application 112). This provides better transparency, choice/consent, and control for the user.


Further to the descriptions throughout this document, a user may be provided with controls (e.g., user interface elements with which a user can interact) allowing the user to make an election as to both if and when systems, programs, or features described herein may enable collection of user information (e.g., information about a user's social network, social actions, or activities, profession, a user's preferences, or a user's current location), and if the user is sent content or communications from a server. In addition, certain data may be treated in one or more ways before it is stored or used, so that personally identifiable information is removed. For example, a user's identity may be treated so that no personally identifiable information can be determined for the user, or a user's geographic location may be generalized where location information is obtained (such as to a city, ZIP code, or state level), so that a particular location of a user cannot be determined. Thus, the user may have control over what information is collected about the user, how that information is used, and what information is provided to the user.


Example Process for Generating and Using Machine Learning Models


FIG. 2 is a swim lane diagram of an example process 200 for training a machine learning model and using the machine learning model to add users to user groups. Operations of the process 200 can be implemented, for example, by the client device 110, the computing systems MPC1 and MPC2 of the MPC cluster 130, and a content platform 150. Operations of the process 200 can also be implemented as instructions stored on one or more computer readable media which may be non-transitory, and execution of the instructions by one or more data processing apparatus can cause the one or more data processing apparatus to perform the operations of the process 200. Although the process 200 and other processes below are described in terms of a two computing system MPC cluster 130, MPC clusters having more than two computing systems can also be used to perform similar processes.


A content platform 150 can initiate the training and/or updating of one of its machine learning models by requesting that applications 112 running on client devices 110 generate a user profile for their respective users and upload secret-shared and/or encrypted versions of the user profiles to the MPC cluster 130. For the purposes of this document, secret shares of user profiles can be considered encrypted versions of the user profiles as the secret shares are not in plaintext. In generation, each application 112 can store data for a user profile and generate the updated user profile in response to receiving a request from the content platform 150. As the content of a user profile and the machine learning models differ for different content platforms 150, the application 112 running on a user's client device 110 can maintain data for multiple user profiles and generate multiple user profiles that are each specific to particular content platforms, or particular model owned by a particular content platform.


An application 112 running on a client device 110 builds a user profile for a user of the client device 110 (202). The user profile for a user can include data related to events initiated by the user and/or events that could have been initiated by the user with respect to electronic resources, e.g., web pages or application content. The events can include views of electronic resources, views of digital components, user interactions, or the lack of user interactions, with (e.g., selections of) electronic resources or digital components, conversions that occur after user interaction with electronic resources, and/or other appropriate events related to the user and electronic resources.


A user profile for a user can be specific to a content platform 150, or selected machine learning models owned by the content platform 150. For example, as described in more detail below with reference to FIG. 3, each content platform 150 can request that the application 112 generate or update a user profile specific to that content platform 150.


The user profile for a user can be in the form of a feature vector. For example, the user profile can be an n-dimensional feature vector. Each of the n dimensions can correspond to a particular feature and the value of each dimension can be the value of the feature for the user. For example, one dimension may be for whether a particular digital component was presented to (or interacted with by) the user. In this example, the value for that feature could be “1” if the digital component was presented to (or interacted with by) the user or “0” if the digital component has not been presented to (or interacted with by) the user. An example process for generating a user profile for a user is illustrated in FIG. 3 and described below.


In some implementations, a content platform 150 may want to train machine learning models based on additional signals, such as contextual signals, signals related to particular digital components, or signals related to the user of which the application 112 may not be aware or to which the application 112 may not have access, such as the current weather at the user's location. For example, the content platform 150 may want to train a machine learning model to predict whether a user will interact with a particular digital component if the digital component is presented to the user in a particular context. In this example, the contextual signals can include, for each presentation of a digital component to a user, the geographic location of the client device 110 at the time (if permission is granted by the user), signals describing the content of the electronic resource with which the digital component is presented, and signals describing the digital component, e.g., the content of the digital component, the type of digital components, where on the electronic resource the digital component is presented, etc. In another example, one dimension may be for whether the digital component presented to the user is of a particular type. In this example, the value could be 1 for travel, 2 for cuisine, 3 for movie, etc. For ease of subsequent description, Pi will represent both user profile and additional signals (e.g., contextual signals and/or digital component-level signals) associated with the i-th user profile.


The application 112 generates shares of the user profile Pi for the user (204). In this example, the application 112 generates two shares of the user profile Pi, one for each computing system of the MPC cluster 130. Note that each share by itself can be a random variable that by itself does not reveal anything about the user profile. Both shares would need to be combined to get the user profile. If the MPC cluster 130 includes more computing systems that participate in the training of a machine learning model, the application 112 would generate more shares, one for each computing system. In some implementations, to protect user privacy, the application 112 can use a pseudorandom function to split the user profile Pi into shares. That is, the application 112 can use pseudorandom function PRF(Pi) to generate two shares {[Pi, 1],[Pi,2]}. The exact splitting can depend on the secret sharing algorithm and crypto library used by the application 112.


In some implementations, the application 112 can also provide one or more labels to the MPC cluster 130. Although the labels may not be used in training the machine learning models of certain architecture (e.g. k-NN), the labels can be used to fine-tune hyperparameters controlling the model training process (e.g., the value of k), or evaluate the quality of the machine learning models trained, or make predictions, i.e. determine whether to suggest a user group for a user. The labels can include, for example, one or more of the user group identifiers for the user and for which the content platform 150 has access. That is, the labels can include the user group identifiers for the user groups managed by the content platform 150 or for which the content platform 150 has read access. In some implementations, a single label includes multiple user group identifiers for the user. In some implementations, the label for a user can be heterogeneous and include all user groups that include the user as a member and additional information, e.g., whether the user interacted with a given digital component. This enables the k-NN model to be used to predict whether another user will interact with the given digital component. A label for each user profile can indicate user group membership for a user corresponding to the user profile.


The label for the user profiles are predictive of user groups to which a user corresponding to an input will be, or should be added. For example, the labels corresponding to the k nearest neighbor user profiles to the input user profile are predictive of user groups that the user corresponding to the input user profile will or should join, e.g., based on the similarity between the user profiles. These predictive labels can be used to suggest user groups to the user or to request that the application added the user to the user groups corresponding to the labels.


If labels are included, the application 112 can also split each label, into shares, e.g., [labeli,1] and [labeli,2]. In this way, without collusion between the computing systems MPC1 and MPC2, neither computing system MPC1 nor MPC2 can reconstruct Pi from [Pi,1] or [Pi,2] or reconstruct label, from [labeli,1] or [labeli,2].


The application 112 encrypts the shares [Pi,1] or [Pi,2] of the user profile Pi and/or the shares [labeli,1] or [labeli,2] of each label labeli (206). In some implementations, the application 112 generates a composite message of the first share [Pi,1] of the user profile Pi and the first share [labeli,1] of the label labeli and encrypts the composite message using an encryption key of the computing system MPC1. Similarly, application 112 generates a composite message of the second share [Pi,2] of the user profile Pi and the second share [labeli,2] of the label labeli and encrypts the composite message using an encryption key of the computing system MPC2. These functions can be represented as PubKeyEncrypt([Pi,1]∥[labeli,1], MPC1) and PubKeyEncrypt([Pi,2]∥[labeli,2], MPC2), where PubKeyEncrypt represents a public key encryption algorithm using the corresponding public key of MPC1 or MPC2. The symbol “∥” represents a reversible method to compose complex messages from multiple simple messages, e.g., JavaScript Object Notation (JSON), Concise Binary Object Representation (CBOR), or protocol buffer.


The application 112 provides the encrypted shares to the content platform 150 (208). For example, the application 112 can transmit the encrypted shares of the user profile and the label to the content platform 150. As each share is encrypted using an encryption key of the computing system MPC1 or MPC2, the content platform 150 cannot access the user's user profile or the label.


The content platform 150 can receive shares of user profiles and shares of labels from multiple client devices. The content platform 150 can initiate the training of a machine learning model by uploading the shares of the user profiles to the computing systems MPC1 and MPC2. Although the labels may not be used in the training process, the content platform 150 can upload the shares of the labels to the computing systems MPC1 and MPC2 for use when evaluating the model quality or querying the model later.


The content platform 150 uploads the first encrypted shares (e.g., PubKeyEncrypt([Pi,1]∥[labeli,1], MPC1)) received from each client device 110 to the computing system MPC1 (210). Similarly, the content platform 150 uploads the second encrypted shares (e.g., PubKeyEncrypt([Pi,2]∥[labeli,2], MPC2)) to the computing system MPC2 (212). Both uploads can be in batches and can include the encrypted shares of user profiles and labels received during a particular time period for training the machine learning model.


In some implementations, the order in which the content platform 150 uploads the first encrypted shares to the computing system MPC1 must match the order in which the content platform 150 uploads the second encrypted shares to the computing system MPC2. This enables the computing systems MPC1 and MPC2 to properly match two shares of the same secret, e.g., two shares of the same user profile.


In some implementations, the content platform 150 may explicitly assign the same pseudorandomly or sequentially generated identifier to shares of the same secret to facilitate the matching. While some MPC techniques can rely on random shuffling of input or intermediate results, the MPC techniques described in this document may not include such random shuffling and may instead rely on the upload order to match.


In some implementations, the operations 208, 210 and 212 can be replaced by alternative process where the application 112 directly uploads [Pi,1]∥[labeli,1] to MPC1, and [Pi,2]∥[labeli,2] to MPC2. This alternative process can reduce the infrastructure cost of the content platform 150 to support the operations 208, 210 and 212, and reduce the latency to start training or updating the machine learning models in MPC1 and MPC2. For example, this eliminates the transmittal of data to the content platform 150, which the content platform 150 then transmits to MPC1 and MPC2. Doing so reduces the amount of data transmitted over the network 105 and reduces the complexity of the logic of the content platform 150 in handling such data.


The computing systems MPC1 and MPC2 generate a machine learning model (214). Each time a new machine learning model is generated based on user profile data can be referred to as a training session. The computing systems MPC1 and MPC2 can train a machine learning model based on the encrypted shares of the user profiles received from the client devices 110. For example, the computing systems MPC1 and MPC2 can use MPC techniques to train a k-NN model based on the shares of the user profiles.


To minimize or at least reduce the crypto computation, and thus the computational burden placed on the computing systems MPC1 and MPC2 to protect user privacy and data during both model training and inference, the MPC cluster 130 can use random projection techniques, e.g., SimHash, to quantify the similarity between two user profiles Pi and Pj quickly, securely, and probabilistically. A SimHash is a technique that enables a quick estimate of the similarity between two data sets. The similarity between the two user profiles Pi and Pj can be determined by determining the Hamming distance between two bit vectors that represent the two user profiles Pi and Pj, which is inversely proportional to the cosine distance between the two user profiles with high probability.


Conceptually, for each training session, m random projection hyperplanes U={U1, U2, . . . , Um} can be generated. The random projection hyperplanes can also be referred to as random projection planes. One objective of the multi-step computation between the computing systems MPC1 and MPC2 is to create a bit vector B of length m for each user profile Pi used in the training of the k-NN model. In this bit vector Bi, each bit Bi,j represents the sign of a dot product of one of the projection planes Uj and the user profile Pi, i.e., Bi,j=sign(Uj⊙Pi) for all j∈[1, m] where ⊙ denotes the dot product of two vectors of equal length. That is, each bit represents which side of the plane Uj the user profile Pi is located. A bit value of one represents a positive sign and a bit value of zero represents a negative sign.


At each end of the multi-step computation, each of the two computing systems MPC1 and MPC2 generates an intermediate result that includes a bit vector for each user profile in cleartext, a share of each user profile, and a share of the label for each user profile. For example, the intermediate result for computing system MPC1 can be the data shown in Table 1 below. The computing system MPC2 would have a similar intermediate result but with a different share of each user profile and each label. To add extra privacy protection, each of the two servers in the MPC cluster 130 can only get half of the m-dimensional bit vectors in cleartext, e.g., computing system MPC1 gets the first m/2 dimension of all the m-dimension bit vectors, computing system MPC2 gets the second m/2 dimension of all the m-dimension bit vectors.











TABLE 1





Bit Vector in Cleartext
MPC1 Share for Pi
MPC1 Share for labeli







. . .
. . .
. . .


Bi
. . .
. . .


Bi+1
. . .
. . .


. . .
. . .
. . .









Given two arbitrary user profile vectors Pi and Pj of unit length i≠j, it has been shown that the Hamming distance between the bit vectors B, and B1 for the two user profile vectors Pi and Pj is proportional to the cosine distance between the user profile vectors Pi and Pj with high probability, assuming that the number of random projections m is sufficiently large.


Based on the intermediate result shown above and because the bit vectors Bi are in cleartext, each computing system MPC1 and MPC2 can independently create, e.g., by training, a respective k-NN model using a k-NN algorithm. The computing systems MPC1 and MPC2 can use the same or different k-NN algorithms. An example process for training a k-NN model is illustrated in FIG. 4 and described below. Once the k-NN models are trained, the application 112 can query the k-NN models to determine whether to add a user to a user group.


The application 112 submits an inference request to the MPC cluster 130 (216). In this example, the application 112 transmits the inference request to computing system MPC1. In other examples, the application 112 can transmit the inference request to computing system MPC2. The application 112 can submit the inference request in response to a request from the content platform 150 to submit the inference request. For example, the content platform 150 can request the application 112 to query the k-NN model to determine whether the user of the client device 110 should be added to a particular user group. This request can be referred to an inference request to infer whether the user should be added to the user group.


To initiate an inference request, the content platform 150 can send, to the application 112, an inference request token Minfer. The inference request token Minfer enables servers in the MPC cluster 130 to validate that the application 112 is authorized to query a particular machine learning model owned by a particular domain. The inference request token Minfer is optional if the model access control is optional. The inference request token Minfer can have the following items shown and described in Table 2 below.











TABLE 2





Item No.
Content
Description







1
Content Platform Domain
Content platform's domain



(e.g., eTLD + 1 domain)
that uniquely identifies the




content platform


2
Model Identifier
Unique identifier for the




content platform's machine




learning model


3
k
How many nearest




neighbors to fetch


4
Aggregation Function
The aggregation function




applied to the labels of the k




nearest neighbors


5
Aggregation Function
Additional parameters used



Parameters
by the aggregation function


6
Token Creation Timestamp
Timestamp indicating when




this token is created


7
Operation
Infer request operation


8
Digital Signature
The content platform's




digital signature over items




1-7









In this example, the inference request token Minfer includes seven items and a digital signature generated based on the seven items using a private key of the content platform 150. The eTLD+1 is the effective top-level domain (eTLD) plus one level more than the public suffix. An example eTLD+1 is “example.com“where”.com” is the top-level domain.


To request an inference for a particular user, the content platform 150 can generate an inference request token Minfer and send the token to the application 112 running on the user's client device 110. In some implementations, the content platform 150 encrypts the inference request token Minfer using a public key of the application 112 so that only the application 112 can decrypt the inference request token Minfer using its confidential private key that corresponds to the public key. That is, the content platform can send, to the application 112, PubKeyEnc(Minfer, application_public_key).


The application 112 can decrypt and verify the inference request token Minfer. The application 112 can decrypt the encrypted inference request token Minfer using its private key. The application 112 can verify the inference request token Minfer by (i) verifying the digital signature using a public key of the content platform 150 that corresponds to the private key of the content platform 150 that was used to generate the digital signature and (ii) ensuring that the token creation timestamp is not stale, e.g., the time indicated by the timestamp is within a threshold amount of time of a current time at which verification is taking place. If the inference request token Minfer is valid, the application 112 can query the MPC cluster 130.


Conceptually, the inference request can include the model identifier for the machine learning model, the current user profile Pi, k (the number of nearest neighbors to fetch), optionally additional signals (e.g., contextual signals or digital component signals), the aggregation function, and the aggregation function parameters. However, to prevent leaking the user profile Pi in plaintext form to either computing system MPC1 or MPC2, and thereby preserve user privacy, the application 112 can split the user profile Pi into two shares [Pi,1] and [Pi,2] for MPC1 and MPC2, respectively. The application 112 can then select one of the two computing systems MPC1 or MPC2, e.g., randomly or pseudorandomly, for the query. If the application 112 selects computing system MPC1, the application 112 can send a single request to computing system MPC1 with the first share [Pi,1] and an encrypted version of the second share, e.g., PubKeyEncrypt([Pi,2], MPC2). In this example, the application 112 encrypts the second share [Pi,2] using a public key of the computing system MPC2 to prevent computing system MPC1 from accessing [Pi,2], which would enable computing system MPC1 to reconstruct the user profile Pi from [Pi,1] and [Pi,2].


As described in more detail below, the computing systems MPC1 and MPC2 collaboratively compute the k nearest neighbors to the user profile Pi. The computing systems MPC1 and MPC2 can then use one of several possible machine learning techniques (e.g., binary classification, multiclass classification, regression, etc.) to determine, based on the k nearest neighbor user profiles, whether to add the user to a user group. For example, the aggregation function can identify the machine learning technique (e.g., binary, multiclass, regression) and the aggregation function parameters can be based on the aggregation function. The aggregation function can define a computation, e.g., a sum, logical AND or OR, or another appropriate function that is performed using the parameters. For example, the aggregation function can be in the form of an equation that includes the function and the parameters that are used in the equation.


In some implementations, the aggregation function parameters can include a user group identifier for a user group for which the content platform 150 is querying the k-NN model for the user. For example, the content platform 150 may want to know whether to add a user to a user group related to hiking and that has a user group identifier “hiking.” In this example, the aggregation function parameter can include the “hiking” user group identifier. In general, the computing systems MPC1 and MPC2 can determine whether to add the user to the user group based on the number of the k nearest neighbors that are a member of the user group, e.g., based on their labels.


The MPC cluster 130 provides an inference result to the application 112 (218). In this example, the computing system MPC1 that received the query sends the inference result to the application 112. The inference result can indicate whether the application 112 should add the user to zero or more user groups. For example, the user group result can specify a user group identifier for the user group. However, in this example, the computing system MPC1 would know the user group. To prevent this, the computing system MPC1 may compute a share of the inference result and the computing system MPC2 may compute another share of the same inference result. The computing system MPC2 can provide an encrypted version of its share to the computing system MPC1, where the share is encrypted using a public key of the application 112. The computing system MPC1 can provide, to the application 112, its share of the inference result and the encrypted version of computing system MPC2's share of the user group result. The application 112 can decrypt computing system MPC2's share and calculate the inference result from the two shares. An example process for querying a k-NN model to determine whether to add a user to a user group is illustrated in FIG. 5 and described below. In some implementations, to prevent computing system MPC1 from falsifying computing system MPC2's result, computing system MPC2 digitally signs its result either before or after encrypting its result using the public key of the application 112. The application 112 verifies computing system MPC2's digital signature using the public key of MPC2.


The application 112 updates the user group list for the user (220). For example, if the inference result is to add the user to a particular user group, the application 112 can add the user to the user group. In some implementations, the application 112 can prompt the user for permission to add the user to the user group.


The application 112 transmits a request for content (222). For example, the application 112 can transmit, to the content platform 150, a request for a digital component in response to loading an electronic resource that has a digital component slot. In some implementations, the request can include one or more user group identifiers for user groups that include the user as a member. For example, the application 112 can obtain one or more user group identifiers from the user group list and provide the user group identifier(s) with the request. In some implementations, techniques can be used to prevent the content platform from being able to associate the user group identifier with the user, the application 112, and/or the client device 112 from which the request is received.


The content platform 150 transmits content to the application 112 (224). For example, the content platform 150 can select a digital component based on the user group identifier(s) and provide the digital component to the application 112. In some implementations, the content platform 150, in collaboration with the application 112, selects a digital component based on the user group identifier(s), without leaking the user group identifier(s) out of the application 112.


The application 112 displays or otherwise implements the received content (226). For example, the application 112 can display a received digital component in a digital component slot of an electronic resource.


Example Process for Generating User Profiles


FIG. 3 is a flow diagram that illustrates an example process 300 for generating a user profile and sending shares of the user profile to an MPC cluster. Operations of the process 300 can be implemented, for example, by the client device 110 of FIG. 1, e.g., by the application 112 running on the client device 110. Operations of the process 300 can also be implemented as instructions stored on one or more computer readable media which may be non-transitory, and execution of the instructions by one or more data processing apparatus can cause the one or more data processing apparatus to perform the operations of the process 300.


An application 112 executing on a user's client device 110 receives data for an event (302). The event can be, for example, a presentation of an electronic resource at the client device 110, a presentation of a digital component at the client device 110, a user interaction with an electronic resource or digital component at the client device 110, or a conversion for a digital component, or the lack of user interaction with or conversion for an electronic resource or digital component presented. When an event occurs, a content platform 150 can provide data related to the event to the application 112 for use in generating a user profile for the user.


The application 112 can generate a different user profile for each content platform 150. That is, the user profile of a user and for a particular content platform 150 may only include event data received from the particular content platform 150. This preserves user privacy by not sharing with content platforms data related to events of other content platforms. In some implementations, the application 112, per the request of the content platform 150, may generate a different user profile for each machine learning model owned by the content platform 150. Based on the design goal, different machine learning models may require different training data. For example, a first model may be used to determine whether to add a user to a user group. A second model may be used to predict whether a user will interact with a digital component. In this example, the user profiles for the second model can include additional data, e.g., whether the user interacted with the digital component, that the user profiles for the first model do not have.


The content platform 150 can send the event data in the form of a profile update token Mupdate. The profile update token Mupdate have the following items shown and described in Table 3 below.











TABLE 3





Item No.
Content
Description







1
Content Platform Domain
Content platform's domain



(e.g., eTLD + 1 domain)
that uniquely identifies the




content platform


2
Model Identifier
Unique identifier for the




content platform's machine




learning model. This item




can have multiple values if




the same feature vector




should be applicable for the




training of multiple machine




learning models for the same




owner domain.


3
Profile Record
n-dimensional feature vector




determined by the content




platform based on the event


4
Token Creation Timestamp
Timestamp indicating when




this token is created


5
Expiration Time
A date and time at which the




feature vector will expire




and not be used for the user




profile calculation.


6
Profile Decay Rate
Optional rate that defines the




rate at which the weight of




this event's data decays in




the user profile


7
Operation
Accumulate user profile


8
Digital Signature
The content platform's




digital signature over items




1-7









The model identifier identifies the machine learning model, e.g., k-NN model, for which the user profile will be used to train or used to make a user group inference. The profile record is an n-dimensional feature vector that includes data specific to the event, e.g., the type of event, the electronic resource or digital component, time at which the event occurred, and/or other appropriate event data that the content platform 150 wants to use in training the machine learning model and making user group inferences. The digital signature is generated based on the seven items using a private key of the content platform 150.


In some implementations, to protect the update token Mupdate during transmission, the content platform 150 encrypts the update token Mupdate prior to sending the update token Mupdate to the application 112. For example, the content platform 150 can encrypt the update token Mupdate using a public key of the application, e.g., PubKeyEnc(Mupdate, application_public_key).


In some implementations, the content platform 150 can send the event data to the application 112 without encoding the event data or the update request in the form of a profile update token Mupdate. For example, a script originated from the content platform 150 running inside the application 112 may directly transmit the event data and the update request to the application 112 via a script API, where the application 112 relies on World Wide Web Consortium (W3C) origin-based security model and/or (Hypertext Transfer Protocol Secure) HTTPS to protect the event data and update request from falsification or leaking.


The application 112 stores the data for the event (304). If the event data is encrypted, the application 112 can decrypt the event data using its private key that corresponds to the public key used to encrypt the event data. If the event data is sent in the form of an update token Mupdate, the application 112 can verify the update token Mupdate before storing the event data. The application 112 can verify the update token Mupdate by (i) verifying the digital signature using a public key of the content platform 150 that corresponds to the private key of the content platform 150 that was used to generate the digital signature and (ii) ensuring that the token creation timestamp is not stale, e.g., the time indicated by the timestamp is within a threshold amount of time of a current time at which verification is taking place. If the update token Mupdate is valid, the application 112 can store the event data, e.g., by storing the n-dimensional profile record. If any verification fails, the application 112 may ignore the update request, e.g., by not storing the event data.


For each machine learning model, e.g., for each unique model identifier, the application 112 can store event data for that model. For example, the application 112 can maintain, for each unique model identifier, a data structure that includes a set of n-dimensional feature vectors (e.g., the profile records of the update tokens) and, for each feature vector, the expiration time. Each feature vector can include feature values for features related to events for the user of the client device 110. An example data structure for a model identifier is shown in Table 4 below.












TABLE 4







Feature Vector
Expiration









n-dimensional feature vector
Expiration time



. . .
. . .










Upon receiving a valid update token Mupdate, the application 112 can update the data structure for the model identifier included in the update token Mupdate by adding the feature vector and expiration time of the update token Mupdate to the data structure. Periodically, the application 112 can purge expired feature vectors from the data structure to reduce storage size.


The application 112 determines whether to generate a user profile (306). For example, the application 112 may generate a user profile for a particular machine learning model in response to a request from the content platform 150. The request may be to generate the user profile and return shares of the user profile to the content platform 150. In some implementations, the application 112 may directly upload the generated user profiles to the MPC cluster 130, e.g., rather than sending them to the content platform 150. To ensure the security of the request to generate and return the shares of the user profile, the content platform 150 can send, to the application 112, an upload token Mupload.


The upload token Mupload can have a similar structure as the update token Mupdate, but with a different operation (e.g., “update server” instead of “accumulate user profile”). The upload token Mupload can also include an additional item for an operation delay. The operation delay can instruct the application 112 to delay calculating and uploading the shares of the user profile while the application 112 accumulates more event data, e.g., more feature vectors. This enables the machine learning model to capture user event data immediately before and after some critical events, e.g., joining a user group. The operation delay can specify the delay time period. In this example, the digital signature can be generated based on the other seven items in Table 3 and the operation delay using the private key of the content platform. The content platform 150 can encrypt the upload token Mupload in a similar manner as the update token Mupdate, e.g., PubKeyEnc(Mupload, application_public_key), using the application's public key to protect the upload token Mupload during transmission.


The application 112 can receive the upload token Mupload, decrypt the upload token Mupload if it is encrypted, and verify the upload token Mupload. This verification can be similar to the way in which the update token Mupdate is verified. The application 112 can verify the upload token Mupload by (i) verifying the digital signature using a public key of the content platform 150 that corresponds to the private key of the content platform 150 that was used to generate the digital signature and (ii) ensuring that the token creation timestamp is not stale, e.g., the time indicated by the timestamp is within a threshold amount of time of a current time at which verification is taking place. If the upload token Mupload is valid, the application 112 can generate the user profile. If any verification fails, the application 112 may ignore the upload request, e.g., by not generating a user profile.


In some implementations, the content platform 150 can request the application 112 to upload a user profile without encoding the upload request in the form of a profile upload token Mupload. For example, a script originated from the content platform 150 running inside the application 115 may directly transmit the upload request to the application 115 via a script API, where the application 115 relies on W3C origin-based security model and/or HTTPS to protect the upload request from falsification or leaking.


If a determination is made to not generate a user profile, the process 300 can return to operation 302 and wait for additional event data from the content platform 150. If a determination is made to generate a user profile, the application 112 generates the user profile (308).


The application 112 can generate the user profile based on the stored event data, e.g., the data stored in the data structure shown in Table 4. The application 112 can access the appropriate data structure based on a model identifier included in the request, e.g., the Content Platform eTLD+1 domain of item 1 and the model identifier of item 2 of the upload token Mupload.


The application 112 can compute the user profile by aggregating the n-dimensional feature vectors in the data structure in the study period that have not yet expired.


For example, the user profile may be the average of the n-dimensional feature vectors in the data structure in the study period that have not yet expired. The result is an n-dimensional feature vector representing the user in the profile space. Optionally, the application 112 may normalize the n-dimensional feature vector to unit length, e.g., using L2 normalization. The content platform 150 may specify the optional study period.


In some implementations, decay rates can be used to calculate the user profiles. As there may be many content platforms 150 that use the MPC cluster 130 to train machine learning models and each content platform 150 may have multiple machine learning models, storing user feature vector data may result in significant data storage requirements.


Using decay techniques can substantially reduce the amount of data that is stored at each client device 110 for the purposes of generating user profiles for training the machine learning models.


Assume that, for a given machine learning model, there are k feature vectors {F1, F2, . . . Fk}, each of which is a n-dimensional vector and their corresponding age (record_age_in_secondsi). The application 112 can compute the user profile using Relationship 1 below:












i
=
1

k


(


e

-



record

_

age



_

in




_

seconds

i




decay

_

rate



_

in



_

seconds





×

F
i


)





Relationship


1







In this relationship, the parameter record_age_in_secondsi is the amount of time in seconds that the profile record has been stored at the client device 110 and the parameter decay_rate_in_seconds is the decay rate of the profile record in seconds (e.g., received in item 6 of the update token Mupdate). In this way, more recent feature vectors carry more weight. This also enables the application 112 to avoid storing feature vectors and only store profile records with constant storage. The application 112 only has to store an n-dimensional vector P and a timestamp user_profile_time for each model identifier, rather than multiple individual feature vectors for each model identifier. This substantially reduces the amount of data that has to be stored at the client device 110, which many client devices typically have limited data storage capacity.


To initialize the n-dimensional vector user profile P and timestamp, the application can set the vector P to a vector of n dimensions where the value of each dimension is zero and set the user_profile_time to epoch. To update the user profile P with a new feature vector Fx at any time, the application 112 can use Relationship 2 below:









P
=



e

-



current

_

time

-


user

_

profile



_

time





decay

_

rate



_

in



_

seconds





×
P

+

F
x






Relationship


2







The application 112 can also update the user profile time to the current time (current time) when updating the user profile with Relationship 2. Note that operations 304 and 308 are omitted if the application 112 calculates user profiles with the above decay rate algorithm.


The application 112 generates shares of the user profile (310). The application 112 can use a pseudorandom function to split the user profile Pi (e.g., the n-dimensional vector P) into shares. That is, the application 112 can use pseudorandom function PRF(Pi) to generate two shares {[Pi, 1],[Pi,2]} of the user profile Pi. The exact splitting can depend on the secret sharing algorithm and crypto library used by the application 112. In some implementations, the application uses Shamir's secret sharing scheme. If shares of one or more labels are being provided, the application 112 can also generate shares of the labels as well.


The application 112 encrypts the shares {[Pi, 1], [Pi,2]} of the user profile Pi (312). For example, as described above, the application 112 can generate composite messages that include shares of the user profile and the label and encrypt the composite messages to obtain encryption results PubKeyEncrypt([Pi,1]∥[labeli,1], MPC1) and PubKeyEncrypt([Pi,2]∥[labeli,2], MPC2). Encrypting the shares using encryption keys of the MPC cluster 130 prevents the content platform 150 from being able to access the user profiles in plaintext. The application 112 transmits the encrypted shares to the content platform (314). Note that operation 314 is omitted if the application 112 transmits the secret shares directly to computing systems MPC1 and MPC2.


Example Process for Generating and Using Machine Learning Models


FIG. 4 is a flow diagram that illustrates an example process 400 for generating a machine learning model. Operations of the process 400 can be implemented, for example, by the MPC cluster 130 of FIG. 1. Operations of the process 400 can also be implemented as instructions stored on one or more computer readable media which may be non-transitory, and execution of the instructions by one or more data processing apparatus can cause the one or more data processing apparatus to perform the operations of the process 400.


The MPC cluster 130 obtains shares of user profiles (402). A content platform 150 can request that the MPC cluster 130 train a machine learning model by transmitting shares of user profiles to the MPC cluster 130. The content platform 150 can access the encrypted shares received from the client devices 110 for the machine learning model over a given time period and upload those shares to the MPC cluster 130.


For example, the content platform 150 can transmit, to computing system MPC1, the encrypted first share of the user profile and the encrypted first share of its label (e.g., PubKeyEncrypt([Pi,1]∥[labeli,1], MPC1) for each user profile Pi. Similarly, the content platform 150 can transmit, to computing system MPC2, the encrypted second share of the user profile and the encrypted second share of its label (e.g., PubKeyEncrypt([Pi,2]∥[labeli,2], MPC2) for each user profile Pi.


In some implementations where the application 112 directly sends secret shares of user profiles to the MPC cluster 130, the content platform 150 can request that the MPC cluster 130 train a machine learning model by transmitting a training request to the MPC cluster 130.


The computing systems MPC1 and MPC2 create random projection planes (404). The computing systems MPC1 and MPC2 can collaboratively create m random projection planes U={U1, U2, . . . , Um}. These random projection planes should remain as secret shares between the two computing systems MPC1 and MPC2. In some implementations, the computing systems MPC1 and MPC2 create the random projection planes and maintain their secrecy using the Diffie-Hellman key exchange technique.


As described in more detail below, the computing systems MPC1 and MPC2 will project their shares of each user profile onto each random projection plane and determine, for each random projection plane, whether the share of the user profile is on one side of the random projection plane. Each computing system MPC1 and MPC2 can then build a bit vector in secret shares from secret shares of the user profile based on the result for each random projection. Partial knowledge of the bit vector for a user, e.g., whether or not the user profile Pi is on one side of the projection plane Uk allows either computing system MPC1 or MPC2 to gain some knowledge about the distribution of Pi, which is incremental to the prior knowledge that the user profile Pi has unit length. To prevent the computing systems MPC1 and MPC2 gaining access to this information (e.g., in implementations in which this is required or preferred for user privacy and/or data security), in some implementations, the random projection planes are in secret shares, therefore neither computing system MPC1 nor MPC2 can access the random projection planes in cleartext. In other implementations, a random bit flipping pattern can be applied over random projection results using secret share algorithms, as described in optional operations 406-408.


To demonstrate how to flip bits via secret shares, assume that there are two secrets x and y whose values are either zero or one with equal probability. An equality operation [x]==[y] will flip the bit of x if y==0 and will keep the bit of x if y==1. In this example, the operation will randomly flip the bit x with 50% probability. This operation can require remote procedure calls (RPCs) between the two computing systems MPC1 and MPC2 and the number of rounds depends on the data size and the secret share algorithm of choice.


Each computing system MPC1 and MPC2 create a secret m-dimensional vector (406). The computing system MPC1 can create a secret m-dimension vector {S1, S2, . . . Sm}, where each element Si has a value of either zero or one with equal probability. The computing system MPC1 splits its m-dimensional vector into two shares, a first share {[S1,1], [S2,1], . . . [Sm,1]} and a second share {[S1,2], [S2,2], . . . [Sm,2]}. The computing system MPC1 can keep the first share secret and provide the second share to computing system MPC2. The computing system MPC1 can then discard the m-dimensional vector {S1, S2, . . . Sm}.


The computing system MPC2 can create a secret m-dimension vector {T1, T2, . . . Tm}, where each element T, has a value of either zero or one. The computing system MPC2 splits its m-dimensional vector into two shares, a first share {[T1,1], [T2,1], . . . [Tm,1]} and a second share {[T1,2], [T2,2], . . . [Tm,2]}. The computing system MPC2 can keep the first share secret and provide the second share to computing system MPC1. The computing system MPC2 can then discard the m-dimensional vector {T1, T2, . . . Tm}.


The two computing systems MPC1 and MPC2 use secure MPC techniques to calculate shares of a bit flipping pattern (408). The computing systems MPC1 and MPC2 can use a secret share MPC equality test with multiple roundtrips between the computing systems MPC1 and MPC2 to compute shares of the bit flipping pattern. The bit flipping pattern can be based on the operation [x]==[y] described above. That is, the bit flipping pattern can be {S1==T1, S2==T2, . . . Sm==Tm}. Let each STi=(Si==Ti). Each STi has a value of either zero or one. After the MPC operation is completed, computing system MPC1 has a first share {[ST1,1], [ST2,1], . . . [STm,1]} of the bit flipping pattern and computing system MPC2 has a second share {[ST1,2], [ST2,2], . . . [STm,2]} of the bit flipping pattern. The shares of each STi enable the two computing systems MPC1 and MPC2 to flip the bits in bit vectors in a way that is opaque to either one of the two computing systems MPC1 and MPC2.


Each computing system MPC1 and MPC2 projects its shares of each user profile onto each random projection plane (410). That is, for each user profile that the computing system MPC1 received a share, the computing system MPC1 can project the share [Pi,1] onto each projection plane U. Performing this operation for each share of a user profile and for each random projection plane Uj results in a matrix R of z×m dimension, where z is the number of user profiles available and m is the number of random projection planes. Each element Ri,j in the matrix R can be determined by computing the dot product between the projection plane Uj and the share [Pi,1], e.g., Ri,j=Uj⊙[P1,1]. The operation ⊙ denotes the dot product of two vectors of equal length.


If bit flipping is used, computing system MPC1 can modify the values of one or more of the elements Ri,j in the matrix using the bit flipping pattern secretly shared between the computing systems MPC1 and MPC2. For each element Ri,j in the matrix R, computing system MPC1 can compute, as the value of the element Ri,j, [STj,1]==sign(Ri,j). Thus, the sign of the element Ri,j will be flipped if its corresponding bit in the bit [STj,1] in the bit flipping pattern has a value of zero. This computation can require multiple RPCs to computing system MPC2.


Similarly, for each user profile that the computing system MPC2 received a share, the computing system MPC2 can project the share [Pi,2] onto each projection plane Uj. Performing this operation for each share of a user profile and for each random projection plane Uj results in a matrix R′ of z×m dimension, where z is the number of user profiles available and m is the number of random projection planes. Each element Ri,j′ in the matrix R′ can be determined by computing the dot product between the projection plane Uj and the share [Pi,2], e.g., Ri,′j=Uj⊙[Pi,2]. The operation ⊙ denotes the dot product of two vectors of equal length.


If bit flipping is used, computing system MPC2 can modify the values of one or more of the elements Ri,j′ in the matrix using the bit flipping pattern secretly shared between the computing systems MPC1 and MPC2. For each element Ri,j′ in the matrix R, computing system MPC2 can compute, as the value of the element Ri,j′, [STj,2]==sign(Ri,j′). Thus, the sign of the element Ri,j′ will be flipped if its corresponding bit in the bit STj in the bit flipping pattern has a value of zero. This computation can require multiple RPCs to computing system MPC1.


The computing systems MPC1 and MPC2 reconstruct bit vectors (412). The computing systems MPC1 and MPC2 can reconstruct the bit vectors for the user profiles based on the matrices R and R′, which have exactly the same size. For example, computing system MPC1 can send a portion of the columns of matrix R to computing system MPC2 and computing system MPC2 can send the remaining portion of the columns of matrix R′ to MPC1. In a particular example, computing system MPC1 can send the first half of the columns of matrix R to computing system MPC2 and computing system MPC2 can send the second half of the columns of matrix R′ to MPC1. Although columns are used in this example for horizontal reconstruction and are preferred to protect user privacy, rows can be used in other examples for vertical reconstruction.


In this example, computing system MPC2 can combine the first half of the columns of matrix R′ with the first half of the columns of matrix R received from computing system MPC1 to reconstruct the first half (i.e., m/2 dimension) of bit vectors in cleartext. Similarly, computing system MPC1 can combine the second half of the columns of matrix R with the second half of the columns of matrix R′ received from computing system MPC2 to reconstruct the second half (i.e. m/2 dimension) of bit vectors in cleartext. Conceptually, the computing systems MPC1 and MPC2 have now combined corresponding shares in two matrixes R and R′ to reconstruct bit matrix B in plaintext. This bit matrix B would include the bit vectors of the projection results (projected onto each projection plane) for each user profile for which shares were received from the content platform 150 for the machine learning model. Each one of the two servers in the MPC cluster 130 owns half of the bit matrix B in plaintext.


However, if bit flipping is used, the computing systems MPC1 and MPC2 have flipped bits of elements in the matrices R and R′ in a random pattern fixed for the machine learning model. This random bit flipping pattern is opaque to either of the two computing systems MPC1 and MPC2 such that neither computing system MPC1 nor MPC2 can infer the original user profiles from the bit vectors of the project results. The crypto design further prevents MPC1 or MPC2 from inferring the original user profiles by horizontally partitioning the bit vectors, i.e. computing system MPC1 holds the second half of bit vectors of the projection results in plaintext and computing system MPC2 holds the first half of bit vectors of the projection results in plaintext.


The computing systems MPC1 and MPC2 generate machine learning models (414). The computing system MPC1 can generate a k-NN model using the second half of the bit vectors. Similarly, computing system MPC2 can generate a k-NN model using the first half of the bit vectors. Generating the models using bit flipping and horizontal partitioning of the matrices applies the defense-in-depth principle to protect the secrecy of the user profiles used to generate the models.


In general, each k-NN model represents cosine similarities (or distances) between the user profiles of a set of users. The k-NN model generated by computing system MPC1 represents the similarity between the second half of the bit vectors and the k-NN model generated by computing system MPC2 represents the similarity between the first half of the bit vectors. For example, each k-NN model can define the cosine similarity between its half of the bit vectors.


The two k-NN models generated by the computing systems MPC1 and MPC2 can be referred to as a k-NN model, which has a unique model identifier as described above. The computing systems MPC1 and MPC2 can store their models and shares of the labels for each user profile used to generate the models. The content platform 150 can then query the models to make inferences for user groups for a user.


Example Process for Using Machine Learning Model to Infer User Groups


FIG. 5 is a flow diagram that illustrates an example process 500 for adding a user to user groups using machine learning models. Operations of the process 500 can be implemented, for example, by the MPC cluster 130 and the client device 110 of FIG. 1, e.g., the application 112 running on the client device 110. Operations of the process 500 can also be implemented as instructions stored on one or more computer readable media which may be non-transitory, and execution of the instructions by one or more data processing apparatus can cause the one or more data processing apparatus to perform the operations of the process 500.


The MPC cluster 130 receives an inference request for a given user profile (502). An application 112 running on a user's client device 110 can transmit the inference request to the MPC cluster 130, e.g., in response to a request from a content platform 150. For example, the content platform 150 can transmit, to the application 112, an upload token Minfer to request that the application 112 submit the inference request to the MPC cluster 130. The inference request can be to query whether the user should be added to any number of user groups.


The inference request token Minfer can include shares of the given user profile of the user, the model identifier for the machine learning model (e.g., k-NN model) and the owner domain to be used for the inference, a number k of nearest neighbors of the given user profile to be used for the inference, additional signals (e.g., contextual or digital component signals), the aggregation function to be used for the inference and any aggregation function parameters to be used for the inference, and the signature over all the above information created by the owner domain using an owner domain confidential privacy key.


As described above, to prevent leaking the given user profile Pi in plaintext form to either computing system MPC1 or MPC2, and thereby preserve user privacy, the application 112 can split the given user profile Pi into two shares [Pi,1] and [Pi,2] for MPC1 and MPC2, respectively. The application 112 can then send a single inference request to computing system MPC1 with the first share [Pi,1] of the given user profile and an encrypted version of the second share, e.g., PubKeyEncrypt([Pi,2], MPC2) of the given user profile. The inference request may also include the inference request token Minfer so that the MPC cluster 130 can authenticate the inference request. By sending an inference request that includes the first share and the encrypted second share, the number of outgoing requests sent by the application 112 is reduced, resulting in computational, bandwidth, and battery savings at the client device 110.


In other implementations, the application 112 can send the first share [Pi,1] of the given user profile to computing system MPC1 and the second share [Pi,2] of the given user profile to computing system MPC2. By sending the second share [Pi,2] of the given user profile to computing system MPC2 without going through computing system MPC1, the second share does not need to be encrypted to prevent computing system MPC1 from accessing the second share [Pi,2] of the given user profile.


Each computing system MPC1 and MPC2 identifies the k nearest neighbors to the given user profile in secret share representation (504). The computing system MPC1 can compute its half of a bit vector of the given user profile using the first share [Pi,1] of the given user profile. To generate the bit vector, computing system MPC1 can use operations 410 and 412 of the process 400 of FIG. 4. That is, computing system MPC1 can use the random projection vectors generated for the k-NN model to project the share [Pi,1] of the given user profile and create a secret share of a bit vector for the given user profile. If bit flipping was used to generate the k-NN model, the computing system MPC1 can then use the first share {[ST1,1], [ST2,1], . . . [STm,1]} of the bit flipping pattern that was used to generate the k-NN model to modify the elements of the secret share of a bit vector for the given user profile.


Similarly, the computing system MPC1 can provide, to computing system MPC2, the encrypted second share PubKeyEncrypt([Pi,2], MPC2) of the given user profile. The computing system MPC2 can decrypt the second share [Pi,2] of the given user profile using its private key and compute its half of the bit vector for the given user profile using the second share [Pi,2] of the given user profile. That is, computing system MPC2 can use the random projection vectors generated for the k-NN model to project the share [Pi,2] of the given user profile and create a bit vector for the given user profile. If bit flipping was used to generate the k-NN model, the computing system MPC2 can then use the second share {[ST1,2], [ST2,2], . . . [STm,2]} of the bit flipping pattern that was used to generate the k-NN model to modify the elements of the bit vector for the given user profile. The computing systems MPC1 and MPC2 then reconstruct the bit vector with horizontal partition, as described in operation 412 in FIG. 4. After the completion of reconstruction, computing system MPC1 has the first half of the overall bit vector for the given user profile and computing system MPC2 has the second half of the overall bit vector for the given user profile.


Each computing system MPC1 and MPC2 uses its half of the bit vector for the given user profile and its k-NN model to identify the k′ nearest neighbor user profiles, where k′=a×k, where a is empirically determined based on actual production data and statistical analysis. For example a=3 or another appropriate number. The computing system MPC1 can compute a Hamming distance between the first half of the overall bit vector and the bit vector for each user profile of the k-NN model. The computing system MPC1 then identifies the k′ nearest neighbors based on the computed Hamming distances, e.g., the k′ user profiles having the lowest Hamming distances. In other words, the computing system MPC1 identifies a set of nearest neighbor user profiles based on a share of a given user profile and a k-nearest neighbor model trained using multiple user profiles. An example result in tabular form is shown in Table 5 below.












TABLE 5






Hamming Distance
Shares of User



Row ID
(in plaintext)
Profile
Shares of Label







i
di,1
[Pi,1]
[labeli,1]


. . .
. . .
. . .
. . .









In Table 5, each row is for a particular nearest neighbor user profile and includes the Hamming distance between the first half of the bit vector for each user profile and the bit vector for the given user profile computed by computing system MPC1. The row for a particular nearest neighbor user profile also includes the first share of that user profile and the first share of the label associated with that user profile.


Similarly, the computing system MPC2 can compute a Hamming distance between the second half of the overall bit vector and the bit vector for each user profile of the k-NN model. The computing system MPC2 then identifies the k′ nearest neighbors based on the computed Hamming distances, e.g., the k′ user profiles having the lowest Hamming distances. An example result in tabular form is shown in Table 5 below.












TABLE 6






Hamming Distance
Shares of User



Row ID
(in plaintext)
Profile
Shares of Label







j
dj,2
[Pj,2]
[labelj,2]


. . .
. . .
. . .
. . .









In Table 6, each row is for a particular nearest neighbor user profile and includes the Hamming distance between that user profile and the given user profile computed by computing system MPC2. The row for a particular nearest neighbor user profile also includes the second share of that user profile and the second share of the label associated with that user profile.


The computing systems MPC1 and MPC2 can exchange the list of row identifiers (row ID) and Hamming distance pairs with each other. Thereafter, each computing system MPC1 and MPC2 can independently select k nearest neighbors with the same algorithm and input data. For example, computing system MPC1 can find row identifiers common to partial query results from both computing systems MPC1 and MPC2. For each i in the common row identifiers, computing system MPC1 calculates combined Hamming distance di from the two partial Hamming distances, e.g., di=di,1+di,2. The computing system MPC1 can then order the common row identifiers based on the combined Hamming distance di and select the k nearest neighbors. The row identifiers for the k nearest neighbors can be represented as ID={id1, . . . idk}. It can be proven that if a is sufficiently large, the k nearest neighbors determined in the above algorithm are the true k nearest neighbors with high probability. However, larger value for a leads to high computation cost. In some implementations, computing systems MPC1 and MPC2 engage in Private Set Intersection (PSI) algorithms to determine row identifiers common to partial query results from both computing systems MPC1 and MPC2. Furthermore, in some implementations, MPC1 and MPC2 engage in enhanced Private Set Intersection (PSI) algorithms to calculate di=di,1+di,2 for row identifiers common to partial query results from both computer systems MPC1 and MPC2, and reveal nothing to either MPC1 or MPC2 but the top k nearest neighbors determined by di.


A determination is made whether to add the user to a user group (506). This determination can be made based on the k nearest neighbor profiles and their associated labels. The determination is also based on the aggregation function used and any aggregation parameters for that aggregation function. The aggregation functions can be chosen based on the nature of the machine learning problem, for example, binary classification, regression (e.g., using arithmetic mean or root mean square), multiclass classification, and weighted k-NN. Each way of determining whether to add a user to a user group can include different interactions between the MPC cluster 130 and the application 112 running on the client 110, as described in more detail below.


If a determination is made to not add the user to the user group, the application 112 may not add the user to the user group (508). If a determination is made to add the user to the user group, the application 112 can add the user to the user group, e.g., by updating the user group list stored at the client device 110 to include the user group identifier of the user group (510).


Example Binary Classification Inference Techniques

For binary classification, the inference request can include, as aggregation function parameters, a threshold, Ltrue, and Lfalse. The label values are Boolean type, i.e. either true or false. The threshold parameter can represent a threshold percentage of k nearest neighbor profiles that must have a label of true value in order for the user to be added to the user group Ltrue. Otherwise the user will be added to user group Lfalse. In one approach, the MPC cluster 130 could instruct the application 112 to add the user to the user group Ltrue (Lfalse otherwise) if the number of nearest neighbor user profiles that has a label value that is true is greater than a product of the threshold and k. However, computing system MPC1 would learn the inference result, e.g., the user group that the user should join.


To preserve user privacy, the inference request can include the threshold in plaintext, a first share [Ltrue,1] and [Lfalse,1] for computing system MPC1, and an encrypted second share PubKeyEncrypt([Ltrue,2]∥[Lfalse,2]∥application_public_key, MPC2) for computing system MPC2. In this example, the application 112 can generate a composite message from [Ltrue,2], [Lfasle,2] and the public key of the application 112, as denoted by the symbols ∥, and encrypt this composite message using a public key of computing system MPC2. The inference response from computing system MPCi to the application 112 can include a first share of the inference result [Lresult,1] determined by computing system MPC1 and a second share of the inference result [Lresult,2] determined by computing system MPC2.


To prevent the second share from being accessed by computing system MPC1 and therefore enabling computing system MPC1 to obtain the inference result in plaintext, computing system MPC2 can send an encrypted (and optionally digitally signed) version of the second share of the inference result [Lresult,2], e.g., PubKeySign(PubKeyEncrypt([Lresult,2], application_public_key), MPC2) to computing system MPC1 for inclusion in the inference response sent to the application 112. In this example, the application 112 can verify the digital signature using the public key of computing system MPC2 that corresponds to the private key of computing system MPC2 used to generate the digital signature, and decrypt the second share of the inference result [Lresult,2] using the private key of the application 112 corresponding to the public key (application_public_key) used to encrypt the second share of the inference result [Lresult,2].


The application 112 can then reconstruct the inference result Lresult from the first share [Lresult,1] and the second share [Lresult,2]. Using the digital signature enables the application 112 to detect falsification of the result from computing system MPC2, e.g., by computing system MPC1. Depending on the level of security desired, what parties operate the computing systems of the MPC cluster 130, and the security model assumed, the digital signature may not be required.


The computing systems MPCi and MPC2 can use MPC techniques to determine the shares [Lresult,1] and [Lresult,2] of the binary classification result. In binary classification, the value of label1 for a user profile is either zero (false) or one (true). Assuming that the selected k nearest neighbors are identified by identifiers {id1, . . . idk}, the computing systems MPC1 and MPC2 can calculate a sum of the labels (sum_of_labels) for the k nearest neighbor user profiles, where the sum is represented by Relationship 3 below:





sum_of_labels=Σi∈{id1, . . . idk} labeli  Relationship 3:


To determine the sum, computing system MPC1 sends ID (i.e., {id1, . . . idk}) to computing system MPC2. The computing system MPC2 can verify that the number of row identifiers in ID is greater than a threshold to enforce k-anonymity. The computing system MPC2 can then calculate a second share of the sum of labels [sum_of_labels2] using Relationship 4 below:





[sum_of_labels2]=ΣI∈{id1, . . . idk}[labeli,2]  Relationship 4:


The computing system MPC1 can also calculate a first share of the sum of labels [sum_of_labels1] using Relationship 5 below:





[sum_of_labels1]=Σi∈{id1, . . . idk}[labeli,1]  Relationship 5:


If the sum of the labels sum_of_labels is confidential information that the computing systems MPC1 and MPC2 should know as little as possible, computing system MPC1 can calculate whether the first share of the sum_of_labels [sum_of_labels1] is below the threshold, e.g., [below_threshold1]=[sum_of_labels1]<threshold×k. Similarly, computing system MPC2 can calculate whether the second share of the sum of labels [sum_of_labels2] is below the threshold, e.g., [below_threshold2]=[sum_of_labels2]<threshold×k. The computing system MPC1 can proceed to calculate inference result [Lresult,1] by [below_threshold1]×[Lfalse,1]+(1−[below_threshold1])×[Ltrue,1]. Similarly, computing system MPC2 can calculate [Lresult,2] by [below_threshold2]×[Lfalse,2]+(1−[below_threshod2])×[Ltrue,2].


If the sum of the labels sum_of_labels is not confidential information, computing systems MPC1 and MPC2 can reconstruct the sum_of_labels from [sum_of_labels1] and [sum_of_labels2]. The computing systems MPC1 and MPC2 can then set the parameter below_threshold to sum_of_labels <threshold×k, e.g., a value of one if it is below the threshold or a value of zero if not below the threshold.


After computing the parameter below_threshold, the computing systems MPC1 and MPC2 can proceed to determine the inference result Lresult. For example, computing system MPC2 can set [Lresult,2] to either [Ltrue,2] or [Lfalse,2] according to the value of below_threshold. For example, computing system MPC2 can set [Lresult,2] to [Ltrue,2] if the sum of labels is not below the threshold or to [Lfalse,2] if the sum of labels is below the threshold. The computing system MPC2 can then return an encrypted second share of the inference result (PubKeyEncrypt(Lresult,2], application_public_key)) or a digitally signed version of this result to computing system MPC1.


Similarly, computing system MPC1 can set [Lresult,1] to either [Ltrue,1] or [Lfalse,1] according to the value of below_threshold. For example, computing system MPC1 can set [Lresult,1] to [Ltrue,1] if the sum of labels is not below the threshold or to [Lfalse,1] if the sum of labels is below the threshold. The computing system MPC1 can transmit the first share of the inference result [Lresult,1] and the encrypted second share of the inference result [Lresult,2] as an inference response to the application 112. The application 112 can then compute the inference result based on the two shares, as described above.


Example Multiclass Classification Inference Techniques

For multiclass classification, the label associated with each user profile can be categorical feature. The content platform 150 can specify a lookup table that maps any possible categorical value to a corresponding user group identifier. The lookup table can be one of the aggregation function parameters included in the inference request.


Within the k nearest neighbors found, the MPC cluster 130 finds the most frequent label value. The MPC cluster 130 can then find, in the lookup table, the user group identifier corresponding to the most frequent label value and request that the application 112 add the user to the user group corresponding to the user group identifier, e.g., by adding the user group identifier to the user group list stored at the client device 110.


Similar to binary classification, it may be preferable to hide the inference result Lresult from the computing systems MPC1 and MPC2. To do so, the application 112 or the content platform 150 can create two lookup tables that each maps categorical values to a respective share of the inference result Lresult. For example, the application can create a first lookup table that maps the categorical values to a first share [Lresult1] and a second lookup table that maps the categorical values to a second share [Lresult2]. The inference request from the application to computing system MPC1 can include the first lookup table in plaintext for computing system MPC1 and an encrypted version of the second lookup table for computing system MPC2. The second lookup table can be encrypted using a public key of computing system MPC2. For example, a composite message that includes the second lookup table and a public key of the application can be encrypted using the public key of the computing system MPC2, e.g., PubKeyEncrypt(lookuptable2∥application_public_key, MPC2).


The inference response sent by computing system MPC1 can include the first share [Lresult1] of the inference result generated by the computing system MPC1. Similar to binary classification, to prevent the second share from being accessed by computing system MPC1 and therefore enabling computing system MPC1 to obtain the inference result in plaintext, computing system MPC2 can send an encrypted (and optionally digitally signed) version of the second share of the inference result [Lresult,2], e.g., PubKeySign(PubKeyEncrypt([Lresult2], application_public_key), MPC2) to computing system MPC1 for inclusion in the inference result sent to the application 112. The application 112 can reconstruct the inference result Lresult from [Lresult1] and [Lresult2].


Assume there are w valid labels {l1, l2, . . . lw} for a multiclass classification problem. To determine the shares [Lresult1] and [Lresult2] of the inference result Lresult in multiclass classification, computing system MPC1 sends ID (i.e., {id1, . . . idk}) to computing system MPC2. The computing system MPC2 can verify that the number of row identifiers in ID is greater than a threshold to enforce k-anonymity. In general, the k in k-NN may be significantly larger than the k in k-anonymity. The computing system MPC2 can then calculate a second frequency share [frequencyj,2] of the j-th label [lj,2] which is defined using Relationship 6 below.





[frequencyj,2]=Σi∈{id1, . . . idk}[labeli,2]==lj)  Relationship 6:


Similarly, computing system MPC1 calculates a first frequency share [frequencyj,1] of the j-th label [lj,1] which is defined using Relationship 7 below.





[frequencyj,1]=Σi∈{id1, . . . idk}[labeli,1]==lj)  Relationship 7:


Assuming that the frequency of a label (frequencyi) within the k nearest neighbors is not sensitive, the computing systems MPC1 and MPC2 can reconstruct frequency, from the two shares [frequencyi,1] and [frequencyi,2] for that label. The computing systems MPC1 and MPC2 can then determine an index parameter (index) where frequencyindex has the largest value, e.g., index=argmaxi(frequencyi).


The computing system MPC2 can then lookup, in its lookup table, the share [Lresult,2] corresponding to the label having the highest frequency and return PubKeyEncrypt([Lresult,2], application_public_key) to the computing system MPC1. The computing system MPC1 can similarly lookup, in its lookup table, the share [Lresult,1] corresponding to the label having the highest frequency. The computing system MPC1 can then send, to the application 112, an inference response that includes the two shares (e.g., [Lresult,1] and PubKeyEncrypt([Lresult,2], application_public_key). As described above, the second share can be digitally signed by computing system MPC2 to prevent computing system MPC1 from falsifying the response of computing system MPC2. The application 112 can then compute the inference result based on the two shares, as described above, and add the user to the user group identified by the inference result.


Example Regression Inference Techniques

For regression, the label associated with each user profile P must be numerical. The content platform 150 can specify an ordered list of thresholds, e.g., (−∞<t0<t1< ⋅ ⋅ ⋅ <tn<∞), and a list of user group identifiers, e.g., {L0, L1, . . . Ln, Ln+1}. In addition, the content platform 150 can specify an aggregation function, e.g., arithmetic mean or root mean square.


Within the k nearest neighbors found, the MPC cluster 130 calculates the mean (result) of the label values and then looks up the mapping using the result to find the inference result Lresult. For example, the MPC cluster 130 can use Relationship 8 below to identify the label based on the mean of the label values:





If result≤t0,Lresult←L0;





If result>tn,Lresult←Ln+1;





If tx<result≤tx+1,Lresult←Lx+1  Relationship 8:


That is, if the result is less than or equal to threshold to, the inference result Lresult is L0. If the result is greater than threshold tn, the inference result Lresult is Lni. Otherwise, if the result is greater than threshold tx and less than or equal to threshold tx, the inference result Lresult is Lx+1. The computing system MPC1 then requests that the application 112 add the user to the user group corresponding to the inference result Lresult, e.g., by sending an inference response that includes the inference result Lresult to the application 112.


Similar to the other classification techniques described above, the inference result Lresult can be hidden from the computing systems MPC1 and MPC2. To do so, the inference request from the application 112 can include first share of the labels [Li,1] for computing system MPC1 and encrypted second shares of the labels [Li,2] (e.g., PubKeyEncrypt([L0,2∥ . . . ∥Ln+1,2∥application_public_key, MPC2)) for computing system MPC2.


The inference result sent by computing system MPC1 can include the first share [Lresult1] of the inference result generated by the computing system MPC1. Similar to binary classification, to prevent the second share from being accessed by computing system MPC1 and therefore enabling computing system MPC1 to obtain the inference result in plaintext, computing system MPC2 can send an encrypted (and optionally digitally signed) version of the second share of the inference result [Lresult,2], e.g., PubKeySign(PubKeyEncrypt([Lresult,2], application_public_key), MPC2) to computing system MPC1 for inclusion in the inference result sent to the application 112. The application 112 can reconstruct the inference result Lresult from [Lresult,1] and [Lresult,2].


When the aggregation function is arithmetic mean, the computing systems MPC1 and MPC2 compute the sum of the labels sum_of_labels, similar to binary classification. If the sum of the labels is not sensitive, the computing systems MPC1 and MPC2 can calculate the two shares [sum_of_labels1] and [sum_of_labels2] and then reconstruct sum_of_labels based on the two shares. The computing systems MPC1 and MPC2 can then compute the average of the labels by dividing the sum of the labels by the quantity of the nearest neighbor labels, e.g., by k.


The computing system MPC1 can then compare the average to the thresholds using Relationship 8 to identify the first share of the label corresponding to the average and set the first share [Lresult,1] to the first share of the identified label. Similarly, the computing system MPC2 can compare the average to the thresholds using Relationship 8 to identify the second share of the label corresponding to the average and set the second share [Lresult,2] to the second share of the identifier label. The computing system MPC2 can encrypt the second share [Lresult,2] using the public key of the application 112, e.g., PubKeyEncrypt([Lresult,2], application_public_key) and send the encrypted second share to computing system MPC1. The computing system MPC1 can provide the first share and the encrypted second share (which can optionally be digitally signed as described above) to the application 112. The application 112 can then add the user to the user group identified by the label (e.g., user group identifier) Lresult.


If the sum of the labels is sensitive, the computing systems MPC1 and MPC2 may not be able to construct sum_of_labels in plaintext. Instead, computing system MPC1 can calculate a mask [maski,1]=[sum_of_labels1]>ti×k for ∀i∈[0, n]. This computation can require multiple roundtrips between the computing systems MPC1 and MPC2. Next, computing system MPC1 can calculate [acci,1]=(1==Σj=0i[maskj,1]) and computing system MPC2 can calculate [acci,2]=(1==Σj=0i[maskj,2]). The equality test in this operation can require multiple roundtrips between the computing systems MPC1 and MPC2.


In addition, computing system MPC1 can calculate [use_default1]=(0==Σj=0n[maskj,1]) and computing system MPC2 can calculate [use_default2]=(0==Σj=0n[maskj,2]). The MPC cluster 130 would then return Li if and only if acci==1, for ∀i∈[0, n] and would return Ln+1 if use_default==1. This condition can be represented in Relationship 9 below.





result=use_default×Ln+1i=0nacci×Li  Relationship 9:


The corresponding cryptographic implementation can be represented by Relationships 10 and 11 below.





[Lresult,1]=[use_default1Ln+1i=0n([acci,1Li)  Relationship 10:





[Lresult,2]=[use_default2Ln+1i=0n([acci,2Li)  Relationship 11:


These computations do not require any roundtrip computations between the computing systems MPC1 and MPC2 if Li are in cleartext, and involve one roundtrip computation if Li are in secret shares. The computing system MPC1 can provide the two shares of the result (e.g., [Lresult,1] and [Lresult,2] to the application 112, with the second share encrypted and optionally digitally signed by computing system MPC2 as described above. In this way, the application 112 can determine the inference result Lresult without the computing systems MPC1 or MPC2 learning anything about the immediate or final result.


For root mean square, computing system MPC1 sends ID (i.e., {id1, . . . idk}) to computing system MPC2. The computing system MPC2 can verify that the number of row identifiers in ID is greater than a threshold to enforce k-anonymity. The computing system MPC2 can calculate a second share of a sum_of_square_labels parameter (e.g., the sum of the squares of the label values) using Relationship 12 below.





[sum_of_square_labels2]=Σi∈{id1, . . . idk}[labeli,2]2  Relationship 12:


Similarly, computing system MPC1 can calculate a first share of the sum_of_square_labels parameter using Relationship 13 below.





[sum_of_square_labels1]=Σi∈{id1, . . . idk}[labeli,1]2  Relationship 13:


Assuming that the sum_of_square_labels parameter is not sensitive, the computing systems MPC1 and MPC2 can reconstruct the sum_of_square_labels parameter from the two shares [sum_of_square_labels1] and [sum_of_square_labels2]. The computing systems MPC1 and MPC2 can compute the root mean square of the labels by dividing the sum_of_squares_labels by the quantity of the nearest neighbor labels, e.g., by k, then calculating the square root.


Regardless of whether the average is calculated via arithmetic mean or root mean square, the computing system MPC1 can then compare the average to the thresholds using Relationship 8 to identify the label corresponding to the average and set the first share [Lresult,1] to the identified label. Similarly, the computing system MPC2 can compare the average to the thresholds using Relationship 8 to identify the label (or secret share of the label) corresponding to the average and set the second share [Lresult,2] to the identifier label (or the secret share of the identifier label). The computing system MPC2 can encrypt the second share [Lresult,2] using the public key of the application 112, e.g., PubKeyEncrypt([Lresult,2], application_public_key) and send the encrypted second share to computing system MPC1. The computing system MPC1 can provide the first share and the encrypted second share (which can optionally be digitally signed as described above) to the application 112 as the inference result. The application 112 can then add the user to the user group identified by the label (e.g., user group identifier) of the Lresult. If the sum_of_square_labels parameter is sensitive, the computing systems MPC1 and MPC2 can perform a similar cryptographic protocol as used in the arithmetic mean example to compute the shares of the inference result.


In the above techniques to infer the result of classification and regressions problems, all k nearest neighbors have equal influence, e.g., equal weight, over the final inference result. For many classification and regression problems, model quality can be improved if each of the k neighbors is assigned a weight that monotonically decreases when the Hamming distance between the neighbor and the query parameter Pi increases. A common kernel function with this property is Epanechnikov (parabolic) kernel function. Both the Hamming distance and the weight can be calculated in plaintext.


Sparse Feature Vector User Profiles

When features of electronic resources are included in user profiles and used to generate the machine learning models, the resulting feature vectors can include high cardinality categorical features, such as domains, URLs, and IP addresses. These feature vectors are sparse, with most of the elements having a value of zero. The application 112 could split the feature vectors into two or more dense feature vectors, but the machine learning platform would consume too much client device upload bandwidth to be practical. To prevent this problem, the systems and techniques described above can be adapted to better handle sparse feature vectors.


When providing feature vectors for events to the client device, computer-readable code (e.g., scripts) of a content platform 150 that is included in an electronic resource can invoke an application (e.g., browser) API to specify the feature vector for the event. This code, or content platform 150, can determine whether (some part of) the feature vector is dense or sparse. If the feature vector (or some part of it) is dense, the code can pass in a vector of numerical values as the API parameter. If the feature vector (or some part of it) is sparse, the code can pass in a map, e.g., indexed key/value pairs for those feature elements with non-zero feature values, where the keys are the names or indices of such feature elements. If the feature vector (or some part of it) is sparse, and the non-zero feature values are always the same value, e.g., 1, the code can pass in a set, whose elements are the names or indices of such feature elements.


When aggregating feature vectors to generate a user profile, the application 112 can handle dense and sparse feature vectors differently. The user profile (or some part of it) calculated from dense vectors remains to be a dense vector. The user profile (or some part of it) calculated from maps remains to be a map, until the fill rate is sufficiently high that map does not save storage cost anymore. At that point, the application 112 will convert the sparse vector representation into dense vector representation.


In some implementations, the application 112 can classify some of the feature vectors, or some parts of the feature vectors as sparse feature vectors and some as dense feature vectors. The application 112 can then handle each type of feature vector differently in generating the user profile and/or the shares of the user profile.


The user profile (or some part of it) calculated from sets can be a map, if the aggregation function is sum. For example, each feature vector can have a categorical feature “domain visited”. The aggregation function, i.e. sum, will calculate the number of times that the user visited the publisher domain. The user profile (or some part of it) calculated from sets can remain to be a set, if the aggregation function is logical OR. For example, each feature vector can have a categorical feature “domain visited”. The aggregation function, i.e. logical OR, will calculate all publisher domains that the user visited, regardless of the frequency of visits.


To send user profiles to the MPC cluster 130 for ML training and prediction, the application 112 may split the dense part of user profiles with any standard crypto libraries that support secret shares. To split the sparse part of user profiles without significantly increasing the client device upload bandwidth and computation cost, a Function Secret Sharing (FSS) technique can be used. In this example, the content platform 150 assigns a unique index to each possible element in the sparse part of the user profile, starting with 1 sequentially. Assume that the valid range of the indices are in the range of [1, N] inclusively.


For the i-th element with non-zero value Pi in a user profile calculated by the application, 1≤i≤N, the application 112 can create two Pseudo-Random Functions (PRF) gi and hi with the following properties:






g
i(j)+hi(j)=0 for any j where 1≤j≤N and j≠i






g
i(j)+hi(j)=Pi otherwise.


Using FSS, either g, or h, can be represented concisely, e.g., by log2(N)×size_of_tag bits and it is impossible to infer i or Pi from either gi or hi. To prevent brute force security attack, size_of_tag is typically 96 bits or larger. Out of the N dimensions, assume that there are n dimensions with non-zero value, where n<<N. For each of the n dimensions, the application 112 can construct two pseudo-random functions g and h as described above. Furthermore, the application 112 can package the concise representation of all n functions g into a vector G, and package the concise representation of n functions h into another vector H in the same order.


In addition, the application 112 can split the dense part of the user profile P into two additive secret shares [P1] and [P2]. The application 112 can then send [P1] and G to computing system MPC1 and send [P2] and H to MPC2. Transmitting G requires |G|×log2(N)×size_of_tag=n×log2(N)×size_of_tag bits, which may be far smaller than N bits needed if the application 112 transmits the sparse part of the user profile in a dense vector, when n<<N.


When computing system MPC1 receives g1 and computing system MPC2 receives h1, the two computing systems MPC1 and MPC2 can create Shamir's secret shares independently. For any j where 1≤j≤N, computing system MPC1 creates a point on two-dimensional coordinate [1, 2×gi(j)] and computing system MPC2 creates a point on two-dimensional coordinate [−1, 2×hi(j)]. If the two computing systems MPC1 and MPC2 collaboratively construct a line y=a0+a1×x that passes through both points, Relationships 14 and 15 are formed.





gi(j)=a0+a1  Relationship 14:





hi(j)=a0−a1  Relationship 15:


If the two relationships are added together, it results in 2×gi(j)+2×hi(j)=(a0+a1)+(a0−a1), which simplifies to a0=gi(j)+hi(j). Therefore, [1, 2×gi(j)] and [−1, 2×hi(j)] are two secret shares of the i-th non-zero element in the sparse array, i.e., Pi.


During the random projection operations of the machine learning training process, the computing system MPC1 can independently assemble its vectors of secret shares for the user profile from both [P1] and G. Per the above description, it is known that |G|=n, where n is the number of non-zero elements in the sparse part of the user profile. In addition, it is known that the sparse part of the user profile is N dimension, where n<<N.


Assume that G={g1, . . . gn}. For the j-th dimension where 1≤j≤N, and 1≤k≤n let [SPj,1]=Σk=1ngk(j). Similarly, let H={h1, . . . hn}. The computing system MPC2 can independently calculate [SPj,2]=Σk=1nhk(j). It is easy to prove that [SPj,1] and [SPj,2] are secret shares of SPj, i.e., the secret value of the j-th element in the original sparse part of the user profile.


Let [SP1]={[SP1,1], . . . [SPN,1]}, i.e., the reconstructed secret share in dense representation of the sparse part of the user profile. By concatenating [P1] and [SP1], computing system MPC1 can reconstruct the full secret share of the original user profile. The computing system MPC1 can then randomly project [P1]∥[SP1]. Similarly, computing system MPC2 can randomly project [P2]∥[SP2]. After projection, the techniques described above can be used to generate the machine learning models in a similar manner.



FIG. 6 is a conceptual diagram of an exemplary framework for generating an inference result for a user profile in a system 600. More particularly, the diagram depicts random projection logic 610, a first machine learning model 620, and final result calculation logic 640 that collectively make up system 600. In some implementations, the functionality of the system 600 may be provided in a secure and distributed manner by way of multiple computing systems in an MPC cluster. The techniques described in reference to system 600 may, for example, be similar to those which have been described above with reference to FIGS. 2-5. For example, the functionality associated with random projection logic 610 may correspond to that of one or more of the random projection techniques described above with reference to FIGS. 2 and 4. Similarly, in some examples, the first machine learning model 620 may correspond to one or more of the machine learning models described above with reference to FIGS. 2, 4, and 5, such as one or more of those described above in connection with steps 214, 414, and 504. In some examples, encrypted label data set 626, which may be maintained and utilized by the first machine learning model 620 and stored in one or more memory units, can include at least one true label for each user profile used to generate or train, or evaluate the quality of training or fine-tune the process of training the first machine learning model 620, such as those which may be associated with k nearest neighbor profiles as described above with reference to step 506 of FIG. 5. That is, the encrypted label data set 626 may include at least one true label for each of n user profiles, where n is the total number of user profiles that were used to train the first machine learning model 620. For example, the encrypted label data set 626 may include at least one true label (Lj) for the jth user profile (Pj) in the n user profiles, at least one true label (Lk) for the kth user profile (Pk) in the n user profiles, at least one true label (Ll) for the lth user profile (Pl) in the n user profiles, where 1≤j, k, l≤n, and so on. Such true labels, as associated with the user profiles that were used to generate or train the first machine learning model 620 and included as part of the encrypted label data set 626, can be encrypted, e.g., represented as secret shares. Additionally, in some examples, final result calculation logic 640 may correspond to logic employed in connection with performing one or more operations for generating an inference result, such as one or more of those described above with reference to step 218 in FIG. 2. The first machine learning model 620 and final result calculation logic 640 can be configured to employ one or more inference techniques including binary classification, regression, and/or multiclass classification techniques.


In the example of FIG. 6, the system 600 is depicted as performing one or more operations at inference time. Random projection logic 610 can be employed to apply a random projection transformation to a user profile 609 (Pi) to obtain a transformed user profile 619 (Pi′). The transformed user profile 619, as obtained by employing random projection logic 610, can be in cleartext. For example, random projection logic 610 may be employed at least in part to obfuscate feature vectors, such as feature vectors included or indicated in the user profile 609 and other user profiles, with random noises to protect user privacy.


The first machine learning model 620 can be trained and subsequently leveraged to receive transformed user profile 619 as input and generate at least one predicted custom-characterlabel 629 (custom-character) in response thereto. The at least one predicted label 629, as obtained using the first machine learning model 620, can be encrypted. In some implementations, the first machine learning model 620 includes a k-nearest neighbor (k-NN) model 622 and a label predictor 624. In such implementations, the k-NN model 622 can be employed by the first machine learning model 620 to identify a number k of nearest neighbor user profiles that are considered to be most similar to the transformed user profile 619. In some examples, a model other than a k-NN model, such as those rooted in one or more prototype methods, may be employed as model 622. The label predictor 624 can then identify a true label for each of the k nearest neighbor user profiles from among the true labels included in encrypted label data set 626 and determine the at least one predicted label 629 based on the identified labels. In some implementations, the label predictor 624 can apply a softmax function to data that it receives and/or generates in determining the at least one predicted label 629.


For implementations in which the first machine learning model 620 and final result calculation logic 640 are configured to employ regression techniques, the at least one predicted label 629 may correspond to a single label that, for example, represents an integer number, such as a sum of the true labels for the k nearest neighbor user profiles as determined by the label predictor 624. Such a sum of the true labels for the k nearest neighbor user profiles as determined by the label predictor 624 is effectively equivalent to an average of the true labels for the k nearest neighbor user profiles as scaled by a factor of k. Similarly, for implementations in which the first machine learning model 620 and final result calculation logic 640 are configured to employ binary classification techniques, the at least one predicted label 629 may correspond to a single label that, for example, represents an integer number determined by the label predictor 624 based at least in part on such a sum. In the case of binary classification, each of the true labels for the k nearest neighbor user profiles may be a binary value of either zero or one, such that the aforementioned average may be an integer value between zero and one (e.g., 0.3, 0.8, etc.) that, for example, is effectively representative of a predicted probability that the true label for the user profile received as input by the first machine learning model 620 (e.g., transformed user profile 619) is equal to one. Additional detail pertaining to the nature of the at least one predicted label 629 and the ways in which the at least one predicted label 629 may be determined for implementations in which the first machine learning model 620 and final result calculation logic 640 are configured to employ regression techniques, as well as for implementations in which the first machine learning model 620 and final result calculation logic 640 are configured to employ binary classification techniques, are provided below with reference to FIGS. 9-11.


For implementations in which the first machine learning model 620 and final result calculation logic 640 are configured to employ multiclass classification techniques, the at least one predicted label 629 may correspond to a vector or set of predicted labels as determined by the label predictor 624. Each predicted label in such a vector or set of predicted labels may correspond to a respective category and be determined by the label predictor 624 based at least in part on a majority vote or a frequency at which true labels that correspond to the respective category in vectors or sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of a first value (e.g., one), as determined by the label predictor 624. Much like binary classification, in the case of multiclass classification, each true label in each vector or set of true labels for user profiles in the k nearest neighbor user profiles may be a binary value of either zero or one. Additional detail pertaining to the nature of the at least one predicted label 629 and the ways in which the at least one predicted label 629 may be determined for implementations in which the first machine learning model 620 and final result calculation logic 640 are configured to employ multiclass classification techniques are provided below with reference to FIGS. 9-11.


Final result calculation logic 640 can be employed to generate an inference result 649 (Resulti) based on the at least one predicted label 629. For example, final result calculation logic 640 can be employed to evaluate the at least one predicted label 629 against one or more thresholds and determine the inference result 649 based on the evaluation results. In some examples, the inference result 649 may be indicative of whether or not a user associated with the user profile 609 is to be added to one or more user groups. In some implementations, the at least one predicted label 629 can be included or otherwise indicated in the inference result 649.


In some implementations, the system 600, as depicted in FIG. 6, can represent a system as implemented by an MPC cluster, such as the MPC cluster 130 of FIG. 1. As such, it is to be understood that, in at least some of these implementations, some or all of the functionality described herein with reference to elements shown in FIG. 6 may be provided in a secure and distributed manner by way of two or more computing systems of an MPC cluster. For example, each of two or more computing systems of an MPC cluster may provide a respective share of the functionality described herein with reference to FIG. 6. In this example, the two or more computing systems may operate in parallel and exchange secret shares so as to collaboratively perform operations similar or equivalent to those described herein with reference to FIG. 6. In at least some of the aforementioned implementations, the user profile 609 may represent a share of a user profile. In such implementations, one or more of the other pieces of data or quantities described herein with reference to FIG. 6 may also be representative of secret shares thereof. It is to be understood that, in providing the functionality described herein with reference to FIG. 6, additional operations may be performed by the two or more computing systems for the purposes of protecting user privacy. Examples of one or more of the aforementioned implementations are described in further detail below, for example, with reference to FIG. 12, and elsewhere herein. In general, “shares” as described below and elsewhere herein may, in at least some implementations, correspond to secret shares.


While the training process for k-NN models, such as k-NN model 622, may be relatively fast and simple in that no knowledge of labels is required, the quality of such models can, in some situations, leave room for improvement. As such, in some implementations, one or more of the systems and techniques described in further detail below may be leveraged to boost the performance of the first machine learning model 620.



FIG. 7 is a conceptual diagram of an exemplary framework for generating an inference result for a user profile with boosted performance in a system 700. In some implementations, one or more of elements 609-629 as depicted in FIG. 7 may be similar or equivalent to one or more of elements 609-629 as described above with reference to FIG. 6, respectively. Much like the system 600, the system 700 includes random projection logic 610 and first machine learning model 620, and is depicted as performing one or more operations at inference time.


However, unlike system 600, the system 700 further includes a second machine learning model 730 that is trained and subsequently leveraged to boost the performance of the first machine learning model 620 by receiving transformed user profile 619 as input and generating, as output, a predicted residue value 739 (Residuei) that is indicative of a predicted amount of error in the at least one predicted label 629. For example, the accuracy of the second machine learning model can be higher than the accuracy of the first machine learning model. The predicted residue value 739, as obtained using the second machine learning model 730, can be in cleartext. Final result calculation logic 740, which is included in the system 700 in place of final result calculation logic 640, can be employed to generate inference result 749 (Resulti) based on the at least one predicted label 629 and based further on the predicted residue value 739. Given that the predicted residue value 739 is indicative of a predicted amount of error in the at least one predicted label 629, relying upon the at least one predicted label 629 and in tandem with the predicted residue value 739 may enable final result calculation logic 740 to effectively offset or counteract at least some of the error that may be expressed in the at least one predicted label 629, thereby enhancing one or both of the accuracy and reliability of the inference result 749 that is produced by the system 700.


For instance, final result calculation logic 740 can be employed to compute a sum of the at least one predicted label 629 and the predicted residue value 739. In some examples, final result calculation logic 740 can be further employed to evaluate such a computed sum against one or more thresholds and determine the inference result 749 based on the results of the evaluation. In some implementations, such a computed sum of the at least one predicted label 629 and the predicted residue value 739 can be included or otherwise indicated in the inference result 649 in FIG. 6 or 749 in FIG. 7.


The second machine learning model 730 may include or correspond to one or more of a deep neural network (DNN), a gradient-boosting decision tree, and a random forest model. That is, the first machine learning model 620 and the second machine learning model 730 may architecturally differ from one another. In some implementations, the second machine learning model 730 can be trained using one or more gradient boosting algorithms, one or more gradient descent algorithms, or a combination thereof.


Using boosting algorithms, which generally employ residues as described in more detail in this document, a weaker machine learning model, e.g., a k nearest neighbor model can be used to train a stronger machine learning model, e.g., a DNN. Unlike the training process for the weak learner, the training label for the strong learner is the residue of the weak learner. Using such residues enables the training of a more accurate strong learner.


The second machine learning model 730 can be trained using the same set of user profiles that were used to train the first machine learning model 620 and data indicating differences between the true labels for such a set of user profiles and predicted labels for such a set of user profiles as determined using the first machine learning model 620. As such, the process of training the second machine learning model 730 is performed after at least a portion of the process of training the first machine learning model 620 is performed. The data that is used for training the second machine learning model 730, such as data indicating differences between predicted labels determined using the first machine learning model 620 and true labels, may be generated or otherwise obtained through a process of evaluating the performance of the first machine learning model 620 as trained. An example of such a process is described in further detail below with reference to FIGS. 10-11.


As mentioned above, random projection logic 610, as included in systems 600 and 700, may be employed at least in part to obfuscate feature vectors, such as feature vectors included or indicated in the user profile 609 and other user profiles, with random noises to protect user privacy. To enable machine learning training and prediction, the random projection transformation that is applied by way of random projection logic 610 needs to preserve some notion of distance among feature vectors. One example of a random projection technique that can be employed in random projection logic 610 includes the SimHash technique. This technique and others described above can serve to obfuscate feature vectors while preserving the cosine distance between such feature vectors.


While the preservation of cosine distance between feature vectors may prove sufficient for training and using k-NN models, such as the k-NN model 622 of the first machine learning model 620, it may be less ideal for training and using other types of models, such as one or more models of the second machine learning model 730. As such, in some implementations, it may be desirable to employ a random projection technique in random projection logic 610 that can serve to obfuscate feature vectors while preserving the Euclidean distance between such feature vectors. One example of such a random projection technique includes the Johnson-Lindenstrauss (J-L) technique or transformation.


As mentioned above, one property of the J-L transformation is that it preserves the Euclidean distance between feature vectors with probability. In addition, the J-L transformation is lossy, non-reversible, and incorporates random noise. As such, even if two or more servers or computing systems of an MPC cluster were to collude, they would not be able to obtain an exact reconstruction of an original user profile (Pi) from a transformed version of the user profile (Pi′) obtained using the J-L transformation technique. In this way, employing the J-L transformation technique for purposes of transforming user profiles in one or more of the systems described herein may serve to provide user privacy protection. Furthermore, the J-L transformation technique can be used as a dimension reduction technique. As such, one advantageous byproduct of employing the J-L transformation technique for purposes of transforming user profiles in one or more of the systems described herein is that it may actually serve to significantly increase the speed at which subsequent processing steps can be performed by such systems.


In general, given an arbitrarily small ε>0, there exists a J-L transformation that can be applied to transform Pi to Pi′, Pj to Pj′ for arbitrary 1≤i, j≤n, where n is the number of training examples, and:





(1−ε)×|Pi−Pj|2≤Pi′−Pj′|2≤(1+ε)×|Pi−Pj|2


That is, applying the J-L transformation may change the Euclidean distance between the two arbitrarily selected training examples by no more than a small fraction ε. For at least the foregoing reasons, in some implementations, the J-L transformation technique may be employed in random projection logic 610 as described herein.


In some implementations, the system 700, as depicted in FIG. 7, can represent a system as implemented by an MPC cluster, such as the MPC cluster 130 of FIG. 1. As such, it is to be understood that, in at least some of these implementations, some or all of the functionality described herein with reference to elements shown in FIG. 7 may be provided in a secure and distributed manner by way of two or more computing systems of an MPC cluster. For example, each of two or more computing systems of an MPC cluster may provide a respective share of the functionality described herein with reference to FIG. 7. In this example, the two or more computing systems may operate in parallel and exchange secret shares so as to collaboratively perform operations similar or equivalent to those described herein with reference to FIG. 7. In at least some of the aforementioned implementations, the user profile 609 may represent a secret share of a user profile. In such implementations, one or more of the other pieces of data or quantities described herein with reference to FIG. 7 may also be representative of secret shares thereof. It is to be understood that, in providing the functionality described herein with reference to FIG. 7, additional operations may be performed by the two or more computing systems for the purposes of protecting user privacy. Examples of one or more of the aforementioned implementations are described in further detail below, for example, with reference to FIG. 12, and elsewhere herein.



FIG. 8 is a flow diagram that illustrates an example process 800 for generating an inference result for a user profile with boosted performance, e.g., higher accuracy, at an MPC cluster. One or more of the operations described with reference to FIG. 8 may, for example, be performed at inference time. Operations of the process 800 can be implemented, for example, by an MPC cluster, such as the MPC cluster 130 of FIG. 1, and can also correspond to one or more of the operations described above with reference to FIG. 7. One or more of the operations described with reference to FIG. 8 may, for example, be performed at inference time.


In some implementations, some or all of the functionality described herein with reference to elements shown in FIG. 8 may be provided in a secure and distributed manner by way of two or more computing systems of an MPC cluster, such as the MPC cluster 130 of FIG. 1. For example, each of two or more computing systems of an MPC cluster may provide a respective share of the functionality described herein with reference to FIG. 8. In this example, the two or more computing systems may operate in parallel and exchange secret shares so as to collaboratively perform operations similar or equivalent to those described herein with reference to FIG. 8. It is to be understood that, in providing the functionality described herein with reference to FIG. 8, additional operations may be performed by the two or more computing systems for the purposes of protecting user privacy. Examples of one or more of the aforementioned implementations are described in further detail below, for example, with reference to FIG. 12, and elsewhere herein. Operations of the process 800 can also be implemented as instructions stored on one or more computer readable media which may be non-transitory, and execution of the instructions by one or more data processing apparatus can cause the one or more data processing apparatus to perform the operations of the process 800.


The MPC cluster receives an inference request associated with a particular user profile (802). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster 130 receiving an inference request from the application 112, as described above with reference to FIG. 1.


The MPC cluster determines a predicted label for the particular user profile based on the particular user profile, a first machine learning model trained using multiple user profiles, and one or more of multiple true labels for the multiple user profiles (804). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the first machine learning model 620 being utilized to obtain at least one predicted label 629 (custom-character), as described above with reference to FIGS. 6-7.


In this example, the multiple true labels for the multiple user profiles may correspond to true labels that are included as part of encrypted label data 626, which are the true labels for the multiple user profiles that were used to train the first machine learning model 620. The one or more true labels, from among the multiple true labels, on which the determination of the predicted label for the particular user profile is based, for instance, may include at least one true label for each of k nearest neighbor user profiles identified by way of the k-NN model 622 of the first machine learning model 620. In some examples, each of the multiple true labels is encrypted, as is the case in the examples of FIGS. 6-7. Some of the various ways in which true labels for k nearest neighbor user profiles can be leveraged to determine predicted labels are described in detail above. As made evident in the above, the way or manner in which such true labels are leveraged to determine predicted labels may at least in part depend on the type(s) of inference technique(s) that are employed (e.g., regression techniques, binary classification techniques, multiclass classification techniques, etc.).


The MPC cluster determines a predicted residue value indicating a predicted error in the predicted label based on the particular user profile and a second machine learning model trained using the multiple user profiles and data indicating differences between the multiple true labels for the multiple user profiles and multiple predicted labels as determined for the multiple user profiles using the first machine learning model (806). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the second machine learning model 730 being utilized to obtain predicted residue value 739 (Residuei), as described above with reference to FIG. 7. As such, in some implementations, the second machine learning model includes at least one of a deep neural network, a gradient-boosting decision tree, and a random forest model.


The MPC cluster generates data representing an inference result based on the predicted label and the predicted residue value (808). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with final result calculation logic 740 being employed to generate an inference result 749 (Resulti), as described above with reference to FIG. 7. As such, in some examples, the inference result includes or corresponds to a sum of the predicted label and the predicted residue value.


The MPC cluster provides the data representing the inference result to a client device (810). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster 130 providing an inference result to the client device 110 on which the application 112 runs, as described above with reference to FIGS. 1-2.


In some implementations, the process 800 further includes one or more operations in which the MPC cluster applies a transformation to the particular user profile to obtain a transformed version of the particular user profile. In these implementations, to determine the predicted label, the MPC cluster determines the predicted label based at least in part on the transformed version of the particular user profile. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with random projection logic 610 being employed to apply a random projection transformation to the user profile 609 (Pi) to obtain the transformed user profile 619 (Pi′), as described above with reference to FIGS. 6-7. As such, in some examples, the aforementioned transformation may be a random projection. Furthermore, in at least some of these examples, the aforementioned random projection may be a Johnson-Lindenstrauss (J-L) transformation. In at least some of the aforementioned implementations, to determine the predicted label, the MPC cluster provides the transformed version of the particular user profile as input to the first machine learning model to obtain the predicted label for the particular user profile as output. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the first machine learning model 620 receiving transformed user profile 619 (Pi′) as input and generating at least one predicted label 629 (custom-character) in response thereto, as described above with reference to FIGS. 6-7.


As mentioned above, in some implementations, the first machine learning model includes a k-nearest neighbor model. In at least some of these implementations, to determine the predicted label, the MPC cluster identifies a number k of nearest neighbor user profiles that are considered most similar to the particular user profile among the multiple user profiles based at least in part on the particular user profile and the k-nearest neighbor model, and determines the predicted label based at least in part on a true label for each of the k nearest neighbor user profiles. In some such implementations, to determine the predicted label based at least in part on a true label for each of the k nearest neighbor user profiles, the MPC cluster determines a sum of the true labels for the k nearest neighbor user profiles. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the first machine learning model 620 being utilized to obtain at least one predicted label 629 (custom-character) in one or more implementations where one or more regression and/or binary classification techniques are employed, as described above with reference to FIGS. 6-7. In some examples, the predicted label includes or corresponds to the sum of the true labels for the k nearest neighbor user profiles.


In some of the aforementioned implementations, to determine the predicted label based at least in part on the true label for each of the k nearest neighbor user profiles, the MPC cluster determines a set of predicted labels based at least in part on a set of true labels for each of the k nearest neighbor user profiles corresponding to a set of categories, respectively, and, to determine the set of predicted labels, the MPC cluster performs operations for each category in the set. Such operations can include one or more operations in which the MPC cluster determines a majority vote or a frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of a first value. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the first machine learning model 620 being utilized to obtain at least one predicted label 629 (custom-character) in one or more implementations where one or more multiclass classification techniques are employed, as described above with reference to FIGS. 6-7.



FIG. 9 is a flow diagram that illustrates an example process 900 for preparing for and carrying out a training of a second machine learning model for boosting inference performance at an MPC cluster. Operations of the process 900 can be implemented, for example, by an MPC cluster, such as the MPC cluster 130 of FIG. 1, and can also correspond to one or more of the operations described above with reference to FIGS. 2, 4, 6, and 7. In some implementations, some or all of the functionality described herein with reference to elements shown in FIG. 9 may be provided in a secure and distributed manner by way of two or more computing systems of an MPC cluster, such as the MPC cluster 130 of FIG. 1. For example, each of two or more computing systems of an MPC cluster may provide a respective secret share of the functionality described herein with reference to FIG. 9. In this example, the two or more computing systems may operate in parallel and exchange secret shares so as to collaboratively perform operations similar or equivalent to those described herein with reference to FIG. 9. It is to be understood that, in providing the functionality described herein with reference to FIG. 9, additional operations may be performed by the two or more computing systems for the purposes of protecting user privacy. Examples of one or more of the aforementioned implementations are described in further detail below, for example, with reference to FIG. 12, and elsewhere herein. Operations of the process 900 can also be implemented as instructions stored on one or more computer readable media which may be non-transitory, and execution of the instructions by one or more data processing apparatus can cause the one or more data processing apparatus to perform the operations of the process 900.


The MPC cluster trains a first machine learning model using multiple user profiles (910). For example, the first machine learning model may correspond to the first machine learning model 620, as described above. Similarly, the multiple user profiles that are used in the training of the first machine learning model may correspond to a number n user profiles that are used to train the first machine learning model 620, the true labels for which may be included in the encrypted label data set 626, as described above.


The MPC cluster evaluates a performance of the first machine learning model as trained using the multiple user profiles (920). Additional details pertaining to what such an evaluation may entail are provided below with reference to FIGS. 10-11.


In some implementations, data generated in such an evaluation can be utilized by the MPC cluster or another system in communication with the MPC cluster to determine whether or not the performance of the first machine learning model, such as the first machine learning model 620, warrants boosting, for example, by way of a second machine learning model, such as the second machine learning model 730. Examples of data generated in such an evaluation that can be utilized is this way are described in further detail below with reference to the profile and residue data set 1070 of FIG. 10 and step 1112 of FIG. 11.


For instance, in some situations, the MPC cluster or another system in communication with the MPC cluster may determine, based on data generated in such an evaluation, that performance (e.g., prediction accuracy) of the first machine learning model satisfies one or more thresholds, and thus does not warrant boosting. In such situations, the MPC cluster may refrain from training and implementing a second machine learning model based on this determination. However, in other situations, the MPC cluster or another system in communication with the MPC cluster may determine, based on data generated in such an evaluation, that performance (e.g., prediction accuracy) of the first machine learning model satisfies one or more thresholds, and thus does warrant boosting. In these situations, the MPC cluster may receive an upgrade in functionality comparable to that which would be gained in transitioning from the system 600 to the system 700, as described above with reference to FIGS. 6-7, based on this determination. To receive such an upgrade in functionality, the MPC cluster may proceed with training and implementing a second machine learning model, such as the second machine learning model 730, for boosting the performance, e.g., accuracy, of the first machine learning model using residue values. In some examples, data generated in such an evaluation may additionally or alternatively be provided to one or more entities associated with the MPC cluster. In some such examples, the one or more entities may make their own determinations regarding whether or not the performance of the first machine learning model warrants boosting, and proceed accordingly. Other configurations are possible.


The MPC cluster uses a set of data including data generated in the evaluation of the performance of the first machine learning model to train a second machine learning model (930). Examples of such data can include that which is described below with reference to the profile and residue data set 1070 of FIG. 10 and step 1112 of FIG. 11.


In some implementations, the process 900 further includes additional steps 912-916, which are described in further detail below. In such implementations, steps 912-916 are performed prior to steps 920 and 930, but can be performed after step 910.



FIG. 10 is a conceptual diagram of an exemplary framework for evaluating a performance of a first machine learning model in a system 1000. In some implementations, one or more of elements 609-629 as depicted in FIG. 10 may be similar or equivalent to one or more of elements 609-629 as described above with reference to FIG. 6-7, respectively. In some examples, one or more of the operations described herein with reference to FIG. 10 may correspond to one or more of those described above with reference to step 920 of FIG. 9. Much like the systems 600 and 700, the system 1000 includes random projection logic 610 and first machine learning model 620.


However, unlike systems 600 and 700, the system 1000 further includes residue calculation logic 1060. Also, in the example of FIG. 10, the user profile 609 (Pi) corresponds to one of the multiple user profiles that were used to train the first machine learning model 620, whereas, in the examples of FIGS. 6 and 7, the user profile 609 (Pi) might not necessarily correspond to one of the multiple user profiles that were used to train the first machine learning model 620, but instead simply correspond to a user profile that is associated with an inference request received at inference time. The aforementioned multiple user profiles that were used to train the first machine learning model 620 can, in some examples, correspond to the multiple user profiles described above with reference to step 910 of FIG. 9. Residue calculation logic 1060 can be employed to generate a residue value 1069 (Residuei) that is indicative of an amount of error in the at least one predicted label 629 based on the at least one predicted label 629 and at least one true label 1059 (Li). Both the at least one predicted label 629 (custom-character) and the at least one true label 1059 (L1) can be encrypted. For example, residue calculation logic 1060 can employ secret shares to calculate a difference in value between the at least one predicted label 629 and the at least one true label 1059. In some implementations, the residue value 1069 may correspond to the aforementioned difference in value.


The residue value 1069 can be stored in association with the transformed user profile 619, for example, in memory as part of the profile and residue data set 1070. In some examples, data included in the profile and residue data set 1070 may correspond to one or both of data as described above with reference to step 930 of FIG. 9 and data as described below with reference to step 1112 of FIG. 11. In some implementations, the residue values 1069 are in the form of secret shares to protect user privacy and data security.


In some implementations, the system 1000, as depicted in FIG. 10, can represent a system as implemented by an MPC cluster, such as the MPC cluster 130 of FIG. 1. As such, it is to be understood that, in at least some of these implementations, some or all of the functionality described herein with reference to elements shown in FIG. 10 may be provided in a secure and distributed manner by way of two or more computing systems of an MPC cluster. For example, each of two or more computing systems of an MPC cluster may provide a respective share of the functionality described herein with reference to FIG. 10. In this example, the two or more computing systems may operate in parallel and exchange secret shares so as to collaboratively perform operations similar or equivalent to those described herein with reference to FIG. 10. In at least some of the aforementioned implementations, the user profile 609 may represent a secret share of a user profile. In such implementations, one or more of the other pieces of data or quantities described herein with reference to FIG. 10 may also be representative of secret shares thereof. It is to be understood that, in providing the functionality described herein with reference to FIG. 10, additional operations may be performed by the two or more computing systems for the purposes of protecting user privacy. Examples of one or more of the aforementioned implementations are described in further detail below, for example, with reference to FIG. 12, and elsewhere herein.



FIG. 11 is a flow diagram that illustrates an example process 1100 for evaluating a performance of a first machine learning model at an MPC cluster. Operations of the process 1100 can be implemented, for example, by an MPC cluster, such as the MPC cluster 130 of FIG. 1, and can also correspond to one or more of the operations described above with reference to FIGS. 9-10. In some examples, one or more of the operations described herein with reference to FIG. 11 may correspond to one or more of those described above with reference to step 920 of FIG. 9. In some implementations, some or all of the functionality described herein with reference to elements shown in FIG. 11 may be provided in a secure and distributed manner by way of two or more computing systems of an MPC cluster, such as the MPC cluster 130 of FIG. 1. For example, each of two or more computing systems of an MPC cluster may provide a respective share of the functionality described herein with reference to FIG. 11. In this example, the two or more computing systems may operate in parallel and exchange secret shares so as to collaboratively perform operations similar or equivalent to those described herein with reference to FIG. 11. It is to be understood that, in providing the functionality described herein with reference to FIG. 11, additional operations may be performed by the two or more computing systems for the purposes of protecting user privacy. Examples of one or more of the aforementioned implementations are described in further detail below, for example, with reference to FIG. 12, and elsewhere herein. Operations of the process 1100 can also be implemented as instructions stored on one or more computer readable media which may be non-transitory, and execution of the instructions by one or more data processing apparatus can cause the one or more data processing apparatus to perform the operations of the process 1100.


The MPC cluster selects the ith user profile and at least one corresponding true label ([Pi, Li]), where i is initially set to a value of one (1102-1104) and incremented through recursion until i equals n (1114-1116), where n is the total number of user profiles that were used to train the first machine learning model. In other words, the process 1100 includes performing steps 1106-1112, as described below, for each of the n user profiles that were used to train the first machine learning model.


In some implementations, the ith user profile may represent a secret share of a user profile. In such implementations, one or more of the other pieces of data or quantities described herein with reference to FIG. 11 may also be representative of shares thereof.


The MPC cluster applies a random projection to the ith user profile (Pi) to obtain a transformed version of the ith user profile (Pi′) (1106). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with random projection logic 610 being employed to apply a random projection transformation to the user profile 609 (Pi) to obtain the transformed user profile 619 (Pi′), as described above with reference to FIG. 10.


The MPC cluster provides the transformed version of the ith user profile (Pi′) as input to a first machine learning model to obtain at least one predicted label (custom-character) for the transformed version of ith user profile (Pi′) as output (1108). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the first machine learning model 620 receiving transformed user profile 619 (Pi′) as input and generating at least one predicted label 629 (custom-character) in response thereto, as described above with reference to FIG. 10.


The MPC cluster calculates a residue value (Residuei) based at least in part on at least one true label (Li) for the ith user profile (Pi) and the at least one predicted label (custom-character) (1110). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with residue calculation logic 1060 being employed to calculate residue value 1069 (Residuei) based at least in part on at least one true label 1059 (Li) and the at least one predicted label 629 (custom-character), as described above with reference to FIG. 10.


The MPC cluster stores the calculated residue value (Residuei) in association with the transformed version of ith user profile (Pi′) (1112). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with residue value 1069 (Residuei) being stored in association with the transformed user profile 619 (Pi′), for example, in memory as part of the profile and residue data set 1070, as described above with reference to FIG. 10. In some examples, this data may correspond to data as described above with reference to step 930 of FIG. 9. As such, in these examples, some or all of the data that is stored in this step may be leveraged as data for training a second machine learning model, such as the second machine learning model 730.


Referring once again to steps 1108-1110, for at least some implementations in which the first machine learning model is configured to employ regression techniques, the at least one predicted label (custom-character) that the MPC cluster obtains at step 1108 can correspond to a single predicted label representing an integer number. In these implementations, the residue value (Residuei) that the MPC cluster calculates at step 1110 can correspond to an integer number indicative of a difference in value between the at least one true label (Li) and the at least one predicted label (custom-character). In at least some of the aforementioned implementations, at step 1108, the first machine learning model identifies a number k of nearest neighbor user profiles that are considered to be most similar to the transformed version of ith user profile (Pi′), identifies at least one true label for each of the k nearest neighbor user profiles, calculates a sum of the true labels for the k nearest neighbor user profiles, and uses this sum as the at least one predicted label (custom-character). As mentioned above, such a sum of the true labels for the k nearest neighbor user profiles as determined in this step is effectively equivalent to an average of the true labels for the k nearest neighbor user profiles as scaled by a factor of k. In some examples, this sum may be utilized as the at least one predicted label (custom-character) instead of the average of the true labels for the k nearest neighbor user profiles such that a division operation need not be performed. Given that the at least one predicted label (custom-character) is effectively equivalent to an average of the true labels for the k nearest neighbor user profiles as scaled by a factor of k, for at least some implementations in which the first machine learning model is configured to employ regression techniques, the calculation that is performed by the MPC cluster at step 1110 is given by:





Residuei=kLicustom-character


Similarly, for at least some implementations in which the first machine learning model is configured to employ binary classification techniques, the at least one predicted label (custom-character) that the MPC cluster obtains at step 1108 can correspond to a single predicted label that, for example, represents an integer number determined based at least in part on a sum of the true labels for the k nearest neighbor user profiles. As mentioned above with reference to implementations in which the first machine learning model is configured to employ regression techniques, such a sum of the true labels for the k nearest neighbor user profiles is effectively equivalent to an average of the true labels for the k nearest neighbor user profiles as scaled by a factor of k.


However, unlike implementations where the first machine learning model is configured to employ regression techniques, in implementations where the first machine learning model is configured to employ binary classification techniques, each of the true labels for the k nearest neighbor user profiles may be a binary value of either zero or one, such that the aforementioned average may be an integer value between zero and one (e.g., 0.3, 0.8, etc.). Although the MPC cluster could, in implementations where binary classification techniques are employed, calculate and use the sum of the true labels for the k nearest neighbor user profiles (sum_of_labels) as the at least one predicted label (custom-character) at step 1108, and use the formula described above with reference to implementations where regression techniques are employed (Residuei=kLicustom-character) to obtain a residue value (Residuei) at step 1110 that is mathematically viable, such a residue value (Residuei) may potentially pose privacy concerns, for instance, later on when used to determine whether boosting the first machine learning model is warranted or later on when used to train a second machine learning model such as the second machine learning model 730. More specifically, because each of the true labels for the k nearest neighbor user profiles may be a binary value of either zero or one, in implementations where binary classification techniques are employed, the sign of such a residue value (Residuei) may potentially be indicative of the value of the at least one true label (Li), and thus may potentially be inferred by one or more systems and/or entities that may handle data indicating residue value (Residuei) in some capacity at or subsequent to step 1112.


For instance, consider a first example where binary classification techniques are to be employed and Li=1, k=15, and custom-character=12. In this first example, the at least one predicted label (custom-character) corresponds to the sum of the true labels for the k nearest neighbor user profiles (sum_of_labels), which is effectively equivalent to an average of the true labels for the k nearest neighbor user profiles as scaled by a factor of k, where the aforementioned average is a non-integer value of 0.8. If the same formula as described above (Residuei=kLicustom-character) was to be utilized in this first example to calculate the residue value (Residue1), e.g., at step 1110, then the residue value (Residue1) in this first example would be given by: Residuei=(15)(1)−12=3. Thus, in this first example, the residue value (Residuei) would be equal to a value of (positive) 3. Now, consider a second example where binary classification techniques are to be employed and Li=0, but k and custom-character are once again equal to values of 15 and 12, respectively. If, once again, the same formula as described above (Residuei=kLicustom-character) was to be utilized in this second example to calculate the residue value (Residuei), e.g., at step 1110, then the residue value (Residuei) in this second example would be given by: Residue1=(15)(0)−12=−12. Thus, in this first example, the residue value (Residuei) would be equal to a value of −12. Indeed, in the case of the first and second examples described above, a positive residue value (Residuei) may correlate to Li=1, while a negative residue value (Residuei) may correlate to Li=0.


To understand why inferring Li from Residuei is possible, consider an example in which the residues for user profiles used to train the first machine learning model whose true labels are equal to 0 are assumed to satisfy a Normal Distribution with the notation custom-character0, σ0), where μ0 and σ0 are the mean and standard deviation of the Normal Distribution of prediction errors (e.g., residue values) for true labels that are equal to 0 (zero) and are associated with user profiles that were used to train the first machine learning model, respectively, and the residues for training examples whose labels are equal to 1 are assumed to satisfy N(μ1, σ1), where μ1 and σ1 are the mean and standard deviation of the Normal Distribution of prediction errors for true labels that are equal to 1 (one) and are associated with user profiles that were used to train the first machine learning model, respectively. Under such assumptions, it is clear that μ0<0, μ1>0, and there is no guarantee that σ01.


In light of the foregoing, as described below, in some implementations, a different approach can be taken to performing one or more operations associated with steps 1108-1110 for implementations where binary classification techniques are employed. In some implementations, to force the residues for the two classes of training examples to have the same normal distribution, the MPC cluster can apply a transformation ƒ to the sum of the true labels for the k nearest neighbor user profiles (sum_of_labels) so that residue values calculated based on Li and custom-character cannot be used to predict Li. The transformation ƒ, when applied to an initial predicted label (e.g., sum of the true labels in the case of binary classification, majority vote of true labels in the case of multiclass classification, etc.), can serve to remove bias that might exist in the first machine learning model's prediction. To achieve such as goal, the transformation ƒ needs to satisfy the following properties:





ƒ(μ0)=0  (i)





ƒ(μ1)=1  (ii)





σ0׃′(μ0)=σ1׃′(μ1)  (iii)


where ƒ′ is the derivative of ƒ.


One example of a transformation with the above properties that may be employed in such implementations is the second degree polynomial transformation of the shape ƒ(x)=a2x2+a1x+a0, where ƒ′(x)=2a2x+a1. In some examples, the MPC cluster can deterministically find the values of coefficients {a2, a1, a0} based on three linear equations from the three constraints as follows:








Let


D

=

1



(


μ
1

-

μ
0


)

2



(


σ
0

+

σ
1


)




,










a
2


=


σ
0

-

σ
1






(
i
)













a
1


=

2


(



σ
1



μ
1


-


σ
0



μ
0



)






(
ii
)













a
0


=


μ
0

(



μ
0



σ
0


+


μ
0



σ
1


-

2


μ
1



σ
1



)





(
iii
)







In these examples, the MPC cluster can calculate coefficients {a2, a1, a0} as: {a2, a1, a0}=D×{a2′, a1′, a0′}. The MPC cluster can calculate {a2′, a1′, a0′} and D using addition and multiplication operations, e.g., over secret shares. The transformation ƒ(x)=a2x2+a1x+a0 is also mirror-symmetrical around:






x
=


-


a
1


2


a
2




=



-
2


(



σ
1



μ
1


-


σ
0



μ
0



)



2


(


σ
0

-

σ
1


)



=





σ
1



μ
1


-


σ
0



μ
0





σ
1

-

σ
0



.







In order to calculate the aforementioned coefficients and other values dependent thereon, the MPC cluster may first estimate the mean and standard deviation of the probability distribution of prediction errors (e.g., residue values) for true labels that are equal to zero, μ0 and σ0, respectively, as well was the mean and standard deviation of the probability distribution of prediction errors (e.g., residue values) for true labels that are equal to one, μ1 and σ1, respectively. In some examples, the variance σ02 of the probability distribution of prediction errors for true labels that are equal to zero may be determined in addition to or instead of the standard deviation σ0, and the variance σ12 of the probability distribution of prediction errors for true labels that are equal to one may be determined in addition to or instead of the standard deviation σ1.


In some instances, a given probability distribution of prediction errors may correspond to a normal distribution and, in other instances, a given probability distribution of prediction errors may correspond to a probability distribution other than a normal distribution, such as a Bernoulli distribution, uniform distribution, binomial distribution, hypergeometric distribution, geometric distribution, exponential distribution, and the like. In such other instances, the distribution parameters that are estimated may, in some examples, include parameters other than mean, standard deviation, and variance, such as one or more parameters that are specific to characteristics of the given probability distribution of prediction errors. For example, the distribution parameters that are estimated for a given probability distribution of prediction errors that corresponds to a uniform distribution may include minimum and maximum value parameters (a and b), while the distribution parameters that are estimated for a given probability distribution of prediction errors that corresponds to an exponential distribution may include at least one rate parameter (λ). In some implementations, one or more operations that are similar to one or more operations that are performed in connection with process 1110 of FIG. 11 may be performed such that data indicative of prediction errors of the first machine learning model can be obtained and utilized for estimating such distribution parameters. In at least some of the aforementioned implementations, data indicative of prediction errors of the first machine learning model can be obtained and utilized to (i) identify, from among several different types of probability distributions (e.g., normal distribution, Bernoulli distribution, uniform distribution, binomial distribution, hypergeometric distribution, geometric distribution, exponential distribution, etc.), a particular type of probability distribution that most closely corresponds to the shape of the probability distribution of a given subset of the prediction errors indicated by the data, and (ii) estimate one or more parameters of the probability distribution of the given subset of the prediction errors indicated by the data in accordance with the particular type of probability distribution identified. Other configurations are possible.


Referring once again to examples in which the estimated distribution parameters include mean and standard deviation, in these example, to estimate such distribution parameters for true labels that are equal to zero, the MPC cluster can calculate:







μ
0

=


sum
0


count
0









σ
0
2

=



sum_of


_square
0



count
0


-

μ
0
2








where
:







sum
0

=



i



(




j


k
-

NNtoP
i





L
j


)

×

(

1
-

L
i


)










count
0

=



i


(

1
-

L
i


)









sum_of


_square
0


=



i




(




j


k
-

NNtoP
i





L
j


)

2

×

(

1
-

L
i


)







In some examples, the MPC cluster calculates standard deviation σ0 based on variance σ02, e.g., by computing the square root of variance σ02. Similarly, to estimate such distribution parameters for true labels that are equal to one, the MPC cluster can calculate:







μ
1

=


sum
1


count
1









σ
1
2

=



sum_of


_square
1



count
1


-

μ
1
2








where
:







sum
1

=



i



(




j


k
-

NNtoP
i





L
j


)

×

L
i










count
1

=



L
i









sum_of


_square
1


=



i




(




j


k
-

NNtoP
i





L
j


)

2

×

L
i







In some examples, the MPC cluster calculates standard deviation σ1 based on variance σ12, e.g., by computing the square root of variance σ12.


Once such distribution parameters are estimated, the coefficients {a2, a1, a0} can be calculated, stored, and later utilized to apply the corresponding transformation ƒ to the sum of the true labels for the k nearest neighbor user profiles (sum_of_labels). In some examples, these coefficients are utilized to configure the first machine learning model, such that, going forward, the first machine learning model applies the corresponding transformation ƒ to the sum of the true labels for the k nearest neighbor user profiles responsive to input.


Much like binary classification, in the case of multiclass classification, each true label in each vector or set of true labels for user profiles in the k nearest neighbor user profiles may be a binary value of either zero or one. For this reason, an approach similar to that described above with reference to binary classification may also be taken in implementations where multiclass classification techniques, such that residue values calculated based on Li and custom-character cannot be used to predict Li. However, in the case of multiclass classification, a respective function or transformation ƒ may be defined and utilized for each category. For instance, if each vector or set of true labels for each user profile were to contain w different true labels corresponding to w different categories, respectively, w different transformations ƒ may be determined and utilized. Also, instead of calculating sums of the true labels, in the case of multiclass classification, a frequency value is calculated for each category. Additional details on how such a frequency value may be calculated are provided above, as well as immediately below. Other configurations are possible.


For an arbitrary chosen jth label, the MPC cluster can partition training examples into two groups, based on whether lj is the training label for the training examples. For the group of training examples where lj is the training label, the MPC cluster can assume that the frequencyj is in a normal distribution and calculate the mean μ1 and variance σ1. On the other hand, for the group of training examples where lj is not the training label, the MPC cluster can assume that the frequencyj is in a normal distribution and calculate the mean μ0 and variance σ0.


Similar to binary classification, in the case of multiclass classification, the k-NN models' prediction is most likely biased, (e.g., μ0>0, where it should have been 0, and μ1<k, where it should have been k). In addition, there is no guarantee that σ0==σ1. As such, similar to binary classification, in the case of multiclass classification, the MPC cluster applies a transformation ƒ over the predicted frequencyj so that, after the transformation, the Residuej for the two groups have substantially the same normal distribution. To achieve such as goal, the transformation ƒ needs to satisfy the following properties:





ƒ(μ0)=0  (i)





ƒ(μ1)=1  (ii)





σ0׃′(μ0)=σ1׃′(μ1)  (iii)


where ƒ′ is the derivative of ƒ.


The above three properties are very similar to their counterparts in the binary classification case. In the case of multiclass classification, one example of a transformation with the above properties that may be employed is the second degree polynomial transformation of the shape ƒ(x)=a2x2+a1x+a0, where ƒ′(x)=2a2x+a1. In some examples, the MPC cluster can deterministically calculate the values of coefficients {a2, a1, a0} based on three linear equations from the three constraints as follows:








Let


D

=

k



(


μ
1

-

μ
0


)

2



(


σ
0

+

σ
1


)




,










a
2


=


σ
0

-

σ
1






(
i
)













a
1


=

2


(



σ
1



μ
1


-


σ
0



μ
0



)






(
ii
)













a
0


=


μ
0

(



μ
0



σ
0


+


μ
0



σ
1


-

2


μ
1



σ
1



)





(
iii
)







Note that the transformation for binary classification and multiclass classification are nearly the same, the only difference being that, in multiclass classification with k-NN models, the value of D can, in some implementations, be enlarged by a factor of k.


Referring once again to FIG. 9, in some implementations, one or more of steps 912-916 may correspond to one or more of the operations described above with approaches for defining at least one function or transformation that can be employed by the MPC cluster such that residue values calculated based on Li and custom-character cannot be used to predict Li. In particular, steps 912-916 may be performed for implementations in which one or more binary and/or multiclass classification techniques are to be employed. As mentioned above, steps 912-916 are performed prior to steps 920 and 930, and may be performed after step 910.


The MPC cluster estimates a set of distribution parameters based on multiple true labels for multiple user profiles (912). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster calculating one or more of parameters μ0, σ02, σ0, μ1, σ12, and σ1, as described above, based on the true labels associated with the same user profiles utilized in step 910.


The MPC cluster derives a function based on the estimated set of distribution parameters (914). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster calculating parameters or coefficients, such as {a2, a1, a0}, which effectively define a function. As such, in some implementations, to derive a function at step 914, the MPC cluster derives a set of parameters of a function, e.g., {a2, a1, a0}.


The MPC cluster configures the first machine learning model to, given a user profile as input, generate an initial predicted label and apply the derived function to the initial predicted label to generate, as output, a predicted label for the user profile (916). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster configuring the first machine learning model, such that, going forward, the first machine learning model applies the corresponding transformation ƒ to the sum of the true labels for the k nearest neighbor user profiles responsive to input (in the case of binary classification). In the case of multiclass classification, the transformation ƒ may represent one of w different functions that the MPC cluster configures the first machine learning model to apply to a respective one of w different values in vector or set corresponding to w different categories. As described above, each of one of these w different values may correspond to a frequency value.


With steps 912-916 having been performed, and the first machine learning model having been configured in such a manner, data that is generated in step 920 and subsequently utilized, e.g., in step 930, may not be used to predict true labels (Li).


Referring once again to FIG. 8, in some implementations, the process 800 may include one or more steps that correspond to one or more of the operations described above with reference to FIGS. 9-11.


In some implementations, the process 800 further includes one or more operations in which the MPC cluster evaluates a performance of the first machine learning model. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster performing step 920 as described above with reference to FIG. 9. In these implementations, to evaluate the performance of the first machine learning model, for each of the multiple user profiles, the MPC cluster determines a predicted label for the user profile based at least in part on (i) the user profile, (ii) the first machine learning model, and (iii) one or more of the multiple true labels for the multiple user profiles, and determines a residue value for the user profile indicating a prediction error in the predicted label based at least in part on the predicted label determined for the user profile and a true label for the user profile included in the multiple true labels. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster performing steps 1108-1106 as described above with reference to FIG. 11. In addition, in these implementations, the process 800 further includes one or more operations in which the MPC cluster trains the second machine learning model using data indicating the residue values determined for the multiple user profiles in evaluating the performance of the first machine learning model. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster performing step 930 as described above with reference to FIG. 9.


In at least some of the aforementioned implementations, the residue value for the user profile is indicative of a difference in value between the predicted label determined for the user profile and the true label for the user profile. For instance, this may be the case for examples in which regression techniques are employed.


In at least some of the aforementioned implementations, before the MPC cluster evaluates the performance of the first machine learning model, the process 800 further includes one or more operations in which the MPC cluster derives a function based at least in part on the multiple true labels, and configures the first machine learning model to, given a user profile as input, use the function to generate, as output, a predicted label for the user profile. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster performing steps 914-916 as described above with reference to FIG. 9. As such, in some implementations, to derive a function at this step, the MPC cluster derives a set of parameters of a function, e.g., {a2, a1, a0}.


In at least some of the aforementioned implementations, the process 800 further includes one or more operations in which the MPC cluster estimates a set of distribution parameters based at least in part on the multiple true labels. In such implementations, to derive the function based at least in part on the multiple true labels, the MPC cluster derives the function based at least in part on the estimated set of distribution parameters. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster performing steps 912-914 as described above with reference to FIG. 9. As such, the aforementioned set of distribution parameters can include one or more parameters of a probability distribution of prediction errors for true labels of a first value in the multiple true labels, e.g., a mean (μ0) and a variance (σ0) of a normal distribution of prediction errors for true labels of a first value in the multiple true labels, and one or more parameters of a probability distribution of prediction errors for true labels of a second value in the multiple true labels, e.g., a mean (μ1) and a variance (σ1) of a normal distribution of prediction errors for true labels of a second, different value in the multiple true labels. As described above, in some examples, the aforementioned set of distribution parameters can include other types of parameters. Furthermore, in at least some of the aforementioned implementations, the function is a second degree polynomial function, e.g., ƒ(x)=a2x2+a1x+a0, where ƒ′(x)=2a2x+a1.


In at least some of the aforementioned implementations, to configure the first machine learning model to, given a user profile as input, use the function to generate, as output, a predicted label for the user profile, the MPC cluster configures the first machine learning model to, given a user profile as input: (i) generate an initial predicted label for the user profile, and (ii) apply the function to the initial predicted label for the user profile to generate, as output, a predicted label for the user profile. For example, for examples in which binary classification techniques are employed, this may correspond to one or more operations in which the MPC cluster configures the first machine learning model to, given a user profile as input: (i) calculate a sum of the true labels for the k nearest neighbor user profiles (sum_of_labels), and (ii) apply the function (transformation ƒ) to the initial predicted label for the user profile to generate, as output, a predicted label for the user profile (custom-characterƒ(sum_of_labels)). Similar operations may be performed for cases in which multiclass classification techniques are employed. In some implementations, to apply function to the initial predicted label for the user profile, the MPC cluster applies a function, as defined based on the derived set of parameters, e.g., {a2, a1, a0}. In some examples, to determine the predicted label based at least in part on the true label for each of the k nearest neighbor user profiles, the MPC cluster determines a sum of the true labels for the k nearest neighbor user profiles. For instance, this may be the case for implementations in which regression or binary classification techniques are employed. In some of the aforementioned examples, the predicted label for the particular user profile may correspond to the sum of the true labels for the k nearest neighbor user profiles. For instance, this may be the case for implementations in which regression classification techniques are employed (custom-character=sum_of_labels). In other such examples, to determine the predicted label based at least in part on the true label for each of the k nearest neighbor user profiles, the MPC cluster applies a function to the sum of the true labels for the k nearest neighbor user profiles to generate a predicted label for the particular user profile. For instance, this may be the case for implementations in which binary classification techniques are employed (custom-character=ƒ(sum_of_labels)).


As mentioned above, in some of the aforementioned implementations, to determine the predicted label based at least in part on the true label for each of the k nearest neighbor user profiles, the MPC cluster determines a set of predicted labels based at least in part on a set of true labels for each of the k nearest neighbor user profiles corresponding to a set of categories, respectively, and, to determine the set of predicted labels, the MPC cluster performs operations for each category in the set. Such operations can include one or more operations in which the MPC cluster determines a frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of a first value. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the first machine learning model 620 being utilized to obtain at least one predicted label 629 (custom-character) in one or more implementations where one or more multiclass classification techniques are employed, as described above with reference to FIGS. 6-7. In at least some of the aforementioned implementations, to determine the set of predicted labels, for each category in the set, the MPC cluster applies a function corresponding to the category to the determined frequency to generate a predicted label corresponding to the category for the particular user profile. For instance, the respective function may correspond to one of w different functions derived by the MPC cluster for w different categories as described above with reference to step 914 of FIG. 9.



FIG. 12 is a flow diagram that illustrates an example process 1200 for generating an inference result for a user profile with boosted performance at a computing system of an MPC cluster. One or more of the operations described with reference to FIG. 12 may, for example, be performed at inference time. At least some of the operations of the process 1200 can be implemented, for example, by a first computing system of an MPC cluster, such as MPC1 of the MPC cluster 130 of FIG. 1, and can also correspond to one or more of the operations described above with reference to FIG. 8. However, in process 1200, one or more operations can be performed over secret shares, so as to provide user data privacy protection. In general, “shares” as described below and elsewhere herein may, in at least some implementations, correspond to secret shares. Other configurations are possible. One or more of the operations described with reference to FIG. 12 may, for example, be performed at inference time.


The first computing system of the MPC cluster receives an inference request associated with a given user profile (1202). For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with MPC1 of the MPC cluster 130 receiving an inference request from the application 112, as described above with reference to FIG. 1. In some implementations, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with step 802 as described above with reference to FIG. 8.


The first computing system of the MPC cluster determines a predicted label for the given user profile (1204-1208). In some implementations, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with step 804 as described above with reference to FIG. 8. However, in steps 1204-1208, the determination of the predicted label for the given user profile can be performed over secret shares, so as to provide user data privacy protection. To determine the predicted label for the given user profile, the first computing system of the MPC cluster (i) determines a first share of the predicted label based at least in part on the first share of the given user profile, a first machine learning model trained using multiple user profiles, and one or more of multiple true labels for the multiple user profiles (1204), (ii) receives, from a second computing system of the MPC cluster, data indicating a second share of the predicted label determined by the second computing system of the MPC cluster based at least in part on a second share of the given user profile and a first set of one or more machine learning models, and (iii) determines the predicted label based at least in part on the first and second shares of the predicted label (1208). For example, the second computing system of the MPC cluster may correspond to MPC2 of the MPC cluster 130 of FIG. 1.


In this example, the multiple true labels for the multiple user profiles may correspond to true labels that are included as part of encrypted label data 626, which are the true labels for the multiple user profiles that were used to train and/or evaluate the first machine learning model 620. In some examples, the multiple true labels may correspond to shares of another set of true labels. The one or more true labels, from among the multiple true labels, on which the determination of the predicted label for the given user profile is based, for instance, may include at least one true label for each of k nearest neighbor user profiles identified by way of the k-NN model 622 of the first machine learning model 620. In some examples, each of the multiple true labels is encrypted, as is the case in the examples of FIGS. 6-7. Some of the various ways in which true labels for k nearest neighbor user profiles can be leveraged to determine predicted labels are described in detail above. As made evident in the above, the way or manner in which such true labels are leveraged to determine predicted labels may at least in part depend on the type(s) of inference technique(s) that are employed (e.g., regression techniques, binary classification techniques, multiclass classification techniques, etc.). Additional details regarding secret share exchanges that may be performed in association with k-NN computations are provided above with reference to FIGS. 1-5.


The first computing system of the MPC cluster determines a predicted residue value indicating a predicted error in the predicted label (1210-1214). In some implementations, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with step 806 as described above with reference to FIG. 8. However, in steps 1210-1214, the determination of the predicted reside value can be performed over secret shares, so as to provide user data privacy protection. To determine the predicted residue value, the first computing system of the MPC cluster (i) determines a first share of the predicted residue value for the given user profile based at least in part on the first share of the given user profile and a second machine learning model trained using the multiple user profiles and data indicating differences between the multiple true labels for the multiple user profiles and multiple predicted labels as determined for the multiple user profiles using the first machine learning model (1210), (ii) receives, from the second computing system of the MPC cluster, data indicating a second share of the predicted residue value for the given user profile determined by the second computing system of the MPC cluster based at least in part on the second share of the given user profile and a second set of one or more machine learning models (1212), and (iii) determines the predicted residue value for the given user profile based at least in part on the first and second shares of the predicted residue value (1214).


The first computing system of the MPC cluster generates data representing an inference result based on the predicted label and the predicted residue value (1216). In some implementations, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with step 808 as described above with reference to FIG. 8. As such, in some examples, the inference result includes or corresponds to a sum of the predicted label and the predicted residue value.


The first computing system of the MPC cluster provides the data representing the inference result to a client device (1218). In some implementations, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with step 810 as described above with reference to FIG. 8. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster 130 providing an inference result to the client device 110 on which the application 112 runs, as described above with reference to FIGS. 1-2.


In some implementations, the process 1200 further includes one or more operations in which the first computing system of the MPC cluster applies a transformation to the first share of the given user profile to obtain a first transformed share of the given user profile. In these implementations, to determine the predicted label, the first computing system of the MPC cluster determines a first share of the predicted label based at least in part on the first transformed share of the given user profile. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with random projection logic 610 being employed to apply a random projection transformation to the user profile 609 (Pi) to obtain the transformed user profile 619 (Pi′), as described above with reference to FIGS. 6-8.


In at least some of the aforementioned implementations, to determine the first share of the predicted label, the first computing system of the MPC cluster provides the first transformed share of the given user profile as input to the first machine learning model to obtain the first share of the predicted label for the given user profile as output. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the first machine learning model 620 receiving transformed user profile 619 (Pi′) as input and generating at least one predicted label 629 (custom-character) in response thereto, as described above with reference to FIGS. 6-7.


In some examples, the aforementioned transformation may be a random projection. Furthermore, in at least some of these examples, the aforementioned random projection may be a Johnson-Lindenstrauss (J-L) transformation.


In some implementations, to apply the J-L transformation, the MPC cluster can generate a project matrix R in ciphertext. To project n dimensional Pi to k dimension, the MPC cluster can generate an n×k random matrix R. For example, the first computing system (e.g., MPC1) can create n×k random matrix A, where Ai,j=1 with 50% probability and Ai,j=0 with 50% probability. The first computing system can split A into two shares [A1] and [A2], discard A, keep [A1] confidentially, and give [A2] to the second computing system (e.g., MPC2). Similarly, the second computing system can create n×k random matrix B, the elements of which have the same distribution of the elements of A. The second computing system can split B into two shares [B1] and [B2], discard B, keep [B2] confidentially, and give [B1] to the first computing system.


The first computing system can then calculate [R1] as 2×([A1]==[B1]) −1. Similarly, the second computing system can then calculate [R2] as 2×([A2]==[B2])−1. In this way, [R1] and [R2] are two secret shares of R whose elements are either 1 or −1 with equal probability.


The actual random projection is between secret shares of Pi of dimension 1×n and projection matrix R of dimension n×k to produce results of 1×k. Assuming that n>>k, the J-L transformation reduces the dimension of training data from n to k. To carry out the above projection in encrypted data, the first computing system can calculate [Pi, 1]⊙[Ri, 1], which requires multiplication between two shares and addition between two shares.


As mentioned above, in some implementations, the first machine learning model includes a k-nearest neighbor model maintained by the first computing system of the MPC cluster, and the first set of one or more machine learning models includes a k-nearest neighbor model maintained by the second computing system of the MPC cluster. In some examples, the two aforementioned k-nearest neighbor models may be identical or nearly identical to one another. That is, in some examples, the first and second computing systems maintain copies of the same k-NN model, and each store their own shares of true labels. In some examples, a model rooted in one or more prototype methods may be implemented in place of one or both of the aforementioned k-nearest neighbor models.


In at least some of these implementations, to determine the predicted label, the first computing system of the MPC cluster (i) identifies a first set of nearest neighbor user profiles based at least in part on the first share of the given user profile and the k-nearest neighbor model maintained by the first computing system of the MPC cluster, (ii) receives, from the second computing system of the MPC cluster, data indicating a second set of nearest neighbor profiles identified by the second computing system of the MPC cluster based at least in part on the second share of the given user profile and the k-nearest neighbor model maintained by the second computing system of the MPC cluster, (iii) identifies a number k of nearest neighbor user profiles that are considered most similar to the given user profile among the multiple user profiles based at least in part on the first and second sets of nearest neighbor profiles, and determines the first share of the predicted label based at least in part on a true label for each of the k nearest neighbor user profiles. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the first machine learning model 620 being utilized to obtain at least one predicted label 629 (custom-character) in one or more implementations where one or more regression and/or binary classification techniques are employed, as described above with reference to FIGS. 6-8. In some examples, the predicted label includes or corresponds to the sum of the true labels for the k nearest neighbor user profiles.


In some of the aforementioned implementations, to determine the first share of the predicted label, the first computing system of the MPC cluster (i) determines a first share of a sum of the true labels for the k nearest neighbor user profiles, (ii) receives, from the second computing system of the MPC cluster, a second share of the sum of the true labels for the k nearest neighbor user profiles, and (iii) determines the sum of the true labels for the k nearest neighbor user profiles based at least in part on the first and second shares of the sum of the true labels for the k nearest neighbor user profiles. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the first machine learning model 620 being utilized to obtain at least one predicted label 629 (custom-character) in one or more implementations where one or more multiclass classification techniques are employed, as described above with reference to FIGS. 6-8.


In some implementations, the second machine learning model includes at least one of a deep neural network (DNN), a gradient-boosting decision tree (GBDT), and a random forest model maintained by the first computing system of the MPC cluster, and the second set of one or more machine learning models includes at least one of a DNN, a GBDT, and a random forest model maintained by the second computing system of the MPC cluster. In some examples, the two models (e.g., DNNs, GBDTs, random forest models, etc.) maintained by the first and second computing systems may be identical or nearly identical to one another.


In some implementations, the process 1200 further includes one or more operations in which the MPC cluster evaluates a performance of the first machine learning model and trains the second machine learning model using data indicating the predicted residue values determined for the multiple user profiles in evaluating the performance of the first machine learning model. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster performing step 920 as described above with reference to FIGS. 8-9.


However, in such implementations, one or more operations can be performed over secret shares, so as to provide user data privacy protection. In these implementations, to evaluate the performance of the first machine learning model, for each of the multiple user profiles, the MPC cluster determines a predicted label for the user profile and determines a residue value for the user profile indicating a prediction error in the predicted label. To determine the predicted label for the user profile, the first computing system of the MPC cluster (i) determines a first share of a predicted label for the user profile based at least in part on a first share of the user profile, the first machine learning model, and one or more of the multiple true labels for the multiple user profiles, (ii) receives, from the second computing system of the MPC cluster, data indicating a second share of the predicted label for the user profile determined by the second computing system of the MPC cluster based at least in part on a second share of the user profile and the first set of one or more machine learning models maintained by the second computing system of the MPC cluster, and (iii) determines the predicted label for the user profile based at least in part on the first and second shares of the predicted label. To determine the residue value for the user profile indicating the error in the predicted label, the first computing system of the MPC cluster (i) determines a first share of the residue value for the user profile based at least in part on the predicted label determined for the user profile and a first share of a true label for the user profile included in the multiple true labels, (ii) receives, from the second computing system of the MPC cluster, data indicating a second share of the residue value for the user profile determined by the second computing system of the MPC cluster based at least in part on the predicted label determined for the user profile and a second share of the true label for the user profile, and (iii) determines the residue value for the user profile based at least in part on the first and second shares of the residue value. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster performing steps 1108-1106 as described above with reference to FIG. 11. In addition, in these implementations, the process 1200 further includes one or more operations in which the MPC cluster trains the second machine learning model using data indicating the residue values determined for the multiple user profiles in evaluating the performance of the first machine learning model. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster performing step 930 as described above with reference to FIG. 9.


In at least some of the aforementioned implementations, the first share of the residue value for the user profile is indicative of a difference in value between the predicted label determined for the user profile by the first machine learning model and the first share of the true label for the user profile, and the second share of the residue value for the user profile is indicative of a difference in value between the predicted label determined for the user profile by the first machine learning model and the second share of the true label for the user profile. For instance, this may be the case for examples in which regression techniques are employed.


In at least some of the aforementioned implementations, before the MPC cluster evaluates the performance of the first machine learning model, the process 1200 further includes one or more operations in which the MPC cluster (i) derives a function and (ii) configures the first machine learning model to, given a user profile as input, generate an initial predicted label for the user profile and apply the function to the initial predicted label for the user profile to generate, as output, a first share of a predicted label for the user profile.


For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster performing steps 914-916 as described above with reference to FIGS. 8-9. To derive the function, the first computing system of the MPC cluster (i) derives a first share of the function based at least in part on a first share of each of the multiple true labels, (ii) receives, from the second computing system of the MPC cluster, data indicating a second share of the function derived by the second computing system of the MPC cluster based at least in part on a second share of each of the multiple true labels, and (iii) derives the function based at least in part on the first and second shares of the function. For example, for examples in which binary classification techniques are employed, this may correspond to one or more operations in which the MPC cluster configures the first machine learning model to, given a user profile as input: (i) calculate a sum of the true labels for the k nearest neighbor user profiles (sum_of_labels), and (ii) apply the function (transformation ƒ) to the initial predicted label for the user profile to generate, as output, a predicted label for the user profile (custom-character=ƒ(sum_of_labels)). Similar operations may be performed for cases in which multiclass classification techniques are employed.


When implemented over secret shares, the first computing system (e.g., MPC1) can calculate:





[sum0,1]=Σij∈k−NN to Pi[Lj,1])×(1−[Li,1])





[count0,1]=Σi(1−[Li,1])





[sum_of_square0,1]=Σij∈k−NN to Pi[Lj,1])2×(1−[Li,1])


Similarly, when implemented over secret shares, the second computing system (e.g., MPC2) can calculate:





[sum0,2]=Σij∈k−NN to Pi[j,2])×(1−[Li,2])





[count0,2]=Σi(1−[Li,2])





[sum_of_square0,2]=Σij∈k−NN to Pi[Lj,2])2×(1−[Li,2])


The MPC cluster can then reconstruct sum0, count0, sum_of_square0 as described above in cleartext, and calculate distribution custom-character0, σ02).


Similarly, to calculate distribution custom-character1, σ12), the first computing system (e.g., MPC1) can calculate:





[sum1,1]=Σij∈k−NN to Pi[j,1])×(1−[Li,1])





[count1,1]=Σi(1−[Li,1])





[sum_of_square1,1]=Σij∈k−NN to Pi[Lj,1])2×(1−[Li,1])


And, the second computing system (e.g., MPC2) can calculate:





[sum1,2]=Σij∈k−NN to Pi[j,2])×(1−[Li,2])





[count1,2]=Σi(1−[Li,2])





[sum_of_square1,2]=Σij∈k−NN to Pi[Lj,2])2×(1−[Li,2])


The MPC cluster can then reconstruct sum1, count1, sum_of_square1 as described above in cleartext, and calculate distribution N(μ1, σ12).


In at least some of the aforementioned implementations, when evaluating the performance of the first machine learning model, the MPC cluster can employ one or more fixed point calculation techniques to determine the residue value for each user profile. More specifically, when evaluating the performance of the first machine learning model, to determine the first share of the residue value for each user profile, the first computing system of the MPC cluster scales the corresponding true label, or share thereof, by a particular scaling factor, scales coefficients {a2, a1, a0} associated with the function by the particular scaling factor and rounds the scaled coefficients to the nearest integer. In such implementations, the second computing system of the MPC cluster may perform similar operations to determine the second share of the residue value for each user profile. The MPC cluster can thus calculate the residue value with secret shares, reconstruct the cleartext residue value from the two secret shares, and divide the cleartext residue value by the scaling factor.


In at least some of the aforementioned implementations, the process 1200 further includes one or more operations in which the first computing system of the MPC cluster estimates a first share of a set of distribution parameters based at least in part on the first share of each of the multiple true labels. In some such implementations, to derive the first share of the function based at least in part on the first share of each of the multiple true labels, the first computing system of the MPC cluster derives a first share of the function based at least in part on the first share of the set of distribution parameters. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the MPC cluster performing steps 912-914 as described above with reference to FIGS. 8-9. As such, the aforementioned set of distribution parameters can include one or more parameters of a probability distribution of prediction errors for true labels of a first value in the multiple true labels, e.g., a mean (μ0) and a variance (σ0) of a normal distribution of prediction errors for true labels of a first value in the multiple true labels, and one or more parameters of a probability distribution of prediction errors for true labels of a second value in the multiple true labels, e.g., a mean (μ1) and a variance (σ1) of a normal distribution of prediction errors for true labels of a second, different value in the multiple true labels. As described above, in some examples, the aforementioned set of distribution parameters can include other types of parameters. Furthermore, in at least some of the aforementioned implementations, the function is a second degree polynomial function, e.g., ƒ(x)=a2×2+a1×+a0, where ƒ′(x)=2a2×+a1, although, in some examples, other functions may be employed.


In some examples, to determine the first share of the predicted label, the first computing system of the MPC cluster (i) determines a first share of a sum of the true labels for the k nearest neighbor user profiles, (ii) receives, from the second computing system of the MPC cluster, a second share of the sum of the true labels for the k nearest neighbor user profiles, and (iii) determines the sum of the true labels for the k nearest neighbor user profiles based at least in part on the first and second shares of the sum of the true labels for the k nearest neighbor user profiles. For instance, this may be the case for implementations in which regression or binary classification techniques are employed. In some of the aforementioned examples, the first share of the predicted label may correspond to the sum of the true labels for the k nearest neighbor user profiles. For instance, this may be the case for implementations in which regression classification techniques are employed (custom-character=sum_of_labels). In other such examples, to determine the first share of the predicted label, the MPC cluster applies a function to the sum of the true labels for the k nearest neighbor user profiles to generate a predicted label for the given user profile. For instance, this may be the case for implementations in which binary classification techniques are employed (custom-character=ƒ(sum_of_labels)).


As mentioned above, in some of the aforementioned implementations, to determine the first share of the predicted label based at least in part on the true label for each of the k nearest neighbor user profiles, the first computing system of the MPC cluster determines a first share of a set of predicted labels based at least in part on a set of true labels for each of the k nearest neighbor user profiles corresponding to a set of categories. To determine the first share of a set of predicted labels, for each category in the set, the first computing system of the MPC cluster (i) determines a first share of a frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of a first value, (ii) receives, a second share of the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value, and (iii) determines the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value based at least in part on the first and second shares of the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value. Such operations can include one or more operations in which the first computing system of the MPC cluster determines a frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of a first value. For example, this may correspond to one or more operations that are similar or equivalent to one or more operations that are performed in connection with the first machine learning model 620 being utilized to obtain at least one predicted label 629 (custom-character) in one or more implementations where one or more multiclass classification techniques are employed, as described above with reference to FIGS. 6-8.


In at least some of the aforementioned implementations, to determine the first share of the set of predicted labels, for each category in the set, the first computing system of the MPC cluster applies a function corresponding to the category to the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value to generate a first share of a predicted label corresponding to the category for the given user profile. For instance, the respective function may correspond to one of w different functions derived by the MPC cluster for w different categories as described above with reference to step 914 of FIGS. 8-9.


For multiclass classification problems, when evaluating the performance (e.g., quality) of the first machine learning model, for each training example/query, the MPC cluster can find the k nearest neighbors and calculate the frequency of their labels over secret shares.


For instance, consider an example in which there are assumed to be w valid labels (e.g., classes) for a multiclass classification problem {l1, l2, . . . lw}. Among the k neighbors identified by {id1, id2, . . . idk}, the first computing system (e.g., MPC1) can calculate the frequency of the jth label as [lj,1] as:





[frequencyj,1]=Σi∈{id1, . . . idk}([labeli,1]==lj)


The first computing system can calculate the frequency from the true label [label1] as:





[expected_frequencyj,1]=k×([label1]==j)


Therefore, the first computing system can calculate:





[Residuej,1]=[expected_frequencyj,1]−[frequencyj,1]


And, [Residuej,1] is equivalent to:





[Residuej,1]=k×([label1]==j)−Σi∈{id1, . . . idk}([labeli,1]==lj)


Similarly, the second computing system (e.g., MPC2) can calculate:





[Residuej,2]=k×([label2]==j)−Σi∈{id1, . . . idk}([labeli,2]==lj)


In the case of binary classification and regression, for each inference, the residue value can be a secret message of integer type. On the contrary, in the case of multiclass classification, for each inference, the residue value can be a secret message of integer vectors, as shown above.



FIG. 13 is a block diagram of an example computer system 1300 that can be used to perform operations described above. The system 1300 includes a processor 1310, a memory 1320, a storage device 1330, and an input/output device 1340. Each of the components 1310, 1320, 1330, and 1340 can be interconnected, for example, using a system bus 1350. The processor 1310 is capable of processing instructions for execution within the system 1300. In some implementations, the processor 1310 is a single-threaded processor.


In another implementation, the processor 1310 is a multi-threaded processor. The processor 1310 is capable of processing instructions stored in the memory 1320 or on the storage device 1330.


The memory 1320 stores information within the system 1300. In one implementation, the memory 1320 is a computer-readable medium. In some implementations, the memory 1320 is a volatile memory unit. In another implementation, the memory 1320 is a non-volatile memory unit.


The storage device 1330 is capable of providing mass storage for the system 1300. In some implementations, the storage device 1330 is a computer-readable medium. In various different implementations, the storage device 1330 can include, for example, a hard disk device, an optical disk device, a storage device that is shared over a network by multiple computing devices (e.g., a cloud storage device), or some other large capacity storage device.


The input/output device 1340 provides input/output operations for the system 1300. In some implementations, the input/output device 1340 can include one or more of a network interface devices, e.g., an Ethernet card, a serial communication device, e.g., and RS-232 port, and/or a wireless interface device, e.g., and 802.11 card. In another implementation, the input/output device can include driver devices configured to receive input data and send output data to external devices 1360, e.g., keyboard, printer and display devices. Other implementations, however, can also be used, such as mobile computing devices, mobile communication devices, set-top box television client devices, etc.


Although an example processing system has been described in FIG. 13, implementations of the subject matter and the functional operations described in this specification can be implemented in other types of digital electronic circuitry, or in computer software, firmware, or hardware, including the structures disclosed in this specification and their structural equivalents, or in combinations of one or more of them.


Embodiments of the subject matter and the operations described in this specification can be implemented in digital electronic circuitry, or in computer software, firmware, or hardware, including the structures disclosed in this specification and their structural equivalents, or in combinations of one or more of them. Embodiments of the subject matter described in this specification can be implemented as one or more computer programs, i.e., one or more modules of computer program instructions, encoded on computer storage media (or medium) for execution by, or to control the operation of, data processing apparatus. Alternatively, or in addition, the program instructions can be encoded on an artificially-generated propagated signal, e.g., a machine-generated electrical, optical, or electromagnetic signal, that is generated to encode information for transmission to suitable receiver apparatus for execution by a data processing apparatus. A computer storage medium can be, or be included in, a computer-readable storage device, a computer-readable storage substrate, a random or serial access memory array or device, or a combination of one or more of them. Moreover, while a computer storage medium is not a propagated signal, a computer storage medium can be a source or destination of computer program instructions encoded in an artificially-generated propagated signal. The computer storage medium can also be, or be included in, one or more separate physical components or media (e.g., multiple CDs, disks, or other storage devices).


The operations described in this specification can be implemented as operations performed by a data processing apparatus on data stored on one or more computer-readable storage devices or received from other sources.


The term “data processing apparatus” encompasses all kinds of apparatus, devices, and machines for processing data, including by way of example a programmable processor, a computer, a system on a chip, or multiple ones, or combinations, of the foregoing. The apparatus can include special purpose logic circuitry, e.g., an FPGA (field programmable gate array) or an ASIC (application-specific integrated circuit). The apparatus can also include, in addition to hardware, code that creates an execution environment for the computer program in question, e.g., code that constitutes processor firmware, a protocol stack, a database management system, an operating system, a cross-platform runtime environment, a virtual machine, or a combination of one or more of them. The apparatus and execution environment can realize various different computing model infrastructures, such as web services, distributed computing and grid computing infrastructures.


A computer program (also known as a program, software, software application, script, or code) can be written in any form of programming language, including compiled or interpreted languages, declarative or procedural languages, and it can be deployed in any form, including as a stand-alone program or as a module, component, subroutine, object, or other unit suitable for use in a computing environment. A computer program may, but need not, correspond to a file in a file system. A program can be stored in a portion of a file that holds other programs or data (e.g., one or more scripts stored in a markup language document), in a single file dedicated to the program in question, or in multiple coordinated files (e.g., files that store one or more modules, sub-programs, or portions of code). A computer program can be deployed to be executed on one computer or on multiple computers that are located at one site or distributed across multiple sites and interconnected by a communication network.


The processes and logic flows described in this specification can be performed by one or more programmable processors executing one or more computer programs to perform actions by operating on input data and generating output. The processes and logic flows can also be performed by, and apparatus can also be implemented as, special purpose logic circuitry, e.g., an FPGA (field programmable gate array) or an ASIC (application-specific integrated circuit).


Processors suitable for the execution of a computer program include, by way of example, both general and special purpose microprocessors. Generally, a processor will receive instructions and data from a read-only memory or a random access memory or both. The essential elements of a computer are a processor for performing actions in accordance with instructions and one or more memory devices for storing instructions and data. Generally, a computer will also include, or be operatively coupled to receive data from or transfer data to, or both, one or more mass storage devices for storing data, e.g., magnetic, magneto-optical disks, or optical disks. However, a computer need not have such devices. Moreover, a computer can be embedded in another device, e.g., a mobile telephone, a personal digital assistant (PDA), a mobile audio or video player, a game console, a Global Positioning System (GPS) receiver, or a portable storage device (e.g., a universal serial bus (USB) flash drive), to name just a few. Devices suitable for storing computer program instructions and data include all forms of non-volatile memory, media and memory devices, including by way of example semiconductor memory devices, e.g., EPROM, EEPROM, and flash memory devices; magnetic disks, e.g., internal hard disks or removable disks; magneto-optical disks; and CD-ROM and DVD-ROM disks. The processor and the memory can be supplemented by, or incorporated in, special purpose logic circuitry.


To provide for interaction with a user, embodiments of the subject matter described in this specification can be implemented on a computer having a display device, e.g., a CRT (cathode ray tube) or LCD (liquid crystal display) monitor, for displaying information to the user and a keyboard and a pointing device, e.g., a mouse or a trackball, by which the user can provide input to the computer. Other kinds of devices can be used to provide for interaction with a user as well; for example, feedback provided to the user can be any form of sensory feedback, e.g., visual feedback, auditory feedback, or tactile feedback; and input from the user can be received in any form, including acoustic, speech, or tactile input. In addition, a computer can interact with a user by sending documents to and receiving documents from a device that is used by the user; for example, by sending web pages to a web browser on a user's client device in response to requests received from the web browser.


Embodiments of the subject matter described in this specification can be implemented in a computing system that includes a back-end component, e.g., as a data server, or that includes a middleware component, e.g., an application server, or that includes a front-end component, e.g., a client computer having a graphical user interface or a Web browser through which a user can interact with an implementation of the subject matter described in this specification, or any combination of one or more such back-end, middleware, or front-end components. The components of the system can be interconnected by any form or medium of digital data communication, e.g., a communication network. Examples of communication networks include a local area network (“LAN”) and a wide area network (“WAN”), an inter-network (e.g., the Internet), and peer-to-peer networks (e.g., ad hoc peer-to-peer networks).


The computing system can include clients and servers. A client and server are generally remote from each other and typically interact through a communication network. The relationship of client and server arises by virtue of computer programs running on the respective computers and having a client-server relationship to each other. In some embodiments, a server transmits data (e.g., an HTML page) to a client device (e.g., for purposes of displaying data to and receiving user input from a user interacting with the client device). Data generated at the client device (e.g., a result of the user interaction) can be received from the client device at the server.


While this specification contains many specific implementation details, these should not be construed as limitations on the scope of any inventions or of what may be claimed, but rather as descriptions of features specific to particular embodiments of particular inventions. Certain features that are described in this specification in the context of separate embodiments can also be implemented in combination in a single embodiment. Conversely, various features that are described in the context of a single embodiment can also be implemented in multiple embodiments separately or in any suitable subcombination. Moreover, although features may be described above as acting in certain combinations and even initially claimed as such, one or more features from a claimed combination can in some cases be excised from the combination, and the claimed combination may be directed to a subcombination or variation of a subcombination.


Similarly, while operations are depicted in the drawings in a particular order, this should not be understood as requiring that such operations be performed in the particular order shown or in sequential order, or that all illustrated operations be performed, to achieve desirable results. In certain circumstances, multitasking and parallel processing may be advantageous. Moreover, the separation of various system components in the embodiments described above should not be understood as requiring such separation in all embodiments, and it should be understood that the described program components and systems can generally be integrated together in a single software product or packaged into multiple software products.


Thus, particular embodiments of the subject matter have been described. Other embodiments are within the scope of the following claims. In some cases, the actions recited in the claims can be performed in a different order and still achieve desirable results. In addition, the processes depicted in the accompanying figures do not necessarily require the particular order shown, or sequential order, to achieve desirable results. In certain implementations, multitasking and parallel processing may be advantageous.

Claims
  • 1. A computer-implemented method comprising: receiving, by a first computing system of a plurality of multi-party computation (MPC) computing systems, an inference request comprising a first share of a given user profile;determining a predicted label for the given user profile based at least in part on a first machine learning model trained using a plurality of user profiles;determining a predicted residue value for the given user profile indicating a predicted error in the predicted label, comprising: determining, by the first computing system, a first share of the predicted residue value for the given user profile based at least in part on the first share of the given user profile and a second machine learning model trained using the plurality of user profiles and data indicating differences between a plurality of true labels for the plurality of user profiles and a plurality of predicted labels as determined for the plurality of user profiles using the first machine learning model;receiving, by the first computing system and from a second computing system of the plurality of MPC computing systems, data indicating a second share of the predicted residue value for the given user profile determined by the second computing system based at least in part on the second share of the given user profile and a second set of one or more machine learning models; anddetermining the predicted residue value for the given user profile based at least in part on the first and second shares of the predicted residue value;generating, by the first computing system, a first share of an inference result based at least in part on the predicted label determined for the given user profile and the predicted residue value; andproviding, by the first computing system and to a client device, the first share of the inference result and a second share of the inference result received from the second computing system.
  • 2. The computer-implemented method of claim 1, wherein determining the predicted label for the given user profile comprises: determining, by the first computing system, a first share of the predicted label based at least in part on (i) the first share of the given user profile, (ii) the first machine learning model trained using the plurality of user profiles, and (iii) one or more of the plurality of true labels for the plurality of user profiles, the plurality of true labels including one or more true labels for each user profile in the plurality of user profiles;receiving, by the first computing system and from the second computing system, data indicating a second share of the predicted label determined by the second computing system based at least in part on a second share of the given user profile and a first set of one or more machine learning models; anddetermining the predicted label based at least in part on the first and second shares of the predicted label.
  • 3. The computer-implemented method of claim 1, further comprising: applying, by the first computing system, a transformation to the first share of the given user profile to obtain a first transformed share of the given user profile, wherein determining, by the first computing system, the first share of the predicted label comprises: determining, by the first computing system, a first share of the predicted label based at least in part on the first transformed share of the given user profile.
  • 4. The computer-implemented method of claim 3, wherein the transformation comprises a Johnson-Lindenstrauss (J-L) transformation.
  • 5. The computer-implemented method of claim 3, wherein determining, by the first computing system, the first share of the predicted label comprises: providing, by the first computing system, the first transformed share of the given user profile as input to the first machine learning model to obtain a first share of the predicted label for the given user profile as output.
  • 6. The computer-implemented method of claim 1, further comprising: evaluating a performance of the first machine learning model, comprising, for each of the plurality of user profiles: determining a predicted label for the user profile, comprising: determining, by the first computing system, a first share of a predicted label for the user profile based at least in part on (i) a first share of the user profile, (ii) the first machine learning model, and (iii) one or more of the plurality of true labels for the plurality of user profiles;receiving, by the first computing system and from the second computing system, data indicating a second share of the predicted label for the user profile determined by the second computing system based at least in part on a second share of the user profile and the first set of one or more machine learning models maintained by the second computing system; anddetermining the predicted label for the user profile based at least in part on the first and second shares of the predicted label;determining a residue value for the user profile indicating an error in the predicted label, comprising: determining, by the first computing system, a first share of the residue value for the user profile based at least in part on the predicted label determined for the user profile and a first share of a true label for the user profile included in the plurality of true labels;receiving, by the first computing system and from the second computing system, data indicating a second share of the residue value for the user profile determined by the second computing system based at least in part on the predicted label determined for the user profile and a second share of the true label for the user profile; anddetermining the residue value for the user profile based at least in part on the first and second shares of the residue value; andtraining the second machine learning model using data indicating the residue values determined for the plurality of user profiles in evaluating the performance of the first machine learning model.
  • 7. The computer-implemented method of claim 6, further comprising: before evaluating the performance of the first machine learning model: deriving a set of parameters of a function, comprising: deriving, by the first computing system, a first share of the set of parameters of the function based at least in part on a first share of each of the plurality of true labels;receiving, by the first computing system and from the second computing system, data indicating a second share of the set of parameters of the function derived by the second computing system based at least in part on a second share of each of the plurality of true labels; andderiving the set of parameters of the function based at least in part on the first and second shares of the set of parameters of the function; andconfiguring the first machine learning model to, given a user profile as input, generate an initial predicted label for the user profile and apply the function, as defined based on the derived set of parameters, to the initial predicted label for the user profile to generate, as output, a first share of a predicted label for the user profile.
  • 8. The computer-implemented method of claim 7, further comprising: estimating, by the first computing system, a first share of a set of distribution parameters based at least in part on the first share of each of the plurality of true labels, wherein deriving, by the first computing system, the first share of the set of parameters of the function based at least in part on the first share of each of the plurality of true labels comprises: deriving, by the first computing system, a first share of the set of parameters of the function based at least in part on the first share of the set of distribution parameters.
  • 9. The computer-implemented method of claim 8, wherein the set of distribution parameters include one or more parameters of a probability distribution of prediction errors for true labels of a first value in the plurality of true labels, and one or more parameters of a probability distribution of prediction errors for true labels of a second value in the plurality of true labels, the second value being different from the first value.
  • 10. The computer-implemented method of claim 6, wherein: the first share of the residue value for the user profile is indicative of a difference in value between the predicted label determined for the user profile and the first share of the true label for the user profile; andthe second share of the residue value for the user profile is indicative of a difference in value between the predicted label determined for the user profile and the second share of the true label for the user profile.
  • 11. The computer-implemented method of claim 1, wherein: the first machine learning model includes a k-nearest neighbor model maintained by the first computing system;the first set of one or more machine learning models includes a k-nearest neighbor model maintained by the second computing system;the second machine learning model includes at least one of a deep neural network (DNN) maintained by the first computing system and a gradient-boosting decision tree (GBDT) maintained by the first computing system; andthe second set of one or more machine learning models includes at least one of a DNN maintained by the second computing system and a GBDT maintained by the second computing system.
  • 12. The computer-implemented method of claim 11, wherein determining, by the first computing system, the first share of the predicted label comprises: identifying, by the first computing system, a first set of nearest neighbor user profiles based at least in part on the first share of the given user profile and the k-nearest neighbor model maintained by the first computing system;receiving, by the first computing system and from the second computing system, data indicating a second set of nearest neighbor profiles identified by the second computing system based at least in part on the second share of the given user profile and the k-nearest neighbor model maintained by the second computing system;identifying a number k of nearest neighbor user profiles that are considered most similar to the given user profile among the plurality of user profiles based at least in part on the first and second sets of nearest neighbor profiles; anddetermining, by the first computing system, the first share of the predicted label based at least in part on a true label for each of the k nearest neighbor user profiles.
  • 13. The computer-implemented method of claim 12, wherein determining, by the first computing system, the first share of the predicted label further comprises: determining, by the first computing system, a first share of a sum of the true labels for the k nearest neighbor user profiles;receiving, by the first computing system and from the second computing system, a second share of the sum of the true labels for the k nearest neighbor user profiles; anddetermining the sum of the true labels for the k nearest neighbor user profiles based at least in part on the first and second shares of the sum of the true labels for the k nearest neighbor user profiles.
  • 14. The computer-implemented method of claim 13, wherein determining, by the first computing system, the first share of the predicted label further comprises: applying a function to the sum of the true labels for the k nearest neighbor user profiles to generate the first share of the predicted label for the given user profile.
  • 15. The computer-implemented method of claim 13, wherein the first share of the predicted label for the given user profile comprises the sum of the true labels for the k nearest neighbor user profiles.
  • 16. The computer-implemented method of claim 12, wherein determining, by the first computing system, the first share of the predicted label based at least in part on the true label for each of the k nearest neighbor user profiles comprises: determining, by the first computing system, a first share of a set of predicted labels based at least in part on a set of true labels for each of the k nearest neighbor user profiles corresponding to a set of categories, respectively, comprising, for each category in the set: determining a first share of a frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of a first value;receiving, by the first computing system and from the second computing system, a second share of the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value; anddetermining the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value based at least in part on the first and second shares of the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value.
  • 17. The computer-implemented method of claim 16, wherein determining, by the first computing system, the first share of the set of predicted labels comprises, for each category in the set: applying a function corresponding to the category to the frequency at which true labels that correspond to the category in the sets of true labels for user profiles in the k nearest neighbor user profiles are true labels of the first value to generate a first share of a predicted label corresponding to the category for the given user profile.
  • 18. The computer-implemented method of claim 1, wherein the client device computes the given user profile using a plurality of feature vectors that each include feature values related to events of a user of the client device and a decay rate for each feature vector.
  • 19. The computer-implemented method of claim 1, wherein the client device computes the given user profile using a plurality of feature vectors that each include feature values related to events of a user of the client device, wherein computing the given user profile comprises: classifying one or more of the plurality of feature vectors as sparse feature vectors; andclassifying one or more of the plurality of feature vectors as dense feature vectors, the method further comprising:generating the first share of the given user profile and respective second shares of the given user profile for the one or more second computing systems using the sparse feature vectors and dense feature vectors, wherein generating the first share and the respective one or more second shares of the given user profile comprises splitting the sparse feature vector using a Function Secret Sharing (FSS) technique.
  • 20. A system comprising: one or more processors; andone or more storage devices storing instructions that, when executed by the one or more processors, cause the one or more processors to perform operations comprising: receiving, by a first computing system of a plurality of multi-party computation (MPC) computing systems, an inference request comprising a first share of a given user profile;determining a predicted label for the given user profile based at least in part on a first machine learning model trained using a plurality of user profiles;determining a predicted residue value for the given user profile indicating a predicted error in the predicted label, comprising: determining, by the first computing system, a first share of the predicted residue value for the given user profile based at least in part on the first share of the given user profile and a second machine learning model trained using the plurality of user profiles and data indicating differences between a plurality of true labels for the plurality of user profiles and a plurality of predicted labels as determined for the plurality of user profiles using the first machine learning model;receiving, by the first computing system and from a second computing system of the plurality of MPC computing systems, data indicating a second share of the predicted residue value for the given user profile determined by the second computing system based at least in part on the second share of the given user profile and a second set of one or more machine learning models; anddetermining the predicted residue value for the given user profile based at least in part on the first and second shares of the predicted residue value;generating, by the first computing system, a first share of an inference result based at least in part on the predicted label determined for the given user profile and the predicted residue value; andproviding, by the first computing system and to a client device, the first share of the inference result and a second share of the inference result received from the second computing system.
  • 21. A non-transitory computer readable storage medium carrying instructions that, when executed by one or more processors, cause the one or more processors to perform operations comprising: receiving, by a first computing system of a plurality of multi-party computation (MPC) computing systems, an inference request comprising a first share of a given user profile:determining a predicted label for the given user profile based at least in part on a first machine learning model trained using a plurality of user profiles:determining a predicted residue value for the given user profile indicating a predicted error in the predicted label, comprising: determining, by the first computing system, a first share of the predicted residue value for the given user profile based at least in part on the first share of the given user profile and a second machine learning model trained using the plurality of user profiles and data indicating differences between a plurality of true labels for the plurality of user profiles and a plurality of predicted labels as determined for the plurality of user profiles using the first machine learning model;receiving, by the first computing system and from a second computing system of the plurality of MPC computing systems, data indicating a second share of the predicted residue value for the given user profile determined by the second computing system based at least in part on the second share of the given user profile and a second set of one or more machine learning models; anddetermining the predicted residue value for the given user profile based at least in part on the first and second shares of the predicted residue value;generating, by the first computing system, a first share of an inference result based at least in part on the predicted label determined for the given user profile and the predicted residue value; andproviding, by the first computing system and to a client device, the first share of the inference result and a second share of the inference result received from the second computing system.
  • 22. (canceled)
Priority Claims (1)
Number Date Country Kind
277910 Oct 2020 IL national
PCT Information
Filing Document Filing Date Country Kind
PCT/US2021/054183 10/8/2021 WO