A method for managing model updates by a first network device includes receiving, at the first network device associated with a first zone model of multiple zone models, a global model from a second network device associated with the global model. The method also includes transmitting, from the first network device, the global model to user equipment (UEs) in a first group of UEs associated with the first zone model, a different group of UEs associated with each of the plurality of zone models. The method further includes receiving, at the first network device, weights associated with the global model from each UE in the first group. The method still further includes updating, at the first network device, the first zone model based on the received weights. The method also includes transmitting, from the first network device, the updated first zone model to each UE in the first group.