International Business Machines Corporation
Federated Learning with Dataset Sketch Commitment Based Malicious Participant Identification

Abstract:

Mechanisms for performing intelligent federated machine learning (ML) model updates are provided. A plurality of ML model updates, and a plurality of dataset sketch commitment data structures (sketches), are received from a plurality of participant computing systems. Each sketch provides statistical characteristics of a corresponding local dataset used by a corresponding participant to train a local ML model. A potentially malicious participant identification operation is performed based on an analysis of the plurality of sketches to identify one or more potentially malicious participants based on differences in sketches. ML model updates received from participant computing systems identified as potentially malicious participants are discarded to thereby generate a modified set of updates. The federated ML computer model is updated based on the modified set of updates.