Predictive Modeling with Entity Representations Computed from Neural Network Models Simultaneously Trained on Multiple Tasks

摘要

This disclosure involves predictive modeling with entity representations computed from neural network models simultaneously trained on multiple tasks. For example, a method includes a processing device performing operations including accessing input data for an entity and transforming the input data into a dense vector entity representation representing the entity. Transforming the input data includes applying, to the input data, a neural network including simultaneously trained propensity models. Each propensity model predicts a different task based on the input data. Transforming the input data also includes extracting the dense vector entity representation from a common layer of the neural network to which the propensity models are connected. The operations performed by the processing device include computing a predicted behavior by applying a predictive model to the dense vector entity representation and transmitting the predicted behavior to a computing device that customizes a presentation of electronic content at a remote user device.