Zone-based federated learning

A system for training models, the system comprising a user equipment (UE) and a zone server:

• the UE comprising:
• a processor;
• a memory coupled with the processor;
• and
• instructions stored in the memory and operable, when executed by the processor, to cause the UE:
• to receive, at the UE associated with a group of UEs of a plurality of groups of UEs, a global model from the zone server associated with a first zone model of a plurality of zone models, the global model associated with a global server, the group of UEs associated with the first zone model, and each group of UEs associated with a respective zone model of the plurality of zone models;
• to identify, at the UE, a device for training the global model based on one or both of a current connectivity state of the UE or a current resource use of the UE, the device being the UE based on the current connectivity state of the UE satisfying a connectivity condition and the current resource use satisfying a resource condition, the device being a proxy device based on one or both of the current connectivity state of the UE failing to satisfy the connectivity condition or the current resource use failing to satisfy the resource condition;
• to transmit, to the zone server, model updates based on identifying the device;
• and
• to receive, from the zone server, the first zone model based on the transmitted model updates.