Federated learning for wireless communications systems
AssigneeApple Inc.
InventorsZhou Zhou, Sharad Sambhwani
Life of this patent
Estimating the term…
Abstract
A wireless communication system may use federated learning to develop a global model. The system may initialize parameters of user equipment (UE) models. A network node may select multiple UEs to participate in federated training. The selected UEs may choose one or more model layers to generate local models. The network node may aggregate the local models to generate a global model. The process may be repeated until the global model converges.
Claims20
A method for federated learning for a wireless communication system, the method comprising:
- initializing parameters of models for user equipment (UE) devices;
- selecting, via a network node, two or more of the UE devices as selected UE devices to participate in federated training;
- selecting, at the selected UE devices, one or more model layers as selected model layers to participate in the federated training;
- performing, at the selected UE devices, forward-forward learning on the selected model layers to generate local models for the selected model layers of the selected UE devices;
- aggregating, at the network node, the local models to generate a global model;
- iteratively selecting the two or more of the UE devices, selecting the one or more of the model layers, performing the forward-forward learning on the selected model layers, and aggregating the local models until the global model converges.
A network node apparatus comprising:
- a processor;
- and
- a memory storing instructions that, when executed by the processor, configure the apparatus to:
- initialize parameters of models for user equipment (UE) devices;
- select two or more of the UE devices as selected UE devices to participate in federated training;
- instruct the selected UE devices to select one or more model layers as selected model layers to participate in the federated training;
- receive local models generated by the selected UE devices performing forward-forward learning on the selected model layers;
- and
- aggregate, at the network node, the local models to generate a global model.
A user equipment (UE) apparatus comprising:
- a processor;
- and
- a memory storing instructions that, when executed by the processor, configure the apparatus to:
- initialize parameters of models for the UE apparatus;
- receive an indication from a network node that the UE apparatus is selected to participate in a federated training;
- select one or more model layers as selected model layers to participate in the federated training;
- perform forward-forward learning on the selected model layers to generate local models for the selected model layers;
- and
- send the local model to the network node for aggregation to generate a global model.
Dependencies are inferred from the claim text.