Multi-Stage Federated Learning in Wireless Networks

A processor of an artificial intelligence (AI) manager configured to perform operations comprising:

• selecting a first group of at least one AI agent to train and report, per each AI agent of the first group, a respective first partial AI or machine learning (ML) (AI/ML) model to the AI manager;
• receiving the respective first partial(AI/ML) model from the at least one AI agent of the first group;
• generating a first version of a global model from the respective first partial (AI/ML) model from the at least one AI agent of the first group;
• if the first version of the global model is determined to be trustworthy, selecting a second group of at least one AI agent to train and report, per each AI agent of the second group, a respective second partial AI/ML model to the AI manager;
• receiving the respective second partial (AI/ML)model from the at least one AI agent of the second group;
• and
• aggregating the respective second partial (AI/ML)model from the at least one AI agent of the second group and the first version of the global model into a second version of the global model.