Systems, methods and computer-readable storage media utilized to train a machine-learning architecture. One method includes receiving, by one or more processing circuits, a data set. The method further includes determining, by the one or more processing circuits, a first portion of the data set associated with a plurality of entities. The method further includes training, by the one or more processing circuits and utilizing the first portion of the data set, an entity model. The method further includes determining, by the one or more processing circuits, a second portion of the data set associated with a first subset of entities and determining a second subset of entities. The method further includes freezing, by the one or more processing circuits, one or more parameters associated with the second subset of entities and training, utilizing the second portion of the data set, the entity model.