Generation of machine learning outputs using a domain index matrix to control a relative contribution of multiple different training data sets

Abstract

A method includes obtaining an input matrix and determining a domain index matrix that includes, for each respective input value of the input matrix, a corresponding domain index value that indicates a corresponding training data distribution of a plurality of training data distributions. The method also includes providing the input matrix and the domain index matrix to a machine learning model that has been trained using the plurality of training data distributions, where each respective training data distribution is associated with a different attribute. The method further includes generating, by the machine learning model and based on the input and the domain index matrices, an output matrix that includes, for each respective input value, a corresponding output value generated based on (i) the respective input value and (ii) the corresponding domain index value such that the corresponding output value exhibits the attribute of the corresponding training data distribution.